KETCHUP: K-Step Return Estimation for Sequential Knowledge Distillation

Jiabin Fan, Guoqing Luo, Michael Bowling, Lili Mou


Abstract
We propose a novel K-step return estimation method (called KETCHUP) for Reinforcement Learning (RL)-based knowledge distillation (KD) in text generation tasks. Our idea is to induce a K-step return by using the Bellman Optimality Equation for multiple steps. Theoretical analysis shows that this K-step formulation reduces the variance of the gradient estimates, thus leading to improved RL optimization, especially when the student model size is large. Empirical evaluation on three text generation tasks demonstrates that our approach yields superior performance in both standard task metrics and large language model (LLM)-based evaluation. These results suggest that our K-step return induction offers a promising direction for enhancing RL-based KD in LLM research.
Anthology ID:
2026.findings-eacl.39
Volume:
Findings of the Association for Computational Linguistics: EACL 2026
Month:
March
Year:
2026
Address:
Rabat, Morocco
Editors:
Vera Demberg, Kentaro Inui, Lluís Marquez
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
778–796
Language:
URL:
https://preview.aclanthology.org/ingest-eacl/2026.findings-eacl.39/
DOI:
Bibkey:
Cite (ACL):
Jiabin Fan, Guoqing Luo, Michael Bowling, and Lili Mou. 2026. KETCHUP: K-Step Return Estimation for Sequential Knowledge Distillation. In Findings of the Association for Computational Linguistics: EACL 2026, pages 778–796, Rabat, Morocco. Association for Computational Linguistics.
Cite (Informal):
KETCHUP: K-Step Return Estimation for Sequential Knowledge Distillation (Fan et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-eacl/2026.findings-eacl.39.pdf
Checklist:
 2026.findings-eacl.39.checklist.pdf