BOCO: Bayesian Online Contextual Optimization for Decision-Focused Online Learning
math.OC, stat.ML
Submitted: 2025-11-25
Updated: 2026-09-23
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Decision-focused learning (DFL) trains predictive models to optimize downstream decisions rather than prediction accuracy alone.
Terminology
Abstract
Decision-focused learning (DFL) trains predictive models to optimize downstream decisions rather than prediction accuracy alone. While recent studies have extended this paradigm to online settings with streaming data, existing online DFL methods generally maintain a point estimate, while their gradient-based updates require either a differentiable optimization layer or a problem-specific surrogate loss. Consequently, they can be unstable under limited data and difficult to apply across heterogeneous optimization problems. We introduce Bayesian Online Contextual Optimization (BOCO), a framework that maintains a decision-focused posterior over model parameters. BOCO aggregates the resulting predictions when prescribing decisions, thereby accounting for parameter uncertainty. To track this posterior in evolving environments, we develop two particle-based inference algorithms: a sequential Monte Carlo sampler for nondifferentiable problems and a function-space Stein variational gradient descent algorithm for differentiable problems. Across both real-world tasks, BOCO reduces running mean regret and temporal regret variability relative to two frequentist online DFL baselines. At the full horizon, relative to the best frequentist baseline, BOCO reduces running mean regret and temporal regret variability by 6.9% and 6.3% on knapsack and by 49.3% and 38.1% on energy scheduling, respectively. The gains are larger early in the data stream, a pattern consistent with a benefit from accounting for parameter uncertainty when data are limited.
Sources
- Online Decision-Focused Learning
- Online Contextual Decision-Making with a Smart Predict-then-Optimize Method
Related papers
- Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noise
- Adam-HNAG: A Convergent Reformulation of Adam with Accelerated Rate
- Incremental Learning in Mirror Flows
- Second-Order Optimization with Asynchronous Curvature
- Online Control via Counterfactual Tracking
- Asynchronous Replanning in Two Population Linear Quadratic Mean Field Games: Information Requirements and Stability