Search papers, labs, and topics across Lattice.
This paper introduces Multi-Agent Protocol Distillation (MAPD), a novel framework that combines knowledge distillation with reinforcement learning to enhance the performance of language models in knowledge-intensive tasks. By utilizing a structured, style-normalized protocol to bridge the distribution gap between proprietary and open-source models, MAPD effectively addresses the limitations of traditional distillation methods. The framework achieves significant improvements in performance across multiple QA benchmarks, with average success rates of 39.4% for Qwen3-1.7B and 44.4% for Qwen3-4B, while maintaining robustness against style drift and verbosity issues.
Bridging the gap between proprietary and open-source models, MAPD achieves up to 44.4% success in QA tasks by transforming sparse RL signals into dense distillation guidance.
Agentic search enables large language models to solve knowledge-intensive tasks by interleaving multi-step reasoning with retrieval, yet optimizing this with outcome-based reinforcement learning (RL) provides only sparse supervision. Knowledge distillation can supply denser guidance, and advanced proprietary models with their strong reasoning capabilities are promising teachers. While distilling from proprietary models can densify this supervisory signal, conventional logit-matching is precluded by hidden logits and mismatched tokenizers, whereas raw natural language trajectory imitation transfers superficial stylistic artifacts rather than core reasoning competence. To address the heterogeneous distillation problem and bridge the distribution gap, we propose Multi-Agent Protocol Distillation (MAPD), a joint distillation and RL framework uses a structured, style-normalized protocol as an intermediate representation. An offline multi-agent system (MAS) decomposes each query, retrieves supporting evidence, repairs failed searches, and converts the resulting exploration trace into a JSON protocol containing the task type, reasoning plan, and extractive grounding facts. During training, the protocol is provided only to a privileged branch of the student policy, whose token distributions furnish a dense distillation signal alongside the sparse RL objective. Extensive evaluations across seven QA benchmarks demonstrate that MAPD consistently outperforms competitive distillation and RL, achieving average success rates of 39.4\% on Qwen3-1.7B and 44.4\% on Qwen3-4B. Crucially, the framework generalizes robustly across diverse proprietary teachers while effectively mitigating the student policy from style drift and verbosity degeneration.
Tsinghua AI