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

Distill Where the Student Goes: Teacher-Regularized RL for English-Evidence Cross-Lingual RAG

About

Cross-lingual retrieval-augmented generation (RAG) is often deployed in an English-evidence regime, where users query in diverse languages but retrieved passages remain English. In this setting, generation can fail despite strong base models: English evidence induces language drift (English or code-switching outputs) and models use evidence unreliably when producing non-English answers. We attribute these failures to two post-training challenges: (i) errors are prefix-dependent, so fixed-trajectory supervision suffers from prefix mismatch; and (ii) sequence-level (partly discrete / judge-based) rewards yield noisy credit assignment and high-variance updates. We propose TR-RAG, a teacher-regularized RL recipe that couples reward optimization with on-policy distillation on student-visited prefixes. A compact student samples on-policy answers, while a stronger frozen teacher is queried only on those prefixes and provides a prefix-wise student-to-teacher reverse-KL anchor. We further introduce a reward decomposition for English-evidence multilingual generation, combining language consistency, character 3-gram recall, and an LLM-judge score for evidence-grounded correctness. Across three benchmarks (BioASQ-ENKB5, Hotpot-ENKB5, and naturally multilingual MKQA) and two backbones, TR-RAG improves the composite of language adherence and evidence-grounded correctness over strong baselines. Crucially, the teacher anchor acts as a safety net: on in-domain languages it prevents the large language-consistency collapses (up to ~27 percentage points) that reward-only RL can suffer by drifting below even the base model, while on distant out-of-distribution languages, where reward-only RL stalls at the base model's ceiling, it still improves evidence grounding; and on character 3-gram recall the compact student sometimes surpasses its 70B teacher.

Haotian Zhou, Weiran Huang, Siqi Liu, Xiting Wang, Xin Zhang, Zhihao Wen• 2026

Related benchmarks

TaskDatasetResultRank
Retrieval-Augmented GenerationHotpot ENKB5 (In-Domain Average)
Language Consistency86.51
15
Cross-lingual Retrieval-Augmented GenerationBioASQ-ENKB5
ID Score90.99
14
Retrieval-Augmented GenerationHotpot-ENKB5 Out-of-Domain Average
Language Consistency96.41
13
Retrieval-Augmented GenerationHotpot-ENKB5 v1 (Indonesian)
Language Consistency55.6
11
Cross-lingual Retrieval-Augmented GenerationHotpot-ENKB5 (all)
Language Consistency (ALL-AVG)86.37
11
Multilingual Question AnsweringMKQA
Composite Score (ALL-AVG)57.82
6
Multilingual Question AnsweringMKQA Norwegian (no)
LC Score9
3
Retrieval-Augmented GenerationHotpot-ENKB5 (Korean)
Language Consistency99.73
3
Multilingual Question AnsweringMKQA Khmer (km)
LC18
3
Showing 9 of 9 rows

Other info

Follow for update