REVIEW 4 major objections 4 minor 24 references
NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations
T0 review · 4 major / 4 minor · reviewed 2026-08-10 · deepseek-v4-flash
Pith's one-line read Novelty-based skill termination lifts maze success by 178%.
desk verdict A plausible skill-termination idea with real transfer merit, but the headline numbers lean on per-environment threshold tuning that the paper only partially owns. 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 state-action novelty module, instantiated as an Intrinsic Curiosity Module (ICM) that predicts the next state's feature encoding from $(s,a)$ and uses the squared $L^2$ prediction error as the novelty score $\chi(s,a)$. The decomposition $\chi(s,a) = \chi(s)\,\chi(a|s)$ gives the method its semantics: state novelty flags unfamiliar regions that deserve more decisions, while conditional action novelty flags crossroads where many actions are plausible and the current skill is likely suboptimal. The same module's error ranks are converted into a Bernoulli termination distribution $p(\beta|s,a)$ via per-environment thresholds near the 97th percentile of offline novelty values, and this termination distribution is trained jointly with the skill VAE so that the skill embedding space itself is shaped by where skills end. In downstream reinforcement learning, the termination signal turns the problem into an SMDP with discounted $Q$-updates over the variable duration $k$ of each skill, regularized toward the learned skill prior; this is the mechanism that lets the agent decide at bottlenecks rather than at fixed intervals.
What would settle it
In a fresh maze, compute $V^{\mu}(s)$ and $Q^{\mu}(s,o)$ for the fixed-length skill policy, mark all states where the termination improvement theorem says terminating raises value, and compare those with NBDI's top-1% state-action novelty points; if the overlap is no better than chance, the claimed mechanism is falsified.
Extended reading notes
Core claim
The central claim is that critical decision points for temporally abstracted behavior—places where a skill should hand control back to a higher-level policy—can be identified from task-agnostic offline trajectories by state-action novelty alone. The paper defines novelty as $\chi(s,a) = \chi(s)\,\chi(a|s)$, so a high value means both that the state is rare in the demonstration data and that, given that state, many different actions have been taken. An Intrinsic Curiosity Module (ICM) estimates this quantity through the squared $L^2$ prediction error of a next-state feature model. State-action pairs whose novelty exceeds roughly the 97th percentile of offline values are labeled as terminations, and a model $p(\beta|s,a)$ is trained to predict them jointly with the skill encoder, decoder, and prior through an evidence-lower-bound objective. Downstream, the agent runs Soft Actor-Critic over the Semi-Markov Decision Process induced by variable-length skills, terminating either at a predicted $\beta=1$ or at the maximum length $H=30$. On this machinery the paper reports that NBDI outperforms the fixed-length skill prior SPiRL by 84.62% and 177.78% success rate in the two maze navigation tasks, by 67.16% in sparse block stacking, and by 22.33% in completed kitchen subtasks, and visualizations show the termination points clustering at crossroads and subtask boundaries.
Load-bearing premise
The method assumes that a state–action pair with high next-state prediction error on the task-agnostic dataset marks a state where ending a skill helps downstream learning, that per-environment novelty thresholds tuned to downstream performance keep that meaning, and that dataset stochasticity stays low enough for prediction error to separate decision points from noise.
Editorial extensions
If this is right
- Switching from fixed-length to novelty-terminated skills improves downstream RL performance in all four tested environments, with the largest relative gain in the harder 40×40 maze (177.78% over SPiRL).
- Conditional action novelty, not state novelty alone, is the main driver of the improvement; the full state-action product performs best in ablations.
- The termination rule transfers to downstream tasks with substantially different environment configurations when observations are agent-centered crops.
- NBDI also improves the skill-based meta-RL method SiMPL when used during skill extraction, roughly quadrupling early success rate in the maze meta-training phase relative to fixed-length skills.
- The approach remains useful on mediocre stochastic behavior-cloning data, but its decision-point detection degrades as dataset stochasticity increases toward random walks.
Reading between the lines
- Because novelty is computed purely from offline data without rewards or task labels, the same termination criterion could apply to offline skill execution and to settings where demonstrations arrive incrementally; the paper names offline skill execution as future work, and the rest is my inference, not a paper claim.
- If the observed correspondence between thresholds and the ~97th percentile holds across new environments, NBDI reduces termination-condition design to a percentile sweep; the paper reports its three thresholds match that percentile but does not claim it as a general law.
- A causal test of the story would compare NBDI's termination points with an oracle built from the termination-improvement theorem: if top-novelty states do not align with states where $Q^{\mu}(s,o) < V^{\mu}(s)$, the reported gains would need another explanation, such as exploration rather than better decision points.
- The measured sensitivity to dataset stochasticity suggests that novelty-based termination inherits data quality; for datasets dominated by near-initial-state noise, a different signal, such as epistemic uncertainty of the dynamics model, would be needed to recover meaningful decision points.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes NBDI, a termination condition for variable-length skill extraction from task-agnostic demonstrations. State-action novelty is computed by the Intrinsic Curiosity Module (ICM) prediction error on offline demonstrations, a threshold converts novelty into Bernoulli termination labels, and a termination distribution β is trained jointly with a skill decoder and skill prior in a latent variable model. In downstream RL, a high-level policy selects skill embeddings and the low-level policy executes until β signals termination or a maximum length H=30 is reached. The method is evaluated on two maze navigation tasks, sparse block stacking, and D4RL Kitchen, showing consistent improvements over SPiRL and other baselines, alongside ablations on novelty components, thresholds, and termination distribution, and comparisons to LOVE and relative novelty.
Significance. If the empirical claim holds, NBDI addresses a real limitation of fixed-length skill extraction by allowing termination at decision points such as crossroads or subtask-completion states, without needing downstream rewards or task labels for the termination model. The paper's strengths include an open-source implementation, a broad set of baselines and ablations, a comparison with other variable-length skill methods, and a demonstration that NBDI improves SiMPL meta-RL. The principal weakness is that the core hyperparameter—the novelty threshold—is tuned per environment on downstream task performance, which tempers the 'task-agnostic transfer' claim. A fixed, pre-specified percentile rule that performs as well without downstream tuning would make the contribution substantially stronger.
major comments (4)
- [5.1, A.1, B.3] The termination labels are produced by thresholding ICM prediction error, and the thresholds are environment-specific and were tuned using downstream task performance: Appendix B.3 states 'the thresholds we tuned through experiments (see Appendix A.1)'. Appendix A.1, Figure 10a shows that in Kitchen both NBDI-th0.1 and NBDI-th0.5 lose the clear margin over SPiRL. Because the threshold determines which state-action pairs train the termination distribution β, and β controls downstream skill switching, the headline claim of a task-agnostic termination condition is conditional on a per-environment hyperparameter fitted to the downstream task. Please either evaluate a fixed, pre-specified rule (e.g., the top 1% or top 3% of the task-agnostic novelty distribution) without downstream tuning, or reframe the contribution as requiring a tuned threshold and report the tuning cost.
- [4.2] Theorem 4.1 justifies terminating when V^μ(s) > Q^μ(s,o), but the paper does not show that high ICM prediction error identifies states satisfying this condition. Figure 3 was computed in an 8x8 grid with exact counts, not with the ICM estimator, and the connection is supported only by qualitative visualizations in Figure 2. Please provide a quantitative test on the task-agnostic datasets—for example, measuring the agreement between ICM error and termination-improvement frequency or a proxy such as action-value dispersion—or weaken the theoretical framing to a heuristic.
- [5.2 (Algorithm 1), B.2 (Algorithm 2), 6.3] The termination distribution is defined inconsistently across the manuscript: Algorithm 1 samples β_{t+k} ∼ p(β_{t+k}|s_{t+k}, a_{t+k}); Appendix B.2, Algorithm 2 samples (a_{t+k}, β_{t+k}) ∼ p_ψ(·|z_t, s_t); and Section 6.3 describes the termination distribution as p(β|z,s). Because the method is defined by how β is learned and used in the downstream loop, this discrepancy affects both interpretation and reproduction. Please specify the exact distribution used and correct the pseudocode accordingly.
- [I.2 vs B.3] The labeling rule is ambiguous: Appendix I.2 states that 'state-actions within the top 1% prediction error percentile serve well as critical decision points', while Appendix B.3 reports thresholds corresponding to 96–97th percentiles of the novelty distribution (i.e., the top 3–4%). These are different labeling rules, and the choice materially changes the termination labels. Please report the exact rule, including how the threshold is converted into Bernoulli labels for the decoder, and align the two appendices.
minor comments (4)
- [Figure 5 table] In Sparse Block Stacking, the 95% confidence intervals for NBDI (1.12 ± 0.16) and SPiRL (0.67 ± 0.29) nearly overlap; a paired-seed comparison or a significance test would strengthen the improvement claim in this environment.
- [Figure 3] The three panels are small and use similar color scales, which makes cross-panel comparison difficult; consider plotting termination improvement, conditional action novelty, and state novelty on a common color scale with larger panels.
- [Acknowledgments and Figure 13] There are typos: 'Institute of Information& Coummunications Technology' in the acknowledgments and 'termianted skills' in the Figure 13 caption.
- [6.6] The claim that novelty estimation is robust to dataset size is based on visual inspection of box plots (Figure 8, right); report quantitative statistics, such as rank correlations between dataset usage and novelty scale, if this robustness is a stated contribution.
Circularity Check
Headline gains are conditional on per-environment novelty thresholds tuned on downstream performance; the post-hoc percentile rule does not establish a parameter-free task-agnostic termination condition.
-
fitted input called prediction
[Appendix B.3; Appendix A.1; Section 5.1; Figure 5/table]
"In practice, the thresholds we tuned through experiments (see Appendix A.1) approximately correspond to the 97th percentile of the novelty values computed over task-agnostic demonstrations—e.g., kitchen = 0.3 (97.47th percentile), maze = 50 (96.86th percentile), and block stacking = 40 (96.12th percentile)."
The termination labels beta are constructed by thresholding the ICM novelty score (Section 5.1), and the thresholds are explicitly tuned through downstream experiments rather than derived from the task-agnostic data alone. The downstream success-rate claims in Figure 5 and its table are then reported for the same environments using these tuned thresholds; Appendix A.1 shows that alternative thresholds (th0.1, th0.5) lose the clear margin over SPiRL, so the reported improvement is conditional on the fitted threshold value. The 'top 1% prediction error percentile' statement in Appendix I.2 is a post-hoc summary rather than a pre-specified rule, and it even conflicts with the 97th-percentile description in Appendix B.3.
full rationale
The central empirical comparison is not fully circular: after the termination distribution is trained, downstream RL performance is measured on held-out tasks with new maze layouts, larger block-stacking configurations, and unseen kitchen subtask sequences, so the learned skills and the RL gains have independent content. There is no load-bearing self-citation chain, no imported uniqueness theorem, and no ansatz smuggled in solely through the authors' prior work; the termination-improvement theorem is used only as motivation, not as a proof that high ICM error marks decision points. The main circularity concern is the fitted threshold: termination labels are produced by thresholding novelty, the thresholds are tuned using downstream performance (Appendix B.3, Appendix A.1), and the reported headline margins over SPiRL are sensitive to those tuned values. This makes the 'task-agnostic' transfer claim partially fitted, but because the downstream gains themselves are evaluated through fresh RL training on held-out tasks, the circularity is partial rather than a by-construction reduction of the result to its inputs.
Assumptions & free parameters
free parameters (2)
- Novelty threshold per environment =
kitchen 0.3, maze 50, block stacking 40
- Maximum skill length H =
30
assumptions (4)
- standard math Termination improvement theorem (Sutton, 1998)
- domain assumption ICM prediction error is a valid estimate of state-action novelty chi(s,a)
- ad hoc to paper High state-action novelty marks decision points where termination improves downstream learning
- domain assumption Agent-centered cropped images preserve decision-point structure across different environment layouts
Cite this review
Pith. "Pith review of NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations." pith.science (2026). https://pith.science/paper/OCCDO2XT
@misc{pith2026250112668,
author = {Pith},
title = {Pith review of: NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations},
year = {2026},
howpublished = {\url{https://pith.science/paper/OCCDO2XT}},
note = {Machine review of arXiv:2501.12668}
}
read the original abstract
Intelligent agents are able to make decisions based on different levels of granularity and duration. Recent advances in skill learning enabled the agent to solve complex, long-horizon tasks by effectively guiding the agent in choosing appropriate skills. However, the practice of using fixed-length skills can easily result in skipping valuable decision points, which ultimately limits the potential for further exploration and faster policy learning. In this work, we propose to learn a simple and effective termination condition that identifies decision points through a state-action novelty module that leverages agent experience data. Our approach, Novelty-based Decision Point Identification (NBDI), outperforms previous baselines in complex, long-horizon tasks, and remains effective even in the presence of significant variations in the environment configurations of downstream tasks, highlighting the importance of decision point identification in skill learning.
Figures
Figures from the paper (14 more)
Reference graph
Works this paper leans on
-
[1]
Variational option discovery algorithms
Achiam, J., Edwards, H., Amodei, D., and Abbeel, P. Variational option discovery algorithms. arXiv preprint arXiv:1807.10299,
-
[4]
CHANNELS 8, 16, 32 # CONVOLUTION LAYERS 3 SKILL DECODER HIDDEN DIM 128 # HIDDEN LAYERS 6 15 A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations B.2. Reinforcement Learning with NBDI In downstream learning, our objective is to learn a skill policy πθ(z|st) that maximizes the expected sum of discounted rewards...
work page 2018
-
[6]
Hierarchical few-shot imitation with skill tran- sition models
Hakhamaneshi, K., Zhao, R., Zhan, A., Abbeel, P., and Laskin, M. Hierarchical few-shot imitation with skill tran- sition models. arXiv preprint arXiv:2107.08981,
-
[8]
D., Saeedi, A., Gautam, S., and Gershman, S
Kulkarni, T. D., Saeedi, A., Gautam, S., and Gershman, S. J. Deep successor reinforcement learning. arXiv preprint arXiv:1606.02396,
-
[16]
Parrot: Data-driven behavioral priors for re- inforcement learning
Singh, A., Liu, H., Zhou, G., Yu, A., Rhinehart, N., and Levine, S. Parrot: Data-driven behavioral priors for re- inforcement learning. arXiv preprint arXiv:2011.10024,
arXiv 2011
-
[17]
The agent starts at the white point and aims to reach the red point
Examples of maze layouts used for training (left) and downstream task (right). The agent starts at the white point and aims to reach the red point. F. Task-Agnostic Dataset Size and Coverage The number of task-agnostic trajectories collected for each environment were as follows: maze (85,000 trajectories), block stacking (37,000 trajectories), and kitchen...
work page 2004
-
[23]
(Kitchen) (CC BY 4.0). Note that task and environment setup differ between the training data and the downstream task, demonstrating the model’s capacity to handle unseen downstream tasks. Kitchen Environment The kitchen environment is provided by the D4RL benchmark (Fu et al., 2021), featuring seven manipulable objects. The training trajectories consist o...
work page 2021
-
[24]
In the kitchen environment, the structure of the dynamic model is the same as its feature encoder. In the maze and sparse block stacking environment, a single fully-connected layer with hidden dimension 52 and 70 have been used, respectively. The state-action novelty χ(s, a) is computed as the squared L2 distance between ˆϕ(ϕ(st), at) and ϕ(st+1), represe...
work page 2021
Show all 24 references
-
[1993]
Offline reinforce- ment learning with implicit q-learning
Kostrikov, I., Nair, A., and Levine, S. Offline reinforce- ment learning with implicit q-learning. arXiv preprint arXiv:2110.06169,
-
[1994]
Ex- ploration by random network distillation
Burda, Y ., Edwards, H., Storkey, A., and Klimov, O. Ex- ploration by random network distillation. arXiv preprint arXiv:1810.12894,
-
[1997]
encoder qϕ(z|τ, β) and a decoder pψ(at, βt|z, st). To learn model parameters ϕ and ψ, the latent variable model receives a randomly sampled experience τ from the training dataset D along with a termination condition vector β from the state-action novelty module, and tries to r...
2016
-
[1998]
Mujoco: A physics engine for model-based control
Todorov, E., Erez, T., and Tassa, Y . Mujoco: A physics engine for model-based control. In 2012 IEEE/RSJ inter- national conference on intelligent robots and systems, pp. 5026–5033. IEEE,
2012
-
[2001]
Q-cut—dynamic discovery of sub-goals in reinforcement learning
Menache, I., Mannor, S., and Shimkin, N. Q-cut—dynamic discovery of sub-goals in reinforcement learning. In Ma- chine Learning: ECML 2002: 13th European Conference on Machine Learning Helsinki, Finland, August 19–23, 2002 Proceedings 13, pp. 295–306. Springer,
2002
-
[2002]
J., and Lim, J
Nam, T., Sun, S.-H., Pertsch, K., Hwang, S. J., and Lim, J. J. Skill-based meta-reinforcement learning. arXiv preprint arXiv:2204.11828,
-
[2003]
J., and Wierstra, D
Gregor, K., Rezende, D. J., and Wierstra, D. Variational intrinsic control. arXiv preprint arXiv:1611.07507,
-
[2004]
P., and Barto, A
Simsek, ¨O., Wolfe, A. P., and Barto, A. G. Identifying useful subgoals in reinforcement learning by local graph partitioning. In Raedt, L. D. and Wrobel, S. (eds.), Ma- chine Learning, Proceedings of the Twenty-Second In- ternational Conference (ICML 2005), Bonn, Germany, Aug...
2005
-
[2005]
URL https://doi
doi: 10.1145/1102351.1102454. URL https://doi. org/10.1145/1102351.1102454. S ¸ims ¸ek, ¨O., Wolfe, A. P., and Barto, A. G. Identifying useful subgoals in reinforcement learning by local graph partitioning. In Proceedings of the 22nd international conference on Machine learnin...
-
[2012]
Trail: Near-optimal imitation learning with suboptimal data
Yang, M., Levine, S., and Nachum, O. Trail: Near-optimal imitation learning with suboptimal data. arXiv preprint arXiv:2110.14770,
-
[2016]
C., Bellemare, M
Machado, M. C., Bellemare, M. G., and Bowling, M. A laplacian framework for option discovery in reinforce- ment learning. In International Conference on Machine Learning, pp. 2295–2304. PMLR, 2017a. Machado, M. C., Rosenbaum, C., Guo, X., Liu, M., Tesauro, G., and Campbell, M....
-
[2018]
Opal: Offline primitive discovery for acceler- ating offline reinforcement learning
Ajay, A., Kumar, A., Agrawal, P., Levine, S., and Nachum, O. Opal: Offline primitive discovery for acceler- ating offline reinforcement learning. arXiv preprint arXiv:2010.13611,
2010 arXiv
-
[2019]
Accelerating reinforcement learning with learned skill priors
Pertsch, K., Lee, Y ., and Lim, J. Accelerating reinforcement learning with learned skill priors. In Conference on robot learning, pp. 188–204. PMLR, 2021a. Pertsch, K., Lee, Y ., Wu, Y ., and Lim, J. J. Guided rein- forcement learning with learned skills. arXiv preprint arXiv...
-
[2020]
Option discovery in hierarchical reinforcement learning using spatio-temporal clustering
Srinivas, A., Krishnamurthy, R., Kumar, P., and Ravin- dran, B. Option discovery in hierarchical reinforcement learning using spatio-temporal clustering. arXiv preprint arXiv:1605.05359,
-
[2021]
Investigating the effectiveness of bpe: The power of shorter sequences
Gall´e, M. Investigating the effectiveness of bpe: The power of shorter sequences. In Proceedings of the 2019 confer- ence on empirical methods in natural language process- ing and the 9th international joint conference on natural language processing (EMNLP-IJCNLP), pp. 1375–1381,
2019
-
[2022]
Sekar, R., Rybkin, O., Daniilidis, K., Abbeel, P., Hafner, D., and Pathak, D
URL https://arxiv.org/ abs/2209.01947. Sekar, R., Rybkin, O., Daniilidis, K., Abbeel, P., Hafner, D., and Pathak, D. Planning to explore via self-supervised world models. In International Conference on Machine Learning, pp. 8583–8592. PMLR,
Reviewed August 10, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.