Diffusion Policy With Bayesian Expert Selection For Active Multi-target Tracking

Abstract

Active multi-target tracking requires a mobile robot to balance exploration for undetected targets with exploitation of uncertain tracked ones. Diffusion policies have emerged as a powerful approach for capturing diverse behavioral strategies by learning action sequences from expert demonstrations. However, existing methods implicitly select among strategies through the denoising process, without uncertainty quantification over which strategy to execute. We formulate expert selection for diffusion policies as an offline contextual bandit problem and propose a Bayesian framework for pessimistic, uncertainty-aware strategy selection. A multi-head Variational Bayesian Last Layer (VBLL) model predicts the expected tracking performance of each expert strategy given the current belief state, providing both a point estimate and predictive uncertainty. Following the pessimism principle for offline decision-making, a Lower Confidence Bound (LCB) criterion then selects the expert whose worst-cas

Stats

• citations0
• S2 citations—
• github stars0
• HF likes0
• heat score0.00
• arxiv keyxiang2026diffusion

Related papers

• Policy-guided Diffusion (2024)0.00
• Sparse Diffusion Policy: A Sparse, Reusable, And Flexible Policy For Robot Learning (2024)0.00
• Lrt-diffusion: Calibrated Risk-aware Guidance For Diffusion Policies (2025)0.00
• Fine-tuning Diffusion Policies With Backpropagation Through Diffusion Timesteps (2025)0.00
• Diffusion Policy Through Conditional Proximal Policy Optimization (2026)0.00
• Steering Your Diffusion Policy With Latent Space Reinforcement Learning (2025)0.00
• Bayesian Residual Policy Optimization: Scalable Bayesian Reinforcement Learning With Clairvoyant Experts (2020)0.00
• Padiff: Predictive And Adaptive Diffusion Policies For Ad Hoc Teamwork (2025)0.00