mlsanityData to 14 Oct 2026

Trapped by Their Own Rollouts: Understanding Aggregation--Rollout Feedback in Federated On-Policy Distillation

arXiv 2609.32573 ↗First posted 26 Sep 2026cs.LG cs.DCcited by 0

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

  1. 2026What Matters in On-Policy Distillation? A Perspective on Data Efficiency and Data Selection4 Sep
  2. 2026Rethinking On-Policy Distillation of Large Language Models II: One Training Example3 Sep
  3. 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation12 May
  4. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
  5. 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar

Other references · 14

  1. 2025Qwen3 Technical Reportcited by 8,820
  2. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
  3. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  4. 2015Trust Region Policy Optimizationcited by 2,580
  5. 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
  6. 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
  7. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
  8. 2020Performative Predictioncited by 327
  9. 2024Improving LoRA in Privacy-preserving Federated Learningcited by 161
  10. 2024FLoRA: Federated Fine-Tuning Large Language Models with Heterogeneous Low-Rank Adaptationscited by 146
  11. 2023FedExP: Speeding Up Federated Averaging via Extrapolationcited by 69
  12. 2024Finite-Time Analysis of On-Policy Heterogeneous Federated Reinforcement Learningcited by 31
2 more
  1. 2023Performative Federated Learning: A Solution to Model-Dependent and Heterogeneous Distribution Shiftscited by 14
  2. 2025Convergence Guarantees for Federated SARSA with Local Training and Heterogeneous Agentscited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Adaptive FastOPD: Progress-Aware Rollout Horizon Expansion for Efficient On-Policy Distillation
  2. 2026REVO: Rollout-Efficient Off-Policy Distillation via Variance-Guided Reuse
  3. 2026Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillation
  4. 2026Adaptive Supervised Anchoring for On-Policy Self-Distillation
  5. 2026AsyncOPD: How Stale Can On-Policy Distillation Be?
  6. 2026Rethinking Self-Distillation for Multi-Teacher Capability Merging
  7. 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts
  8. 2026boldsymbolf-OPD: Stabilizing Long-Horizon On-Policy Distillation with Freshness-Aware Control