Pith. sign in

REVIEW 25 references

Neural Contextual Bandits Under Delayed Feedback Constraints

T0 review · reviewed 2026-08-16 · deepseek-v4-flash

Pith's one-line read Under sub-exponential reward delays, Delayed NeuralUCB achieves regret O(d~√T logT + d~^{3/2}D+ log^{3/2}T), with D+ depending on the expected delay.

arxiv 2504.12086 v1 pith:U5TKSCZE submitted 2025-04-16 cs.LG

classification cs.LG
keywords delayedrewardalgorithmfeedbackbanditsboundcalledcontextual
verification ladder T0 review T1 audit T2 compute T3 formal

The pith

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The reading

In a contextual bandit, a system sees a situation, picks one of several actions, and receives a reward. Most learning algorithms assume the reward arrives immediately. In real settings, the reward for a recommendation or a treatment may only arrive days later. This paper adapts a well-known algorithm, NeuralUCB, to that delayed setting. NeuralUCB uses a neural network to predict rewards and adds an uncertainty bonus to encourage trying actions it is unsure about. The new Delayed NeuralUCB only updates its network and its uncertainty matrix when a delayed reward actually arrives.

The authors prove a regret bound: the total loss compared with always picking the best action grows like the square root of the number of rounds times an effective dimension, plus a delay term that depends on the expected delay but does not grow with the number of rounds. This means that, in theory, even with large delays the algorithm does not fall disastrously behind. They also propose Delayed NeuralTS, a Thompson sampling version. In experiments on MNIST (digit classification as a 10-arm bandit) and Mushroom (2-arm), the algorithms run under uniform, exponential, and Pareto delays.

The experiments are not conclusive. The baselines are algorithms that receive feedback immediately, which is an oracle advantage, and the plots show only means over five runs with no error bars. The theoretical result is the main content, and it depends strongly on the delay distribution having an exponential tail.

Extended reading notes

Core claim

Theorem 1: with probability at least 1-delta, Delayed NeuralUCB has cumulative regret bounded by Eq. (7), which the paper summarizes as O(d~ sqrt(T) log T + d~^{3/2} D+ log(T)^{3/2}), where d~ is the NTK effective dimension and D+ = 1 + 2E[tau] + D_tau + psi_tau depends on the mean delay but not linearly on T.

Load-bearing premise

The delay variables are iid (alpha,b)-sub-exponential (Assumption 2), which is what makes the missing-reward concentration step (Lemmas 3-5 of [17], used to prove Eq. (14)) hold. If delays are heavy-tailed, dependent, or controlled by the environment, the D+ term and the regret bound fail; the paper's own Pareto-delay experiments lie outside this assumption and have no theoretical guarantee.

Share X Bluesky LinkedIn Reddit HN

Signed reviews

No signed human review yet.

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, and a circularity audit.

Assumptions & free parameters 4 free parameters · 8 assumptions · 0 invented entities

The central regret bound rests on the reward-noise, delay, and NTK assumptions, plus a chain of cited technical lemmas from [9], [17], and [23]. No new physical or mathematical entity is introduced and no constants are fit to data; the only hand-chosen numbers are algorithm hyperparameters.

free parameters (4)
  • Exploration parameter nu = 1 (experiments)
    Scales the UCB/TS exploration bonus and appears linearly in the regret bound. Chosen by hand, not estimated from data.
  • Regularization parameter lambda = 1 (experiments)
    Regularizes the design matrix; theory requires lambda >= max(1,S^{-2},L^2). Chosen by hand.
  • Norm bound S = 0.0001 (experiments)
    Appears as sqrt(lambda) S in the regret bound. Theory requires S >= sqrt(2 h^T H^{-1} h), which is not verified and likely violated by this tiny value.
  • Network width m = 128 (experiments)
    Theory requires m >= poly(T,K,L,lambda^{-1},lambda_0^{-1},S^{-1},log(1/delta)), while experiments use m=128, so the experiments operate outside the theorem's regime.
assumptions (8)
  • domain assumption Reward noise xi_t satisfies E[exp(gamma xi_t)|F_{t-1}] <= exp(gamma^2 R^2/2) for all gamma (sub-Gaussian).
    Assumption 1, Section II-C. Required for concentration of the estimated reward model.
  • domain assumption Delays tau_t are non-negative, iid (alpha,b)-sub-exponential random variables with MGF bound E[exp(gamma(tau_t-E[tau]))] <= exp(alpha^2 gamma^2/2) for |gamma| <= 1/b.
    Assumption 2, Section II-C. Drives D+ and the missing-reward concentration; Pareto delays in experiments do not satisfy it.
  • domain assumption The NTK matrix H satisfies H >= lambda_0 I and each context has ||x_i||=1 with [x_i]_j=[x_i]_{j+d/2}.
    Assumption 3, Section III-A, taken from [10] to ensure f(x;theta_0)=0 and positive-definite kernel.
  • standard math For sufficiently wide networks, the true reward function has the linear representation h(x_i)=g(x_i;theta_0)^T(theta*-theta_0) with sqrt(m)||theta*-theta_0||_2 <= sqrt(2 h^T H^{-1} h).
    Lemma 1 quoted from [9], Section IV.
  • standard math Gradient descent training in Algorithm 2 keeps parameters near initialization: ||theta_t-theta_0||_2 <= 2 sqrt(|I_t|/(m lambda)) and ||theta*-theta_t||_{Z_t} <= gamma_t sqrt(m).
    Lemma 2, proof relies on Lemma 5.2 of [9].
  • standard math Matrix decomposition Z_t^{-1}=V_t^{-1}+V_t^{-1} W_t Z_t^{-1} where V_t=Z_t+W_t.
    Lemma 4 proof, cited as Lemma 2 of [17].
  • standard math For iid sub-exponential delays, the missing-reward matrix/term concentration yields Eq. (14) with constant D+ from Eq. (8).
    Lemma 4 proof uses Lemmas 3-5 of [17] without reproducing the proof.
  • standard math Elliptical potential lemma: sum_t ||x_t||_{V_{t-1}^{-1}}^2 <= 2 log(det(V_T)/det(lambda I)).
    Lemma 11 of [23], used in Eq. (15).

how reviews work

0 comments
Cite this review

Pith. "Pith review of Neural Contextual Bandits Under Delayed Feedback Constraints." pith.science (2026). https://pith.science/paper/U5TKSCZE

@misc{pith2026250412086,
  author       = {Pith},
  title        = {Pith review of: Neural Contextual Bandits Under Delayed Feedback Constraints},
  year         = {2026},
  howpublished = {\url{https://pith.science/paper/U5TKSCZE}},
  note         = {Machine review of arXiv:2504.12086}
}
read the original abstract

This paper presents a new algorithm for neural contextual bandits (CBs) that addresses the challenge of delayed reward feedback, where the reward for a chosen action is revealed after a random, unknown delay. This scenario is common in applications such as online recommendation systems and clinical trials, where reward feedback is delayed because the outcomes or results of a user's actions (such as recommendations or treatment responses) take time to manifest and be measured. The proposed algorithm, called Delayed NeuralUCB, uses an upper confidence bound (UCB)-based exploration strategy. Under the assumption of independent and identically distributed sub-exponential reward delays, we derive an upper bound on the cumulative regret over a T-length horizon. We further consider a variant of the algorithm, called Delayed NeuralTS, that uses Thompson Sampling-based exploration. Numerical experiments on real-world datasets, such as MNIST and Mushroom, along with comparisons to benchmark approaches, demonstrate that the proposed algorithms effectively manage varying delays and are well-suited for complex real-world scenarios.

Figures

Figures reproduced from arXiv: 2504.12086 by the authors.

Figure 1
Figure 1. Comparison of the cumulative regret of the algorithms with no delayed feedback – LinTS/UCB, Neural-TS/UCB – with our proposed algorithms Delayed Neural-UCB/TS under uniform delay with E[τ ] = 30, as a function of the number of iterations on (left) MNIST and (right) Mushroom datasets. We run 5 experiments and plot the mean regret [PITH_FULL_IMAGE:figures/full_fig_p007_1.png] view at source ↗
Figure 2
Figure 2. (Left) Comparison of the cumulative regret of the Delayed NeuralTS under uniform, exponential and Pareto delays with E[τ ] = 30 as a function of the number of iterations on MNIST. (Right) Comparison of the cumulative regret of the Delayed NeuralUCB under the three delays on Mushroom. C. Experiment and Results For all our algorithms, we choose a two-layer neural network f(x; θ) = √ mW2σ(W1x) with network width m = 12… view at source ↗

Discussion (0). Continue with ORCID to comment.

Reference graph

Works this paper leans on

25 extracted references · 19 canonical work pages

  1. [1]

    Contextual bandits for adapting treatment in a mouse model of de novo carcinogenesis,

    A. Durand, C. Achilleos, D. Iacovides, K. Strati, G. D. Mitsis, and J. Pineau, “Contextual bandits for adapting treatment in a mouse model of de novo carcinogenesis,” in Machine learning for healthcare conference, pp. 67–82, PMLR, 2018

  2. [2]

    Portfolio choices with orthogonal bandit learning,

    W. Shen, J. Wang, Y .-G. Jiang, and H. Zha, “Portfolio choices with orthogonal bandit learning,” in Twenty-fourth international joint conference on artificial intelligence , 2015

  3. [3]

    Explore, exploit, and explain: personalizing explainable recommendations with bandits,

    J. McInerney, B. Lacker, S. Hansen, K. Higley, H. Bouchard, A. Gru- son, and R. Mehrotra, “Explore, exploit, and explain: personalizing explainable recommendations with bandits,” inProceedings of the 12th ACM conference on recommender systems , pp. 31–39, 2018

  4. [4]

    Distributed stochastic contextual bandits for protein drug interaction,

    J. Lin, K. A. Sajeevan, B. Acharya, S. Moothedath, and R. Chowdhury, “Distributed stochastic contextual bandits for protein drug interaction,” in ICASSP 2024-2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), pp. 7160–7164, IEEE, 2024

  5. [5]

    Stochastic conservative contextual linear bandits,

    J. Lin, X. Y . Lee, T. Jubery, S. Moothedath, S. Sarkar, and B. Ganap- athysubramanian, “Stochastic conservative contextual linear bandits,” IEEE Conference on Decision and Control , 2022

  6. [6]

    Contextual bandits with linear payoff functions,

    W. Chu, L. Li, L. Reyzin, and R. Schapire, “Contextual bandits with linear payoff functions,” in Proceedings of the fourteenth international conference on artificial intelligence and statistics, pp. 208–214, JMLR Workshop and Conference Proceedings, 2011

  7. [7]

    Thompson sampling for contextual bandits with linear payoffs,

    S. Agrawal and N. Goyal, “Thompson sampling for contextual bandits with linear payoffs,” in International conference on machine learning, pp. 127–135, PMLR, 2013

  8. [8]

    Parametric bandits: The generalized linear case,

    S. Filippi, O. Cappe, A. Garivier, and C. Szepesv ´ari, “Parametric bandits: The generalized linear case,” Advances in neural information processing systems, vol. 23, 2010

Show all 25 references
  1. [9]

    Neural contextual bandits with ucb- based exploration,

    D. Zhou, L. Li, and Q. Gu, “Neural contextual bandits with ucb- based exploration,” in International Conference on Machine Learning, pp. 11492–11502, PMLR, 2020

  2. [10]

    Neural thompson sampling,

    W. Zhang, D. Zhou, L. Li, and Q. Gu, “Neural thompson sampling,” arXiv preprint arXiv:2010.00827 , 2020

  3. [11]

    Online learning under delayed feedback,

    P. Joulani, A. Gyorgy, and C. Szepesv ´ari, “Online learning under delayed feedback,” in International conference on machine learning , pp. 1453–1461, PMLR, 2013

  4. [12]

    Ban- dits with delayed, aggregated anonymous feedback,

    C. Pike-Burke, S. Agrawal, C. Szepesvari, and S. Grunewalder, “Ban- dits with delayed, aggregated anonymous feedback,” in International Conference on Machine Learning , pp. 4105–4113, PMLR, 2018

  5. [13]

    Stochastic bandit models for delayed conversions,

    C. Vernade, O. Capp ´e, and V . Perchet, “Stochastic bandit models for delayed conversions,” arXiv preprint arXiv:1706.09186 , 2017

  6. [14]

    The queue method: Handling delay, heuristics, prior data, and evaluation in ban- dits,

    T. Mandel, Y .-E. Liu, E. Brunskill, and Z. Popovi ´c, “The queue method: Handling delay, heuristics, prior data, and evaluation in ban- dits,” in Proceedings of the AAAI Conference on Artificial Intelligence, vol. 29, 2015

  7. [15]

    Learning in generalized linear contextual bandits with stochastic delays,

    Z. Zhou, R. Xu, and J. Blanchet, “Learning in generalized linear contextual bandits with stochastic delays,” Advances in Neural In- formation Processing Systems , vol. 32, 2019

  8. [16]

    Delay-adaptive learning in general- ized linear contextual bandits,

    J. Blanchet, R. Xu, and Z. Zhou, “Delay-adaptive learning in general- ized linear contextual bandits,” Mathematics of Operations Research , vol. 49, no. 1, pp. 326–345, 2024

  9. [17]

    Delayed feedback in generalised linear bandits revisited,

    B. Howson, C. Pike-Burke, and S. Filippi, “Delayed feedback in generalised linear bandits revisited,” in International Conference on Artificial Intelligence and Statistics , pp. 6095–6119, PMLR, 2023

  10. [18]

    Neural contextual bandits without regret,

    P. Kassraie and A. Krause, “Neural contextual bandits without regret,” in International Conference on Artificial Intelligence and Statistics , pp. 240–278, PMLR, 2022

  11. [19]

    Neural contextual bandits with deep representation and shallow exploration,

    P. Xu, Z. Wen, H. Zhao, and Q. Gu, “Neural contextual bandits with deep representation and shallow exploration,” arXiv preprint arXiv:2012.01780, 2020

  12. [20]

    Scalable neural contextual bandit for recommender systems,

    Z. Zhu and B. Van Roy, “Scalable neural contextual bandit for recommender systems,” in Proceedings of the 32nd ACM International Conference on Information and Knowledge Management , pp. 3636– 3646, 2023

  13. [21]

    Provably and practically efficient neural contextual ban- dits,

    S. Salgia, “Provably and practically efficient neural contextual ban- dits,” in International Conference on Machine Learning , pp. 29800– 29844, PMLR, 2023

  14. [22]

    Neural tangent kernel: Con- vergence and generalization in neural networks,

    A. Jacot, F. Gabriel, and C. Hongler, “Neural tangent kernel: Con- vergence and generalization in neural networks,” Advances in neural information processing systems , vol. 31, 2018

  15. [23]

    Improved algorithms for linear stochastic bandits,

    Y . Abbasi-Yadkori, D. P ´al, and C. Szepesv ´ari, “Improved algorithms for linear stochastic bandits,” Advances in neural information process- ing systems, vol. 24, 2011

  16. [24]

    Gradient-based learning applied to document recognition,

    Y . LeCun, L. Bottou, Y . Bengio, and P. Haffner, “Gradient-based learning applied to document recognition,” Proceedings of the IEEE , vol. 86, no. 11, pp. 2278–2324, 1998

  17. [25]

    Mushroom

    “Mushroom.” UCI Machine Learning Repository, 1981. DOI: https://doi.org/10.24432/C5959T

Pith tools

Reviewed August 16, 2026 · model on record in the stance chip above.