stat.MLSep 1, 2026

Pooling and Drift in Delayed Bandits

Authors: Melika Baghi

Abstract

A system often has to act long before it learns whether the act worked: a recommender sees a click in seconds and a purchase in days. With KK actions and a delay of dd rounds, the best rate known for this setting is O~((K+d)T)\widetilde{O}(\sqrt{(K+d)T}) over TT rounds, so a longer menu is always more expensive to learn from. It need not be: if the outcome depends on the action only through the state it produced, then one late outcome informs every action that could have produced the observed state, and the price is set by how many genuinely different states the actions produce rather than by how many actions there are. We measure this using an effective dimension vtv_t between 11 and the number of states, and prove O~((d+1)VlogK)\widetilde{O}(\sqrt{(d+1)V\log K}) for a rotating algorithm and O~(V+dT)\widetilde{O}(\sqrt{V^{-}}+\sqrt{dT}) for the single-copy algorithm used in practice, for any budget fixed in advance; merging similar states lowers the price further, at an explicit bias. Even when given the exact losses from dd rounds ago, no algorithm escapes Ω(dEmin{1+logJ,T/d})Ω(\sqrt{dE\min\{1+\log J,T/d\}}), where JJ counts the drifting directions and EE bounds how far losses move while the learner waits. On generated data, the state channel cuts regret by up to 79 percent against action-level weighting and, on the funnel family, by 32 to 68 percent against a tuned minimax-optimal method.

Explore similar work

Jan 14, 2025cs.LG

Impatient Bandits: Optimizing for the Long-Term Without Delay

Increasingly, recommender systems are tasked with improving users' long-term satisfaction. In this context, we study a content exploration task, which we formalize as a bandit problem with delayed rewards. There is an apparent trade-off in choosing the learning signal: waiting for the full reward to become available might take several weeks, slowing the rate of learning, whereas using short-term proxy rewards reflects the actual long-term goal only imperfectly. First, we develop a predictive model of delayed rewards that incorporates all information obtained to date. Rewards as well as shorter-term surrogate outcomes are combined through a Bayesian filter to obtain a probabilistic belief. Second, we devise a bandit algorithm that quickly learns to identify content aligned with long-term success using this new predictive model. We prove a regret bound for our algorithm that depends on the Value of Progressive Feedback, an information-theoretic metric that captures the quality of short-term leading indicators that are observed prior to the long-term reward. We apply our approach to a podcast recommendation problem, where we seek to recommend shows that users engage with repeatedly over two months. We empirically validate that our approach significantly outperforms methods that optimize for short-term proxies or rely solely on delayed rewards, as demonstrated by an A/B test in a recommendation system that serves hundreds of millions of users.
Kelly W. Zhang, Thomas Baldwin-McDonald, Kamil Ciosek +2
Jun 15, 2026cs.LG

Near-Optimal Stochastic Linear Bandits with Delay

We study stochastic linear bandits with delayed feedback under several delay models and establish near-optimal regret guarantees. Our results identify when delayed linear bandits exhibit the same qualitative behavior as multi-armed bandits (MAB), and when the linear structure creates fundamentally new challenges. Specifically, (1) for \emph{loss-independent delays}, where the delay does not depend on the realized loss (but potentially depends on the arm), we show that delays incur only an additive regret penalty. Under stochastic delays, this penalty scales with the expected delay, while under adversarial delays, it scales with the maximum number of outstanding observations. Notably, both delay penalties are dimension-free, improving upon the state-of-the-art results; (2) for \emph{loss-dependent delays}, we show that linear bandits are substantially harder than MAB: unlike in MAB, we prove matching (up to log factors) upper and lower bounds in linear bandits, whose delay penalty depends on the square root of the dimension. (3) for the \emph{delay-as-payoff model}, a special case of loss-dependent delay, we show that the optimal MAB guarantee, which depends only on the delay of the optimal arm, is also unattainable in linear bandits. Together, these results provide a sharp characterization of how delayed feedback interacts with linear generalization.
Ofir Schlisselberg, Mengxiao Zhang, Yishay Mansour
Sep 14, 2026cs.LG

Nearly Minimax-Optimal Regret for Linear Contextual Bandits with Arbitrary Adaptive Action Sets

We study stochastic linear contextual bandits with arbitrary action menus that may depend on the fixed parameter and the interaction history. We establish matching upper and lower bounds, up to logarithmic factors. Let dd be the dimension, KK be the menu size, and TT the time horizon. For 2Kd2\le K\le d, we prove an upper bound O~(K1/4dT)\widetilde O(K^{1/4}\sqrt{dT}). When Td2T\ge d^2, we further prove a lower bound Ω(K1/4dT)Ω(K^{1/4}\sqrt{dT}). Thus, for Td2T\ge d^2 and 2Kd2\le K\le d, the upper and lower bounds match up to logarithmic factors, and the polynomial dependence on KK is optimal. Compared with the previous O~(dKT)\widetilde O(\sqrt{dKT}) bound, our upper bound improves the dependence on KK by a factor of K1/4K^{1/4}. For KdK\ge d, we prove an upper bound O~d,T(dTmin{d,(dlogK)1/4})\widetilde O_{d,T}\left(\sqrt{dT}\min\{\sqrt d,(d\log K)^{1/4}\}\right) and a lower bound Ω(dTmin{d,(dlogKlog(2d))1/4})Ω\left(\sqrt{dT}\min\left\{\sqrt d,\left(\frac{d\log K}{\log(2d)}\right)^{1/4}\right\}\right). Here, O~d,T\widetilde O_{d,T} omits logarithmic factors only in dd and TT. In particular, for polynomially large KdK\ge d, the upper and lower bounds both scale as d3/4Td^{3/4}\sqrt T up to logarithmic factors, improving the standard O~(dT)\widetilde O(d\sqrt T) rate by a factor of d1/4d^{1/4}. As KK grows further, the regret smoothly recovers the dTd\sqrt T scale once logK\log K reaches order dd.
Tianyuan Jin