Trapped by Their Own Rollouts: Understanding Aggregation--Rollout Feedback in Federated On-Policy Distillation
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
On-policy distillation (OPD) is a promising approach to language-model adaptation, aligning teacher supervision with the student's own generated trajectories. When adaptation prompts are distributed across clients, can this process benefit from federated collaboration? We study federated OPD and find that substantial collaboration gains can be obscured by learning-rate sensitivity: FedAvg can perform no better than independent local training at a small learning rate, yet recover a clear advantage at a larger rate. We explain this phenomenon through the student's dual role as learner and generator of future training data. An aggregation-induced optimization lag can delay access to useful teacher supervision, which in turn slows subsequent learning. Our theory establishes this aggregation--rollout feedback in a solvable model with a common optimum and stable updates, and identifies two coupled roles of learning rate: learning from current supervision and reaching future supervision. Guided by this analysis, we propose FedTOPS (Federated Teacher-guided On-Policy Scaling), which reuses teacher feedback on current trajectories to adapt the FedAvg update magnitude under clientwise predictive-change constraints. Across six mathematical reasoning benchmarks, FedTOPS improves macro Avg@8 over FedAvg by 4.56--14.57 percentage points across the evaluated student models and local learning rates.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 19 of its 23 references that are in the map. First posted on arXiv 26 Sep 2026.
Builds on
19 of 23 references.
Closest ancestors · 5
Same direction, newest first
- 2026What Matters in On-Policy Distillation? A Perspective on Data Efficiency and Data Selection4 Sep
- 2026Rethinking On-Policy Distillation of Large Language Models II: One Training Example3 Sep
- 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation12 May
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
- 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar
Other references · 14
- 2025Qwen3 Technical Reportcited by 8,820
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2015Trust Region Policy Optimizationcited by 2,580
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2020Performative Predictioncited by 327
- 2024Improving LoRA in Privacy-preserving Federated Learningcited by 161
- 2024FLoRA: Federated Fine-Tuning Large Language Models with Heterogeneous Low-Rank Adaptationscited by 146
- 2023FedExP: Speeding Up Federated Averaging via Extrapolationcited by 69
- 2024Finite-Time Analysis of On-Policy Heterogeneous Federated Reinforcement Learningcited by 31
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Adaptive FastOPD: Progress-Aware Rollout Horizon Expansion for Efficient On-Policy Distillation
- 2026REVO: Rollout-Efficient Off-Policy Distillation via Variance-Guided Reuse
- 2026Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillation
- 2026Adaptive Supervised Anchoring for On-Policy Self-Distillation
- 2026AsyncOPD: How Stale Can On-Policy Distillation Be?
- 2026Rethinking Self-Distillation for Multi-Teacher Capability Merging
- 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts
- 2026boldsymbolf-OPD: Stabilizing Long-Horizon On-Policy Distillation with Freshness-Aware Control