Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

TemplateRL: Structured Template-Guided Reinforcement Learning for LLM Reasoning

About

Reinforcement learning (RL) has emerged as an effective paradigm for enhancing model reasoning. However, existing RL methods like GRPO typically rely on unstructured self-sampling to fit scalar rewards, often producing inefficient rollouts that fail to capture transferable problem-solving strategies. To address this limitation, we propose **TemplateRL**, a structured template-guided RL framework that augments policy optimization with explicit template guidance. Our approach first constructs a problem-solving template library via MCTS on a small seed set, then seamlessly integrates this high-level structured guidance into RL training. By guiding rollout generation to align with proven template structures, TemplateRL significantly improves high-quality trajectory hit rates while reducing ineffective exploration. This structure-guided design steers the policy toward validated strategic patterns, stabilizing training dynamics, and enhancing RL sampling efficiency. Notably, the explicit template library is interpretable, editable, and supports online updates-enabling continuous updates during both training and inference. Extensive experiments demonstrate that TemplateRL outperforms GRPO by 99% on AIME and 41% on AMC, with superior stability on weak models and remarkable cross-domain generalization, highlighting its potential for broader tasks.

Jinyang Wu, Chonghua Liao, Mingkuan Feng, Shuai Zhang, Zhengqi Wen, Haoran Luo, Ling Yang, Huazhe Xu, Jianhua Tao• 2025

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH 500
Accuracy (Acc)83.4
543
Mathematical ReasoningAIME 2024
Accuracy33.3
479
Mathematical ReasoningMathVista
Accuracy70.2
382
Mathematical ReasoningAMC
Accuracy (%)77.5
368
Mathematical ReasoningMinerva
Pass@1 Accuracy38.2
289
Visual PerceptionBLINK
Accuracy48.3
241
Mathematical ReasoningMathVerse
Accuracy36.4
183
Mathematical ReasoningOlympiad
Accuracy0.462
134
Language UnderstandingMMLU-Pro
Accuracy49.5
116
Mathematical ReasoningMathVision
Accuracy34.6
66
Showing 10 of 13 rows

Other info

Follow for update