TD-Grokking: Learning from Zero-Reward Problems by Training-Time Decomposition
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
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023Let's Verify Step by Stepcited by 3,598
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
14 more
- 2023Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotationscited by 895
- 2022Decomposed Prompting: A Modular Approach for Solving Complex Taskscited by 554
- 2023Plan-and-Solve Prompting: Improving Zero-Shot Chain-of-Thought Reasoning by Large Language Modelscited by 550
- 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 525
- 2024Improve Mathematical Reasoning in Language Models by Automated Process Supervisioncited by 426
- 2021A Diverse Corpus for Evaluating and Developing English Math Word Problem Solverscited by 314
- 2024ProcessBench: Identifying Process Errors in Mathematical Reasoningcited by 277
- 2024Benchmarking Large Language Models on Answering and Explaining Challenging Medical Questionscited by 106
- 2025Scaf-GRPO: Scaffolded Group Relative Policy Optimization for Enhancing LLM Reasoningcited by 48
- 2025QuestA: Expanding Reasoning Capacity in LLMs via Question Augmentationcited by 39
- 2025Nudging the Boundaries of LLM Reasoningcited by 25
- 2025RL Grokking Recipe: How Does RL Unlock and Transfer New Algorithms in LLMs?cited by 23
- 2026Self-Hinting Language Models Enhance Reinforcement Learningcited by 19
- 2026Learning to Hint for Reinforcement Learningcited by 9
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Reinforcement Learning for Hierarchical Reasoning Rewards: Minimax-Optimal Rates with Transformers
- 2025Zero Reinforcement Learning Towards General Domains
- 2025A Minimalist Approach to LLM Reasoning: from Rejection Sampling to Reinforce
- 2026Resource-Efficient Reinforcement for Reasoning Large Language Models via Dynamic One-Shot Policy Refinement
- 2026Training Reasoning Models on Saturated Problems via Failure-Prefix Conditioning
- 2026GRPO Beyond English: A Large-Scale Study of GRPO in Non-English and Multilingual Settings
- 2025TreeRPO: Tree Relative Policy Optimization
- 2025DRPO: Efficient Reasoning via Decoupled Reward Policy Optimization