Machine Learning PapersLandmark papers
← All papers
RESEARCH PAPER / 2026-10-17

Near-Optimal Regret for Semi-Bandit Learning of Monotone Stochastic Problems

Zhengjia Zhuo, Viswanath Nagarajan

University of Michigan · arXiv

Affiliations: University of Michigan

Abstract

We consider the task of learning an (approximately) optimal policy for stochastic optimization problems where the true distributions are unknown. In each period, the algorithm plays a policy for the stochastic problem and observes semi-bandit feedback, i.e., it only receives new samples from the random variables that were selected by the policy. We obtain an algorithm that works for a large class of ``monotone'' stochastic problems, including prophet inequality, Pandora's box, series testing, stochastic knapsack and stochastic matching. We obtain $\tilde{O}(\sqrt{nkT})$ regret for ``weakly'' monotone problems and $\tilde{O}(\sqrt{nT})$ regret for ``strongly'' monotone problems, where $T$ is the number of periods, $n$ is the problem dimension (number of random variables) and $k$ is the maximum support size of any random variable. This improves over previous regret bounds by a $\sqrt{n}$ factor. In fact, our dependence on $n$ and $T$ is the best-possible due the $Ω(\sqrt{nT})$ lower bound for multi-armed-bandits. Our main technique is the use of Hellinger distance in bounding the ``error'' between the true and learned distributions, and the use of a constant-factor chain rule for the Hellinger distance of dependent random variables.

The following analysis is generated by AI and may contain errors. It is separate from the authors’ abstract and is not peer review.

Research summary

The paper establishes near-optimal regret bounds for semi-bandit learning of monotone stochastic problems by leveraging Hellinger distance and adaptive chain rules. This theoretical advance improves upon prior bounds by a $\sqrt{n}$ factor and provides a unified framework for analyzing a wide class of stochastic optimization tasks, though it lacks empirical validation.

Methodology

The paper proposes a semi-bandit learning algorithm for a broad class of "monotone" stochastic optimization problems (e.g., stochastic knapsack, matching, prophet inequality). The core technical contribution is the application of the Hellinger distance to bound the stability of policies under distributional shifts. By utilizing Jayram's adaptive chain rule for Hellinger distance, the authors derive a stability lemma that depends on the probability of probing each item, rather than a uniform bound. This allows for a tighter regret analysis that leverages the semi-bandit feedback structure (where only probed items are observed). The method generalizes previous work that relied on Total Variation distance, which yielded suboptimal $\sqrt{nT}$ dependencies. The extension to general distributions using Radon-Nikodym derivatives and measure-theoretic tools is rigorous and well-structured.

Results & evaluation

The paper is purely theoretical. It provides no empirical experiments, benchmarks, or simulations to validate the theoretical bounds on real-world or synthetic datasets. The "results" are entirely mathematical regret bounds. While this is standard for this specific subfield of theoretical computer science/ML, it limits the immediate practical impact and verifiability of the improvements.

Reproducibility

The theoretical results are fully reproducible in the sense that the proofs are provided in detail. However, there is no code, algorithm implementation, or experimental setup to reproduce. The algorithm described is a meta-algorithm relying on an offline approximation oracle, which is not specified in a way that allows for direct implementation without choosing a specific problem instance.

Limitations

The primary limitation is the lack of empirical validation. The theoretical improvements (removing a $\sqrt{n}$ factor) are significant in asymptotic analysis but may not translate to practical performance gains in finite-time regimes typical of real-world applications. Additionally, the results assume access to an offline approximation algorithm for the specific stochastic problem, which may not exist or be efficient for all monotone problems. The assumption of independent random variables, while standard, restricts applicability to problems with correlated uncertainties.

Broader impact

This work contributes to the foundational theory of online learning in combinatorial settings. It provides a general framework for analyzing semi-bandit feedback in stochastic optimization, which can be applied to a wide range of problems in operations research and economics. The technique of using Hellinger distance with adaptive chain rules may inspire further theoretical work in bandit learning and statistical estimation. However, the impact is primarily academic and theoretical rather than industrial. The paper establishes near-optimal regret bounds for semi-bandit learning of monotone stochastic problems by leveraging Hellinger distance and adaptive chain rules. This theoretical advance improves upon prior bounds by a $\sqrt{n}$ factor and provides a unified framework for analyzing a wide class of stochastic optimization tasks, though it lacks empirical validation.

Author of this paper? Send a correction.