mlsanityData to 7 Oct 2026

TD-Grokking: Learning from Zero-Reward Problems by Training-Time Decomposition

arXiv 2606.09883 ↗First posted 3 Jun 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Large language models (LLMs) have made remarkable progress in reasoning tasks, largely driven by post-training paradigms, especially reinforcement learning with verifiable rewards (RLVR). However, a critical bottleneck persists: RLVR fails on highly challenging zero-reward problems, where all sampled reasoning trajectories yield uniformly failed outcomes, providing no optimization signal to drive model improvement. Prior efforts to address this limitation, such as dense process supervision, partial reward assignment, or prefix-guided exploration, suffer from inherent task constraints or do not fully equip the policy model with the capabilities necessary to solve the original intractable problems. To address this, we propose TD-Grokking, a training-time decomposition framework for zero-reward problems. It recursively decomposes intractable root problems into self-contained, verifiable subproblems, forming hierarchical trees where solvable leaves provide non-zero rewards. Evaluations on mathematical and medical tasks show that TD-Grokking outperforms vanilla GRPO as well as all baseline approaches. Together with detailed analysis, these results confirm that training-time decomposition effectively converts zero-reward examples into usable training signals, enabling consistent performance gains. Our code and datasets are available at https://anonymous.4open.science/r/TD-Grokking-6567/.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 26 of its 45 references that are in the map. First posted on arXiv 3 Jun 2026.

Builds on

26 of 45 references.

References · 26

  1. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  3. 2025Qwen3 Technical Reportcited by 8,195
  4. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  6. 2023Let's Verify Step by Stepcited by 3,598
  7. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
  8. 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
  9. 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
  10. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
  11. 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
  12. 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
14 more
  1. 2023Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotationscited by 895
  2. 2022Decomposed Prompting: A Modular Approach for Solving Complex Taskscited by 554
  3. 2023Plan-and-Solve Prompting: Improving Zero-Shot Chain-of-Thought Reasoning by Large Language Modelscited by 550
  4. 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 525
  5. 2024Improve Mathematical Reasoning in Language Models by Automated Process Supervisioncited by 426
  6. 2021A Diverse Corpus for Evaluating and Developing English Math Word Problem Solverscited by 314
  7. 2024ProcessBench: Identifying Process Errors in Mathematical Reasoningcited by 277
  8. 2024Benchmarking Large Language Models on Answering and Explaining Challenging Medical Questionscited by 106
  9. 2025Scaf-GRPO: Scaffolded Group Relative Policy Optimization for Enhancing LLM Reasoningcited by 48
  10. 2025QuestA: Expanding Reasoning Capacity in LLMs via Question Augmentationcited by 39
  11. 2025Nudging the Boundaries of LLM Reasoningcited by 25
  12. 2025RL Grokking Recipe: How Does RL Unlock and Transfer New Algorithms in LLMs?cited by 23
  13. 2026Self-Hinting Language Models Enhance Reinforcement Learningcited by 19
  14. 2026Learning to Hint for Reinforcement Learningcited by 9

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Reinforcement Learning for Hierarchical Reasoning Rewards: Minimax-Optimal Rates with Transformers
  2. 2025Zero Reinforcement Learning Towards General Domains
  3. 2025A Minimalist Approach to LLM Reasoning: from Rejection Sampling to Reinforce
  4. 2026Resource-Efficient Reinforcement for Reasoning Large Language Models via Dynamic One-Shot Policy Refinement
  5. 2026Training Reasoning Models on Saturated Problems via Failure-Prefix Conditioning
  6. 2026GRPO Beyond English: A Large-Scale Study of GRPO in Non-English and Multilingual Settings
  7. 2025TreeRPO: Tree Relative Policy Optimization
  8. 2025DRPO: Efficient Reasoning via Decoupled Reward Policy Optimization