Huazhe Xu
2026
TemplateRL: Structured Template-Guided Reinforcement Learning for LLM Reasoning
Jinyang Wu | Chonghua Liao | Mingkuan Feng | Shuai Zhang | Zhengqi Wen | Haoran Luo | Ling Yang | Huazhe Xu | Jianhua Tao
Findings of the Association for Computational Linguistics: ACL 2026
Jinyang Wu | Chonghua Liao | Mingkuan Feng | Shuai Zhang | Zhengqi Wen | Haoran Luo | Ling Yang | Huazhe Xu | Jianhua Tao
Findings of the Association for Computational Linguistics: ACL 2026
Reinforcement learning (RL) has emerged as an effective paradigm for enhancing model reasoning. However, existing RL methods like GRPO often rely on unstructured self-sampling to fit scalar rewards, often producing inefficient rollouts that fail to capture transferable problem-solving strategies. To address these limitations, 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.
2025
World Models with Hints of Large Language Models for Goal Achieving
Zeyuan Liu | Ziyu Huan | Xiyao Wang | Jiafei Lyu | Jian Tao | Xiu Li | Furong Huang | Huazhe Xu
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)
Zeyuan Liu | Ziyu Huan | Xiyao Wang | Jiafei Lyu | Jian Tao | Xiu Li | Furong Huang | Huazhe Xu
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)
Reinforcement learning struggles in the face of long-horizon tasks and sparse goals due to the difficulty in manual reward specification. While existing methods address this by adding intrinsic rewards, they may fail to provide meaningful guidance in long-horizon decision-making tasks with large state and action spaces, lacking purposeful exploration. Inspired by human cognition, we propose a new multi-modal model-based RL approach named Dreaming with Large Language Models (DLLM). DLLM integrates the proposed hinting subgoals from the LLMs into the model rollouts to encourage goal discovery and reaching in challenging tasks. By assigning higher intrinsic rewards to samples that align with the hints outlined by the language model during model rollouts, DLLM guides the agent toward meaningful and efficient exploration. Extensive experiments demonstrate that the DLLM outperforms recent methods in various challenging, sparse-reward environments such as HomeGrid, Crafter, and Minecraft by 41.8%, 21.1%, and 9.9%, respectively.