REVIEW 2 major objections 4 minor 1 cited by
Near-optimal Regret Using Policy Optimization in Online MDPs with Aggregate Bandit Feedback
T0 review · 2 major / 4 minor · reviewed 2026-08-08 · deepseek-v4-flash
Pith's one-line read In online MDPs with aggregate bandit feedback and known dynamics, policy optimization achieves the first optimal regret bound $\tilde\Theta(H^2\sqrt{SAK})$.
desk verdict The U-function is a genuine new tool and the known-dynamics bound is near-optimal in the SA >= H^2 regime, but the abstract's unqualified Theta(H^2 sqrt(SAK)) overstates what Theorems 2 and 4 actually deliver. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the U-function $U^\pi_h(s,a;\ell)$, defined as the expected total cost of the episode conditional on reaching $(s,a)$ at step $h$ under policy $\pi$; the paper shows that replacing the Q-function by U in the value-difference lemma costs nothing because the discrepancy is an action-independent offset (Lemma 2). The corresponding estimator $\hat U^k_h(s,a)=I\{s^k_h=s,a^k_h=a\}L^k_{1:H}/(\mu^k_h(s,a)+\gamma)$ uses only the aggregated trajectory loss, which is all the feedback available. Around this estimator the algorithm builds a multiplicative-weights update with a carefully chosen bonus $B^k$, computed as a Bellman value with local loss $b^k_h(s)=\sum_a 3\gamma H\pi^k_h(a|s)/(\mu^k_h(s)\pi^k_h(a|s)+\gamma)$; the bonus cancels the distribution-mismatch terms in the regret decomposition. In the unknown-dynamics version the same machinery is retained, with optimistic upper and lower occupancy estimates $\overline\mu^k_h(s,a)$ and $\underline\mu^k_h(s,a)$ obtained from Bernstein confidence sets.
What would settle it
Implement the lower-bound MDP from Appendix D with small $S,A$ and large $H$, and measure Algorithm 1's regret against both $H^2\sqrt{SAK}$ and $H^3\sqrt K$: the theorem's $H^3\sqrt K$ term should be visible whenever $SA<H^2$. More decisively, test whether the imported multi-task bandit lower bound used in Lemma 4 holds when each state receives a random number $T_i\sim\mathrm{Bin}(K,1/S)$ of rounds and the adversary picks per-state instances independently; if it fails for small $T_i$, the $\Omega(H^2\sqrt{SAK})$ lower bound no longer follows from the reduction.
Extended reading notes
Core claim
The central claim is that aggregate bandit feedback does not prevent optimal-order regret in online MDPs, provided the regret decomposition is built on the U-function rather than the Q-function. For a fixed policy $\pi$, $U^\pi_h(s,a;\ell)$ is the expected total loss of the episode given that $\pi$ chooses action $a$ in state $s$ at time $h$; Lemma 2 shows $U^\pi_h(s,a;\ell)-Q^\pi_h(s,a;\ell)$ is a state-only term independent of $a$. This makes the value-difference lemma valid with U in place of Q and yields an estimator $\hat U^k_h(s,a)=I\{s^k_h=s,a^k_h=a\}L^k_{1:H}/(\mu^k_h(s,a)+\gamma)$ that uses only the observed aggregate loss $L^k_{1:H}$. The analysis follows the policy-optimization template: a multiplicative-weights update on $\hat U^k$ with a variance-reducing bonus $B^k$, a regret decomposition whose bias and bonus terms cancel, and tuned $\eta,\gamma$ to balance the remaining terms. The result is a high-probability regret bound of $\tilde O(H^2\sqrt{SAK}+H^3\sqrt K)$ under known dynamics, and the paper's lower bound $\Omega(H^2\sqrt{SAK})$ is built by embedding a hard multi-task bandit into each state of an MDP.
Load-bearing premise
The claim of optimality rests on an imported multi-task bandit lower bound that the paper cites without proof or adaptation, together with the condition $SA\ge H^2$ that makes the $H^3\sqrt K$ term negligible.
Editorial extensions
If this is right
- With known dynamics, aggregate-bandit online MDPs admit $\tilde\Theta(H^2\sqrt{SAK})$ regret, so no future algorithm in this feedback model can improve the order when $SA\ge H^2$.
- With unknown dynamics, the $\tilde O(H^3 S\sqrt{AK})$ bound matches the best policy-optimization guarantee for semi-bandit feedback, showing the information gap between semi-bandit and aggregate feedback is not as large as previously thought.
- The closed-form multiplicative-weights update avoids the per-iteration convex optimization of prior work, so the improvement comes with a computational gain rather than a trade-off.
- The lower-bound construction transfers multi-task bandit hardness to MDPs by splitting visits across states, so any improved unknown-dynamics guarantee must close the remaining $H\sqrt S$ gap rather than change the known-dynamics exponent.
Reading between the lines
- Beyond the paper's claims, the U-function estimator could be adapted to preference-based RL, where feedback is a binary outcome over a whole trajectory rather than a scalar loss; the same importance-weighted ratio would provide an unbiased trajectory-level gradient surrogate.
- The $H^3\sqrt K$ term in Theorem 2 means the optimality claim is regime-dependent; an experiment sweeping $S,A,H$ separately could reveal whether this term is an artifact of the bonus construction and whether a tighter analysis can remove it.
- Because the lower bound only uses states that are independent after the first step, a more informative probe of the unknown-dynamics gap $H\sqrt S$ would be to test whether intermediate state observations, but not intermediate losses, help; if they do, the gap reflects an information difference rather than a transition-estimation difficulty.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies episodic finite-horizon online MDPs with adversarially chosen losses and aggregate bandit feedback, where the learner observes only the total episode loss rather than per-step losses. It introduces a new `U-function` decomposition (Lemma 2 and Corollary 1) that replaces the Q-function in the value-difference lemma with the expected total episode loss conditional on visiting a state-action pair, and it gives the first policy-optimization algorithms for this setting. For known dynamics, Algorithm 1 combines a multiplicative-weights update with a nearly unbiased estimator of the U-function and a state-dependent bonus; Theorem 2 claims regret \tilde O(H^2\sqrt{SAK}+H^3\sqrt K). For unknown dynamics, Algorithm 2 uses optimistic occupancy estimates and Bernstein-style confidence sets, giving Theorem 3 with regret \tilde O(H^3 S\sqrt{AK}+H^4 S^3 A). Theorem 4 states a lower bound \Omega(H^2\sqrt{SAK}) through a reduction to a multi-task bandit lower bound of Cohen et al. (2017). The abstract and contribution list describe the known-dynamics result as the first optimal bound \tilde\Theta(H^2\sqrt{SAK}).
Significance. The U-function decomposition is a natural and potentially reusable idea for aggregate-bandit feedback: it makes the full episode loss a valid unbiased signal for the quantity appearing in the regret decomposition, avoiding the need to estimate Q-functions from unavailable per-step losses. The proof structure in Appendix B is detailed and internally consistent; the key cancellation between the bonus term and the distribution-mismatch terms in Bias1 and Reg appears correct. If the optimality claim is properly qualified, the known-dynamics result is a genuine improvement over both inefficient reduction-to-linear-bandits bounds (\tilde O(H^2 S\sqrt{AK}) and \tilde O(H^2 SA\sqrt K)) and gives a matching lower bound in the SA \ge H^2 regime. The unknown-dynamics bound \tilde O(H^3 S\sqrt{AK}+H^4 S^3 A) improves on Cohen et al. (2021b) by a large factor and, surprisingly, matches the best known policy-optimization bound for the more informative semi-bandit setting. The paper also ships high-probability bounds with explicit logarithmic factors and a clear appendix, which strengthens the value of the contribution.
major comments (2)
- [Abstract and Section 4 (Theorem 2)] The abstract and the contribution list state an unqualified optimal bound \tilde\Theta(H^2\sqrt{SAK}), but Theorem 2 only establishes \tilde O(H^2\sqrt{SAK}+H^3\sqrt K). The H^3\sqrt K term is not a proof artifact: with \eta=(H\sqrt{SAK}+H^2\sqrt K)^{-1}, the OMD term H\ln A/\eta contributes H^2\sqrt{SAK}\ln A + H^3\sqrt K\ln A. When SA < H^2, the H^3\sqrt K term dominates, and the lower bound in Theorem 4 (\Omega(H^2\sqrt{SAK})) does not close the gap. The paper itself acknowledges the condition SA \ge H^2 in the remarks after Theorem 2, so the abstract and contribution list must be revised to either state the optimality claim only in the regime SA \ge H^2 or to remove the second term from the upper bound.
- [Section 6 and Appendix D (Lemma 4, Theorem 4)] The matching lower bound relies entirely on Lemma 4, imported as Theorem 1 of Cohen et al. (2017) and not proved in the paper. The reduction in Appendix D encodes a hard multi-task bandit problem in each state, but the number of rounds at a given state is a random variable T_i (roughly K/S in expectation), and the adversary's loss functions are fixed for all K episodes before the random state transitions are realized. The proof neither states the exact hypotheses of the cited theorem (e.g., fixed vs. random number of rounds, oblivious vs. adaptive adversary) nor verifies that the induced per-state subproblem satisfies them. This is load-bearing for the claimed \tilde\Theta(H^2\sqrt{SAK}) optimality, so the authors should either include a proof of Lemma 4 or provide a detailed verification/adaptation of its conditions in the MDP reduction.
minor comments (4)
- [Lemma 2 statement] The quantifier in Lemma 2 is written as "(h,s,a)\,\in\,[H]\times A\times S", which reorders the state and action sets; it should be [H]\times S\times A.
- [Section 2 notation] The definition of the indicator I^k_h(s,a) contains a typographical error: I^k_h(s,a)=I{s^k_h=s,\,a^k_h=s} should read a^k_h=a in the second component.
- [Table 1] The table entries abbreviate the regret bounds without explicitly saying that logarithmic and lower-order terms are suppressed; adding a footnote to that effect would improve readability for readers comparing the rows.
- [Appendix D] The notation in the lower-bound proof uses both "SAT_i" and "\sqrt{SA}" inconsistently; the derivation is correct once the missing parentheses are understood, but the display would benefit from a clearer statement of how the sum over states is evaluated.
Circularity Check
No circular step: the regret upper bound is derived from first principles, and the lower bound is an external multi-task bandit result; the abstract's unqualified Theta is a scoping overclaim, not a circularity.
full rationale
The paper's central derivation is self-contained rather than self-referential. The U-function is defined in Section 3 as the expected total loss conditional on a time-h state-action visit; Lemma 2 proves U-Q equals an action-independent prefix term using the Markov property; and Corollary 1 derives the U-based value-difference decomposition from the standard value-difference lemma. None of these steps assumes the target regret bound. The estimator in Eq. (3) is analyzed in Lemma 3, which computes its conditional expectation directly from the definition and the law of total expectation, and the subsequent concentration arguments (Lemmas 5-8) are standard martingale bounds. The bonus B_k is a deterministic proof device designed to cancel distribution-mismatch terms through the value-difference lemma; its constants are chosen in the analysis, not fitted to data, and the resulting bound is stated explicitly, including the H^3 sqrt(K) term. The abstract's Theta(H^2 sqrt(SAK)) claim is formally narrower than Theorem 2, because the H^3 sqrt(K) term dominates when SA < H^2; this is a correctness or scoping issue about what 'optimal' means, not circularity. The lower bound Theorem 4 is obtained by reducing an MDP with a uniformly random first transition to the multi-task bandit problem and invoking Lemma 4 from Cohen et al. (2017), an independent external theorem proved outside this paper; importing it is a normal use of an external benchmark, and the proof does not assume the target bound. The only self-citations (Even-Dar et al. 2009 for the value-difference lemma; Cohen et al. 2021a for a Freedman consequence) are standard technical lemmas and are not load-bearing: deleting or replacing them would not change any conclusion. Hence no prediction or lower bound reduces to its own input.
Assumptions & free parameters
free parameters (2)
- learning rate eta =
sqrt(iota)/(H sqrt(SAK)+H^2 sqrt(K))
- exploration parameter gamma =
2 eta H
assumptions (5)
- domain assumption Lemma 4: multi-task bandit lower bound (Cohen et al. 2017, Theorem 1)
- standard math Confidence-set transition bounds (Luo et al. 2021 Lemma 24; Jin et al. 2020 Lemmas 26-27)
- standard math Entropy-regularized OMD guarantee (Lemma 25, Hazan et al.)
- standard math Freedman-type concentration inequality (Lemma 22, Beygelzimer et al. 2011)
- domain assumption Oblivious adversary and Markovian policies
Cite this review
Pith. "Pith review of Near-optimal Regret Using Policy Optimization in Online MDPs with Aggregate Bandit Feedback." pith.science (2026). https://pith.science/paper/VSOARE5S
@misc{pith2026250204004,
author = {Pith},
title = {Pith review of: Near-optimal Regret Using Policy Optimization in Online MDPs with Aggregate Bandit Feedback},
year = {2026},
howpublished = {\url{https://pith.science/paper/VSOARE5S}},
note = {Machine review of arXiv:2502.04004}
}
abstract
We study online finite-horizon Markov Decision Processes with adversarially changing loss and aggregate bandit feedback (a.k.a full-bandit). Under this type of feedback, the agent observes only the total loss incurred over the entire trajectory, rather than the individual losses at each intermediate step within the trajectory. We introduce the first Policy Optimization algorithms for this setting. In the known-dynamics case, we achieve the first \textit{optimal} regret bound of $\tilde \Theta(H^2\sqrt{SAK})$, where $K$ is the number of episodes, $H$ is the episode horizon, $S$ is the number of states, and $A$ is the number of actions. In the unknown dynamics case we establish regret bound of $\tilde O(H^3 S \sqrt{AK})$, significantly improving the best known result by a factor of $H^2 S^5 A^2$.
Forward citations
Cited by 1 Pith paper
-
Outcome-Based Online Reinforcement Learning: Algorithms and Fundamental Limits
Outcome-based online RL is tractable under coverability with general function approximation, but there are MDPs where trajectory-level feedback costs exponentially more samples than per-step feedback.
Reference graph
Works this paper leans on
-
[1]
J. D. Abernethy, E. Hazan, and A. Rakhlin. Competing in the dark: An efficient algorithm for bandit linear optimization. In COLT, pages 263--274. Citeseer, 2008
work page 2008
-
[2]
A. Beygelzimer, J. Langford, L. Li, L. Reyzin, and R. Schapire. Contextual bandit algorithms with supervised learning guarantees. In Proceedings of the Fourteenth International Conference on Artificial Intelligence and Statistics. JMLR Workshop and Conference Proceedings, 2011
work page 2011
- [3]
-
[4]
Q. Cai, Z. Yang, C. Jin, and Z. Wang. Provably efficient exploration in policy optimization. In International Conference on Machine Learning, pages 1283--1294. PMLR, 2020
2020
- [5]
-
[6]
N. Cesa-Bianchi and G. Lugosi. Combinatorial bandits. Journal of Computer and System Sciences, 78 0 (5): 0 1404--1422, 2012
work page 2012
-
[7]
N. Chatterji, A. Pacchiano, P. Bartlett, and M. Jordan. On the theory of reinforcement learning with once-per-episode feedback. Advances in Neural Information Processing Systems, 34: 0 3401--3412, 2021
work page 2021
-
[8]
L. Chen, H. Luo, and A. Rosenberg. Policy optimization for stochastic shortest path. In P. Loh and M. Raginsky, editors, Conference on Learning Theory, 2-5 July 2022, London, UK , volume 178 of Proceedings of Machine Learning Research, pages 982--1046. PMLR , 2022 a
work page 2022
Show all 34 references
-
[9]
X. Chen, H. Zhong, Z. Yang, Z. Wang, and L. Wang. Human-in-the-loop: Provably efficient preference-based reinforcement learning with general function approximation. In International Conference on Machine Learning, pages 3773--3793. PMLR, 2022 b
2022
-
[10]
Cohen, T
A. Cohen, T. Hazan, and T. Koren. Tight bounds for bandit combinatorial optimization. In Conference on Learning Theory, pages 629--642. PMLR, 2017
2017
-
[11]
Cohen, Y
A. Cohen, Y. Efroni, Y. Mansour, and A. Rosenberg. Minimax regret for stochastic shortest path. Advances in Neural Information Processing Systems, 34, 2021 a
2021
-
[12]
Cohen, H
A. Cohen, H. Kaplan, T. Koren, and Y. Mansour. Online markov decision processes with aggregate bandit feedback. In Conference on Learning Theory, pages 1301--1329. PMLR, 2021 b
2021
-
[13]
Y. Dai, H. Luo, C.-Y. Wei, and J. Zimmert. Refined regret for adversarial mdps with linear function approximation. arXiv preprint arXiv:2301.12942, 2023
2023 arXiv
-
[14]
V. Dani, S. M. Kakade, and T. Hayes. The price of bandit information for online optimization. Advances in Neural Information Processing Systems, 20, 2007
2007
-
[15]
Efroni, N
Y. Efroni, N. Merlis, and S. Mannor. Reinforcement learning with trajectory feedback. In Proceedings of the AAAI conference on artificial intelligence, volume 35, pages 7288--7295, 2021
2021
-
[16]
Even-Dar, S
E. Even-Dar, S. M. Kakade, and Y. Mansour. Online markov decision processes. Mathematics of Operations Research, 34 0 (3): 0 726--736, 2009
2009
-
[17]
Hazan and Z
E. Hazan and Z. Karnin. Volumetric spanners: an efficient exploration basis for learning. Journal of Machine Learning Research, 2016
2016
-
[18]
Hazan et al
E. Hazan et al. Introduction to online convex optimization. Foundations and Trends in Optimization , 2 0 (3-4): 0 157--325, 2016
2016
-
[19]
Jaksch, R
T. Jaksch, R. Ortner, and P. Auer. Near-optimal regret bounds for reinforcement learning. Journal of Machine Learning Research, 11 0 (4), 2010
2010
-
[20]
C. Jin, Z. Allen-Zhu, S. Bubeck, and M. I. Jordan. Is q-learning provably efficient? In Advances in Neural Information Processing Systems, pages 4863--4873, 2018
2018
-
[21]
C. Jin, T. Jin, H. Luo, S. Sra, and T. Yu. Learning adversarial markov decision processes with bandit feedback and unknown transition. In International Conference on Machine Learning, pages 4860--4869. PMLR, 2020
2020
-
[22]
S. M. Kakade. A natural policy gradient. Advances in neural information processing systems, 14: 0 1531--1538, 2001
2001
-
[23]
Lattimore and C
T. Lattimore and C. Szepesv \'a ri. Bandit algorithms. Cambridge University Press, 2020
2020
-
[24]
Liu, C.-Y
H. Liu, C.-Y. Wei, and J. Zimmert. Towards optimal regret in adversarial linear mdps with bandit feedback. arXiv preprint arXiv:2310.11550, 2023
2023 arXiv
-
[25]
Luo, C.-Y
H. Luo, C.-Y. Wei, and C.-W. Lee. Policy optimization in adversarial mdps: Improved exploration via dilated bonuses. Advances in Neural Information Processing Systems, 34, 2021
2021
-
[26]
Rosenberg and Y
A. Rosenberg and Y. Mansour. Online stochastic shortest path with bandit feedback and unknown transition function. In Advances in Neural Information Processing Systems, pages 2209--2218, 2019 a
2019
-
[27]
Rosenberg and Y
A. Rosenberg and Y. Mansour. Online convex optimization in adversarial markov decision processes. In International Conference on Machine Learning, pages 5478--5486. PMLR, 2019 b
2019
-
[28]
A. Saha, A. Pacchiano, and J. Lee. Dueling rl: Reinforcement learning with trajectory preferences. In International Conference on Artificial Intelligence and Statistics, pages 6263--6289. PMLR, 2023
2023
-
[29]
Schulman, S
J. Schulman, S. Levine, P. Abbeel, M. Jordan, and P. Moritz. Trust region policy optimization. In International conference on machine learning, pages 1889--1897, 2015
2015
-
[30]
Schulman, F
J. Schulman, F. Wolski, P. Dhariwal, A. Radford, and O. Klimov. Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347, 2017
2017 arXiv
-
[31]
Shani, Y
L. Shani, Y. Efroni, A. Rosenberg, and S. Mannor. Optimistic policy optimization with bandit feedback. In International Conference on Machine Learning, pages 8604--8613. PMLR, 2020
2020
-
[32]
Sherman, T
U. Sherman, T. Koren, and Y. Mansour. Improved regret for efficient online reinforcement learning with linear function approximation. arXiv preprint arXiv:2301.13087, 2023
2023 arXiv
-
[33]
Wu and W
R. Wu and W. Sun. Making rl with preference-based feedback efficient via randomization. arXiv preprint arXiv:2310.14554, 2023
2023 arXiv
-
[34]
Zimin and G
A. Zimin and G. Neu. Online learning in episodic markovian decision processes by relative entropy policy search. In Advances in Neural Information Processing Systems 26: 27th Annual Conference on Neural Information Processing Systems 2013., 2013
2013
Reviewed August 8, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.