REVIEW 5 major objections 5 minor 17 references
Adaptive Data Exploitation in Deep Reinforcement Learning
T0 review · 5 major / 5 minor · reviewed 2026-08-10 · deepseek-v4-flash
Pith's one-line read The paper claims that the number of update epochs per episode in on-policy RL is a tunable resource that a multi-armed bandit can schedule online, and reports significant gains in data efficiency and compute.
desk verdict Bandit-scheduled update epochs is a sensible and well-tested idea, but without fixed K=1/K=2 baselines the paper cannot yet show that the adaptation itself helps. 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 number of update epochs (NUE), the count of gradient steps the agent takes on a collected episode's experience. The carrying mechanism is a multi-armed bandit whose arms are candidate NUE values: the reward for selecting an arm is the episode's mean value-predicted return, averaged over a sliding FIFO window, and the selection rules are UCB, Gaussian Thompson sampling, or round-robin. This creates a self-referential loop in which the value network both produces the bandit's reward signal and is the network being trained for the chosen number of epochs.
What would settle it
Run vanilla PPO with a fixed NUE of 1 on the same Procgen environments and compare its returns and measured FLOPs against PPO+ADEPT(U); if the fixed one-epoch schedule already matches the adaptive agent's performance and overhead, then the adaptive bandit is not what produces the gains. Conversely, corrupt the value-prediction reward signal with artificial noise and check whether scheduling quality degrades in step with the noise, which would confirm the loop's dependence on that signal.
Extended reading notes
Core claim
The paper's central claim is that data reuse per episode is itself a controllable resource that can be scheduled online. ADEPT frames each candidate number of update epochs (NUE) as an arm of a multi-armed bandit; after each episode it selects an arm, updates the policy and value networks for that many epochs, and feeds the mean value-predicted return of the episode into a sliding-window estimate of that arm's quality. Three schedulers are instantiated: UCB (ADEPT(U)), Gaussian Thompson sampling (ADEPT(G)), and round-robin (ADEPT(R)). On the sixteen Procgen environments, PPO+ADEPT(U) is reported to outperform vanilla PPO on 14 environments, reach the highest performance on 6, and produce the lowest computational overhead on 11, while DrAC+ADEPT(U) reaches the highest performance on 7 environments at about 69.1% of the overhead of vanilla DrAC. The paper interprets these results as evidence that different tasks and learning stages call for different amounts of data reuse, and that a lightweight bandit scheduler can find those amounts automatically.
Load-bearing premise
The scheduling loop assumes the value network's predicted returns are reliable enough to rank the update-epoch counts, even though the same network is being updated a different number of times depending on what the bandit selects.
Editorial extensions
If this is right
- Practitioners can wrap PPO or DrAC with ADEPT and expect equal or better returns at roughly 70% of the update-phase FLOPs, without hand-tuning a fixed epoch count.
- The agent regulates its own overfitting by switching to fewer update epochs when value predictions plateau, reducing repetition of the same batch and preserving plasticity.
- Because ADEPT adds no auxiliary networks or extra loss terms, it can be combined with exploration and augmentation methods rather than competing with them.
- The fixed three-epoch schedule reported as best for Procgen is not optimal at every learning stage, and an automatic scheduler can specialize the reuse count per environment.
Reading between the lines
- The same bandit view could be applied to replay ratio or gradient steps per sample in off-policy RL, another hand-tuned reuse knob; this would be a direct transfer of the NUE mechanism to a different training loop.
- Replacing the value-predicted return with the true episode return (available in simulation) would isolate how much of the gain comes from the scheduler itself rather than from the quality of its self-referential signal, an experiment the paper leaves open.
- If ADEPT's generalization gains come from limiting data reuse, the effect should be weaker on a single fixed level than on the full procedural distribution, since the overfitting mechanism it targets is stronger under distribution shift; this is a testable prediction.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper introduces ADEPT, a multi-armed bandit framework that adaptively selects the number of update epochs (NUE) per episode for on-policy RL algorithms (PPO and DrAC). Three schedulers are proposed: UCB (ADEPT(U)), Gaussian Thompson sampling (ADEPT(G)), and round-robin (ADEPT(R)). The method is evaluated on Procgen, MiniGrid, and PyBullet, with claims of improved data efficiency, better generalization, and reduced computational overhead. The central empirical claim is that PPO+ADEPT(U) outperforms vanilla PPO in 14 Procgen environments while using the least training FLOPS in 11. The paper also provides ablation studies, hyperparameter searches, and a FLOPS calculation appendix.
Significance. If the headline results held, ADEPT would be a simple, useful plug-in for reducing RL training compute while preserving or improving performance. The paper has tangible strengths: it releases code, evaluates on three benchmarks with multiple seeds, includes ablations, and uses proper aggregate metrics (mean, median, IQM, OG). However, the current evidence does not establish that adaptation, rather than a lower average number of update epochs, drives the gains. The missing fixed-epoch baselines are the decisive omission, and the FLOPS metric is defined in a way that makes the compute savings partially tautological. These issues are fixable within the scope of an empirical paper, but they currently prevent acceptance.
major comments (5)
- [Section 5.2, Figure 5] The headline performance claim is confounded by the absence of fixed-epoch baselines. Figure 5 shows that ADEPT(U) selects K=1 roughly 60% of the time, for an average NUE near 1.6, while the vanilla PPO comparison uses the fixed schedule NUE=3 (Section 5.1 and Appendix B.1). PPO+ADEPT(U) is reported to outperform vanilla PPO in 14 environments, but the paper never runs PPO with fixed K=1 or K=2 under the same hyperparameters and seeds. If fixed K=1 attains comparable or better test returns, the gains are explained by the regularization effect of fewer gradient steps, not by adaptive choice, and the claim in Section 4 that 'a dynamic NUE value can reduce reliance on specific data and preserve plasticity' is untested. The fixed low-NUE controls are essential to support Q1 and the abstract's claim of adaptive improvement.
- [Appendix G, Eqs. (16)-(17)] The computational-overhead result is largely a mechanical consequence of the FLOPS definition. O_forward is defined as proportional to N_update_epochs and O_backward = 2*O_forward, so total training FLOPS is linear in the average NUE. Because ADEPT(U) mostly chooses K=1 (Figure 5), its average NUE of ~1.6 mechanically yields an update-phase FLOPS ratio near 1.6/3 = 53% relative to vanilla PPO's NUE=3. The reported 70% total overhead (Section 5.2) follows from combining this with the constant sampling overhead. The authors should report the average NUE per scheduler and compare against fixed-NUE baselines at matched FLOPS and matched performance; without this, 'minimum computational overhead' does not demonstrate the benefit of adaptation.
- [Appendices B.1 and F.1, Figure 4] Hyperparameters are selected on the evaluation benchmark itself. The authors searched c ∈ {0.1, 1.0, 5.0} and W ∈ {10, 50, 100} on Procgen and then report the best configuration as the main result; the same procedure is used for PyBullet (Appendix B.3). This is test-set tuning, which inflates the reported gains and invalidates the claim that ADEPT is parameter-free. In addition, Figure 4's 'ADEPT(*)' bars take the best result among RRS, UCB, and GTS for each metric; this is a multiple-comparisons selection without correction. Please either fix the hyperparameters a priori, use a separate validation split, or report each variant's results and justify the 'best-of-three' aggregation.
- [Section 5.2, Figure 3] The Procgen 'data efficiency' comparison is plotted against FLOPS, not environment steps. Q1 asks whether ADEPT improves data efficiency, but the main training curves show performance per FLOP, which is a computational-efficiency measure; data efficiency in the sense of returns per environment step is not shown for Procgen (the MiniGrid and PyBullet figures do use environment steps). The paper should include environment-step curves to support the data-efficiency claim, or consistently use the term 'computational efficiency' for the Procgen results.
- [Equation (3) and Section 6] The bandit's reward signal is the mean value-predicted return after K update epochs, but this value network is the same network being updated by the selected K. The Discussion concedes that inaccurate value predictions will directly affect scheduling quality, yet the paper provides no evidence that this self-referential estimate is reliable across tasks and learning stages (e.g., a correlation analysis between the bandit reward and final policy return). Without such evidence, the claim that ADEPT automatically selects the optimal extent value (Section 1) is not validated beyond the effect of lowering the average NUE.
minor comments (5)
- [Section 4.2, Eq. (5)] The initialization of µ and σ² for the Gaussian Thompson sampling arm is not specified; please report the initial values used in the experiments.
- [Figure 3 and Appendix D] The captions state that the dotted line and dashed line represent the highest score and the lowest overhead, respectively, but these appear as markers rather than lines; please clarify the visual convention.
- [Appendix G, Eq. (14)] The formula for sampling overhead counts (Nepisode length + 1) observations per environment; note that this assumes a specific PPO implementation detail (the extra bootstrap prediction), and it should be stated explicitly that the reported FLOPS exclude environment-execution cost, which is acknowledged only in Figure 2(a).
- [Section 2.1] The sentence 'Our method provides a systematic guarantee for optimizing long-term returns' overclaims, as no formal guarantee is derived; please rephrase to 'aims to optimize' or provide a proof.
- [Section 5.1] The description of the evaluation metrics says 'the average floating point operations (FLOPS) over 16 environments and all the runs'; consider clarifying that the FLOPS totals are per method and are then averaged, and specify the seed count in the main text rather than only in Appendix.
Circularity Check
Compute-savings claim reduces to definition of FLOPS; fixed low-NUE baselines missing.
-
self definitional
[Appendix G Eqs. (15)-(18); Section 4 key insight (iii); Section 5.2; Figure 5]
"O_forward = Obs1 ∗ B ∗ Nbatches ∗ N_update epochs ... O_total = (O_sampling + O_update) ∗ N_episodes (Appendix G). "By minimizing unnecessary updates through adaptive NUE tuning, we can significantly reduce the overall computational overhead" (Section 4). "ADEPT(U) primarily selects K = 1, while K = 2 and K = 3 each account for approximately 20%" (Figure 5). "Meanwhile, it produces the minimum computational overhead in 11 environments" (Section 5.2)."
The overhead metric is defined to be linear in N_update_epochs: O_forward = Obs1*B*Nbatches*N_update_epochs and O_total sums O_update per episode. Consequently, any scheduler whose average selected NUE is below the vanilla fixed-3 schedule is guaranteed to show lower measured FLOPS. Figure 5 shows ADEPT(U) chooses K=1 for roughly 60% of updates (K=2 and K=3 each about 20%), so its average NUE is about 1.6 versus 3 for vanilla PPO. The 'minimum computational overhead in 11 environments' is therefore an arithmetic consequence of the metric definition plus the scheduler's preference for K=1; it does not measure the value of adaptation. The FLOPS-savings 'prediction' is fixed by the definition of the metric itself.
full rationale
The central performance claim is not circular: ADEPT's returns are evaluated against environment scores on Procgen/MiniGrid/PyBullet, which are external to the scheduling mechanism, and the paper honestly reports learning curves and aggregate metrics. However, the computational-efficiency claim is a definitional artifact: Appendix G defines FLOPS so that update cost is proportional to N_update_epochs, and ADEPT(U) mostly selects K=1, so its lower measured overhead follows by construction rather than from adaptive scheduling. The paper also omits fixed NUE=1 and NUE=2 baselines, so the performance comparison cannot isolate adaptation from the regularizing effect of fewer gradient steps; this is a confound, though not a circular reduction. The Discussion's admitted reliance on value-predicted returns creates a self-referential bandit signal that is not validated against true returns, but the paper explicitly flags it. I found no load-bearing self-citations or imported uniqueness theorems: the authors' prior bandit work (AIRS) is cited only as context, and the NUE-candidate justification comes from external Procgen/DrAC papers. Overall, one advertised result (compute savings) reduces by construction, while the main empirical performance claim retains independent content, giving a partial circularity score of 6.
Assumptions & free parameters
free parameters (5)
- UCB exploration coefficient c =
5.0 (Procgen), 1.0 (MiniGrid)
- Sliding window length W =
10 (UCB on Procgen), 50 (GTS on Procgen), 50 (MiniGrid)
- GTS step size eta =
1.0 (PPO, Procgen), 0.1 (DrAC, Procgen)
- Initial GTS variance =
unspecified
- NUE candidate set K =
{3,2,1} for Procgen; {4,2,1}, {6,4}, {8,4} for MiniGrid; {10,5,1}, {10,5} for PyBullet
assumptions (4)
- standard math MAB algorithms (UCB, GTS) provide valid solutions for the NUE-selection problem
- domain assumption The value network's predicted return is a usable reward signal for scheduling
- ad hoc to paper Arm rewards for GTS follow a normal distribution
- domain assumption FLOPS of an update is forward FLOPS plus twice backward FLOPS and excludes environment execution
Cite this review
Pith. "Pith review of Adaptive Data Exploitation in Deep Reinforcement Learning." pith.science (2026). https://pith.science/paper/BPZWB572
@misc{pith2026250112620,
author = {Pith},
title = {Pith review of: Adaptive Data Exploitation in Deep Reinforcement Learning},
year = {2026},
howpublished = {\url{https://pith.science/paper/BPZWB572}},
note = {Machine review of arXiv:2501.12620}
}
read the original abstract
We introduce ADEPT: Adaptive Data ExPloiTation, a simple yet powerful framework to enhance the **data efficiency** and **generalization** in deep reinforcement learning (RL). Specifically, ADEPT adaptively manages the use of sampled data across different learning stages via multi-armed bandit (MAB) algorithms, optimizing data utilization while mitigating overfitting. Moreover, ADEPT can significantly reduce the computational overhead and accelerate a wide range of RL algorithms. We test ADEPT on benchmarks including Procgen, MiniGrid, and PyBullet. Extensive simulation demonstrates that ADEPT can achieve superior performance with remarkable computational efficiency, offering a practical solution to data-efficient RL. Our code is available at https://github.com/yuanmingqi/ADEPT.
Figures
Figures from the paper (33 more)
Reference graph
Works this paper leans on
-
[1]
∗ Nenvironments ∗ Obs1 (14) where +1 is for predicting the returns of the next observations at the end of the episode, as shown in the PPO implementation of CleanRL (Huang et al., 2022). For the model update phase, the computational overhead is Oupdate = Oforward + Obackward, (15) where Oforward = Obs1 ∗ B ∗ Nbatches ∗ Nupdate epochs (16) and Obackward = ...
work page 2022
-
[2]
(17) Here, B is the batch size, and the overhead ratio of a forward pass to a backward pass is 1:2 as suggested by (Dario & Danny, 2018). Finally, the total computational overhead is Ototal = (Osampling + Oupdate) ∗ Nepisodes (18) In the Procgen experiments, we have Obs1 = 528384FLOPS Nenvironments = 64 Nepisode length = 256 Nepisodes = 1525 B = 2048 Nbat...
work page 2018
-
[7]
Dropout: a simple way to prevent neural networks from overfitting
Srivastava, N., Hinton, G., Krizhevsky, A., Sutskever, I., and Salakhutdinov, R. Dropout: a simple way to prevent neural networks from overfitting. The journal of machine learning research, 15(1):1929–1958,
1929
-
[9]
Rewarding episodic visitation discrepancy for exploration in reinforcement learning
Yuan, M., Li, B., Jin, X., and Zeng, W. Rewarding episodic visitation discrepancy for exploration in reinforcement learning. In Deep RL Workshop NeurIPS 2022, 2022a. Yuan, M., Pun, M.-O., and Wang, D. R ´enyi state entropy maximization for exploration acceleration in reinforce- ment learning. IEEE Transactions on Artificial Intelli- gence, 2022b. Yuan, M....
work page 2022
-
[10]
39 Adaptive Data Exploitation in Deep Reinforcement Learning G
The mean and standard deviation are computed across all the environments. 39 Adaptive Data Exploitation in Deep Reinforcement Learning G. Calculation of Computational Overhead To compare the computational efficiency of the baseline algorithms and ADEPT, we utilize the floating point operations (FLOPS) as the KPI. Moreover, we only count the computational ...
work page 2020
-
[12]
of DrAC for the experiments, and Table 1 lists the shared and fixed hyperparameters. For each Procgen environment, Table 2 lists the best augmentation method of DrAC as reported in (Raileanu & Fergus, 2021). The candidates of NUE are also set as K = {3, 2, 1} for all the experiments. For ADEPT(U) and ADEPT(G), we run the same hyperparameter search as the ...
work page 2021
-
[50]
Finally, Table 3 illustrates the PPO hyperparameters, which remain fixed throughout all the experiments. B.3. PyBullet Figure 11: Screenshots of the four PyBullet environments. From left to right: Ant, Hopper, HalfCheetah, and Walker2D. Finally, we perform the experiments on the PyBullet benchmark using the PPO implementation of (Kostrikov, 2018). Since (...
work page 2018
-
[255]
In this part, we use the official implementation (Raileanu et al.,
Reward normalization Yes for PPO, No for DrAC LSTM No Stacked frames No Environment steps 25000000 Episode steps 256 Number of workers 1 Environments per worker 64 Optimizer Adam Learning rate 5e-4 GAE coefficient 0.95 Action entropy coefficient 0.01 Value loss coefficient 0.5 Value clip range 0.2 Max gradient norm 0.5 Batch size 2048 Discount factor 0.99...
Show all 17 references
-
[1988]
Master- ing visual continuous control: Improved data-augmented reinforcement learning
Yarats, D., Fergus, R., Lazaric, A., and Pinto, L. Master- ing visual continuous control: Improved data-augmented reinforcement learning. In International Conference on Learning Representations, 2021a. Yarats, D., Kostrikov, I., and Fergus, R. Image aug- mentation is all you n...
2020
-
[2002]
Badia, A
doi: 10.1023/A:1013689704352. Badia, A. P., Piot, B., Kapturowski, S., Sprechmann, P., Vitvitskyi, A., Guo, Z. D., and Blundell, C. Agent57: Outperforming the atari human benchmark. In Inter- national conference on machine learning, pp. 507–517. PMLR, 2020a. Badia, A. P., Spre...
-
[2015]
Proximal policy optimization algorithms
Schulman, J., Wolski, F., Dhariwal, P., Radford, A., and Klimov, O. Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347,
-
[2016]
Schulman, J., Moritz, P., Levine, S., Jordan, M., and Abbeel, P
URL http: //arxiv.org/abs/1511.05952. Schulman, J., Moritz, P., Levine, S., Jordan, M., and Abbeel, P. High-dimensional continuous control using generalized advantage estimation. Proceedings of the International Conference on Learning Representations,
-
[2017]
The policy loss is defined as: Lπ(θ) = −Eτ ∼π [min (ρt(θ)At, clip (ρt(θ), 1 − ϵ, 1 + ϵ) At)] , (7) where ρt(θ) = πθ(at|st) πθold (at|st) , (8) and ϵ is a clipping range coefficient
is an on-policy algorithm that is designed to improve the stability and sample efficiency of policy gradient methods, which uses a clipped surrogate objective function to avoid large policy updates. The policy loss is defined as: Lπ(θ) = −Eτ ∼π [min (ρt(θ)At, clip (ρt(θ), 1 − ...
2015
-
[2020]
W., Hilton, J., Klimov, O., and Schulman, J
Cobbe, K. W., Hilton, J., Klimov, O., and Schulman, J. Phasic policy gradient. In International Conference on Machine Learning, pp. 2020–2027. PMLR,
2020
-
[2021]
and Bai, Y
Coumans, E. and Bai, Y . Pybullet, a python module for physics simulation for games, robotics and machine learn- ing. URL http://pybullet.org, 2016–2018. Dario, A. and Danny, H. Ai and compute. OpenAI blog,
2016
-
[2022]
Then we test the PPO agent with three ADEPT algorithms
reported the best K = 10, we set the NUE candidates as K = {10, 5, 1} and K = {10, 5}. Then we test the PPO agent with three ADEPT algorithms. Moreover, we run a similar hyperparameter search as the Procgen experiments and report the best results of each method. Table 3 illust...
-
[2023]
Lever- aging procedural generation to benchmark reinforcement learning
Cobbe, K., Hesse, C., Hilton, J., and Schulman, J. Lever- aging procedural generation to benchmark reinforcement learning. In International conference on machine learn- ing, pp. 2048–2056. PMLR,
Reviewed August 10, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.