Rewarding How Models Think Pedagogically: Integrating Pedagogical Reasoning and Thinking Rewards for LLMs in Education
Unggi Lee, Jiyeong Bae, Jaehyeon Park, Haeun Park, Taejun Park, Younghoon Jeon, Sungmin Cho, Junbo Koh, Yeil Jeong, Gyeonggeon Lee
TL;DR
This work reframes LLM tutoring as an optimization problem for both visible outputs and internal reasoning by introducing PedagogicalRL-Thinking, a framework that integrates Pedagogical Reasoning Prompting grounded in Polya's four-step method with Thinking Reward to train reasoning traces. The composite reward r = $r_{\text{sol}} + (r_{\text{ped}} - 1.0) \cdot \lambda_{\text{ped}} + (r_{\text{think}} - \theta) \cdot \lambda_{\text{think}}$ guides the tutor toward higher problem-solving impact, pedagogical quality, and thinking quality, with weights $\lambda_{\text{ped}}=0.75$, $\lambda_{\text{think}}=0.3$, and threshold $\theta=0.6$. Across five ablated conditions on BigMath with synthetic student simulations, thinking-enabled tutors dramatically improve $\Delta$Solve and Helpful rates, and Ped. Think Reward achieves the best overall performance by reducing leakage while maintaining or increasing pedagogical quality. JL RL-based training also yields out-of-distribution gains on WBEB, improving pedagogical knowledge, writing scores, and decision-making while largely preserving subject knowledge, supported by a comprehensive 82-code educational codebook and both quantitative and qualitative analyses. These findings suggest that shaping internal reasoning, not just outputs, is crucial for effective AI-driven education and may generalize to other domains beyond mathematics.
Abstract
Large language models (LLMs) are increasingly deployed as intelligent tutoring systems, yet research on optimizing LLMs specifically for educational contexts remains limited. Recent works have proposed reinforcement learning approaches for training LLM tutors, but these methods focus solely on optimizing visible responses while neglecting the model's internal thinking process. We introduce PedagogicalRL-Thinking, a framework that extends pedagogical alignment to reasoning LLMs in education through two novel approaches: (1) Pedagogical Reasoning Prompting, which guides internal reasoning using domain-specific educational theory rather than generic instructions; and (2) Thinking Reward, which explicitly evaluates and reinforces the pedagogical quality of the model's reasoning traces. Our experiments reveal that domain-specific, theory-grounded prompting outperforms generic prompting, and that Thinking Reward is most effective when combined with pedagogical prompting. Furthermore, models trained only on mathematics tutoring dialogues show improved performance on educational benchmarks not seen during training, while preserving the base model's factual knowledge. Our quantitative and qualitative analyses reveal that pedagogical thinking reward produces systematic reasoning trace changes, with increased pedagogical reasoning and more structured instructional decision-making in the tutor's thinking process.
