No Task Vector Is an Island: A Comprehensive Study on the Composability of Task Vectors from On-Policy Distillation
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
Task vectors provide a simple mechanism for composing learned capabilities through model merging. However, the composability of task vectors produced by on-policy distillation (OPD) remains largely unexplored. OPD trains a student using teacher feedback on student-generated trajectories, yielding parameter updates that differ from those produced by the teacher model, usually by reinforcement learning (RL). We therefore ask whether OPD task vectors can complement their RL teacher updates and compose effectively across tasks. Across five domains and two model architectures, we find evidence for both forms of composability. Within a task, merging OPD and RL task vectors can outperform both constituent models, even when the OPD student is weaker than its RL teacher. Across tasks, OPD task-vector compositions achieve higher average scores than corresponding RL compositions in seven of eight backbone-merging-rule comparisons. Parameter-space analyses reveal substantial non-collinearity between OPD and RL updates. Experiment in CODE domain on SMOLLM3-3B shows that the combined direction outperforms either constituent direction at the tested global update norm, supporting directional complementarity in this configuration. Across tasks, OPD updates also show lower overlap among the top-10% feed-forward channels ranked by update energy. Together, these results show that weaker standalone performance does not imply weaker task-vector composability. OPD task vectors can complement stronger RL teacher updates and combine effectively across tasks, highlighting composability as a distinct property for understanding and evaluating post-training updates.
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 26 of its 43 references that are in the map. First posted on arXiv 30 Sep 2026.
Builds on
26 of 43 references.
Closest ancestors · 6
Same direction, newest first
- 2026Teacher Should Think Ahead: Adaptive Continuations for Reliable On-Policy Distillation6 Sep
- 2026Open-MOPD: Diagnosing and Fixing Capability Imbalance in Multi-Teacher On-Policy Distillation19 Aug
- 2026MOPD: Multi-Teacher On-Policy Distillation for Capability Integration in LLM Post-Training29 Jun
- 2026Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillation11 Jun
- 2026On the Geometry of On-Policy Distillation5 Jun
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
Other references · 20
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,228
- 2022Editing Models with Task Arithmeticcited by 1,120
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 221
8 more
- 2025Generalizing Verifiable Instruction Followingcited by 194
- 2024Task Singular Vectors: Reducing Task Interference in Model Mergingcited by 147
- 2025OpenCodeReasoning: Advancing Data Distillation for Competitive Codingcited by 113
- 2026Consolidating RLVR Capabilities Across Domains: A Deep Dive into Fusion Paradigmscited by 4
- 2026Sparsity Curse: Understanding RLVR Model Parameter Space from Model Mergingcited by 2
- 2026Enough is as good as a feast: A Comprehensive Analysis of How Reinforcement Learning Mitigates Task Conflicts in LLMscited by 2
- 2026Decouple before Integration: Test-time Synthesis of SFT and RLVR Task Vectorscited by 1
- 2026CAMFT: Conflict-Aware Mergeable Fine-Tuning for Large Language Modelscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Reward-Aligned Reweighting for On-Policy Distillation
- 2026Demystifying On-Policy Distillation: Roles, Pathologies, and Regulations
- 2026From Gradients to Capabilities: Understanding Multi-Teacher On-Policy Distillation
- 2026Demystifying On-Policy Distillation: Where Does the Teacher's Advantage Live?
- 2026Understanding the Synergy between SFT, RLVR, and OPD in LLM Post-Training
- 2026RL Starts before RL: On Policy Distillation for Better Reinforcement Learning
- 2026PMOPD: Task Ordering, Cycling, and Parameter-Update Subspace Protection in Multi-Teacher On-Policy Distillation
- 2026DuoOPD: Learning from Joint Teacher-Student Outcomes for Multi-Task On-Policy Distillation