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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Retrieval-Augmented Generation | Hotpot ENKB5 (In-Domain Average) | Language Consistency86.51 | 15 | |
| Cross-lingual Retrieval-Augmented Generation | BioASQ-ENKB5 | ID Score90.99 | 14 | |
| Retrieval-Augmented Generation | Hotpot-ENKB5 Out-of-Domain Average | Language Consistency96.41 | 13 | |
| Retrieval-Augmented Generation | Hotpot-ENKB5 v1 (Indonesian) | Language Consistency55.6 | 11 | |
| Cross-lingual Retrieval-Augmented Generation | Hotpot-ENKB5 (all) | Language Consistency (ALL-AVG)86.37 | 11 | |
| Multilingual Question Answering | MKQA | Composite Score (ALL-AVG)57.82 | 6 | |
| Multilingual Question Answering | MKQA Norwegian (no) | LC Score9 | 3 | |
| Retrieval-Augmented Generation | Hotpot-ENKB5 (Korean) | Language Consistency99.73 | 3 | |
| Multilingual Question Answering | MKQA Khmer (km) | LC18 | 3 |