English

Bregman Centroid Guided Cross-Entropy Method

Machine Learning 2025-07-02 v2 Artificial Intelligence Systems and Control Systems and Control

Abstract

The Cross-Entropy Method (CEM) is a widely adopted trajectory optimizer in model-based reinforcement learning (MBRL), but its unimodal sampling strategy often leads to premature convergence in multimodal landscapes. In this work, we propose Bregman Centroid Guided CEM (BC\mathcal{BC}-EvoCEM), a lightweight enhancement to ensemble CEM that leverages Bregman centroids\textit{Bregman centroids} for principled information aggregation and diversity control. \textbf{\mathcal{BC}-EvoCEM} computes a performance-weighted Bregman centroid across CEM workers and updates the least contributing ones by sampling within a trust region around the centroid. Leveraging the duality between Bregman divergences and exponential family distributions, we show that \textbf{\mathcal{BC}-EvoCEM} integrates seamlessly into standard CEM pipelines with negligible overhead. Empirical results on synthetic benchmarks, a cluttered navigation task, and full MBRL pipelines demonstrate that \textbf{\mathcal{BC}-EvoCEM} enhances both convergence and solution quality, providing a simple yet effective upgrade for CEM.

Cite

@article{arxiv.2506.02205,
  title  = {Bregman Centroid Guided Cross-Entropy Method},
  author = {Yuliang Gu and Hongpeng Cao and Marco Caccamo and Naira Hovakimyan},
  journal= {arXiv preprint arXiv:2506.02205},
  year   = {2025}
}
R2 v1 2026-07-01T02:55:24.490Z