arXivDaily arXiv每日学术速递 周一至周五更新

大厂专区

ByteDance

2026-07-16 至 2026-07-16 收录 2
2607.13124 2026-07-16 cs.LG cs.AI cs.CL 新提交

ShortOPD: Recovering Pruned LLMs with Short-to-Long On-Policy Distillation

ShortOPD:通过短到长的策略蒸馏恢复剪枝后的语言模型

Qingyu Zhang, Qianhao Yuan, Hongyu Lin, Yaojie Lu, Xianpei Han, Le Sun, Xiang Li, Ming Xu, Jiarui Li, Xiuyin Zhao

发表机构 * ByteDance(字节跳动) Institute of Software, Chinese Academy of Sciences(中国科学院软件研究所) University of Chinese Academy of Sciences(中国科学院大学)

AI总结 研究结构化剪枝在语言模型自由形式生成任务中存在的问题,提出ShortOPD方法,通过短到长的策略蒸馏,检测重复后缀,合理分配展开预算,有效提升压缩模型分数,减少训练时间和展开令牌,推动结构化剪枝接近可部署的生成质量。

详情
AI中文摘要

结构化剪枝是一种对硬件友好的语言模型压缩方式,但大多在多项选择识别任务中得到验证,相同的压缩检查点在实际部署所需的自由形式生成任务中可能会崩溃。本文通过两项观察发现了这种差距。首先,贪心的\textsc{pass}@$1$在压缩后几乎消失,但\textsc{pass}@$k$在重复采样下能大幅恢复。其次,可恢复机制主要因后缀重复而失败。因此,恢复应在压缩模型自身的策略状态上进行密集的令牌级监督训练,策略蒸馏(OPD)通过将预压缩模型用作冻结教师来提供这种监督。然而,长时间的策略展开会将早期恢复预算花费在低信息重复后缀上,延迟损失下降。为缓解这种浪费,本文提出了\textbf{\shortopd},一种短到长的OPD调度,它能检测教师确认的重复后缀,将幸存的前缀视为每次展开的有效长度,并将未来的展开预算分配给策略当前可使用的有效长度。在数学、代码和开放式生成任务中,\shortopd\将压缩模型的分数提高到未恢复值的约$9$倍,以及标准恢复方法(无知识蒸馏的监督微调、知识蒸馏和序列知识蒸馏)的$1.6$ - $4.4$倍,并且在两点内匹配固定的$8192$令牌展开范围,使用四分之一的训练时间($8.5$小时对$35.9$小时)和减少$71\%$的展开令牌。希望该方法有助于使结构化剪枝超越在困惑度和多项选择基准上的微小收益,更接近可部署的生成质量。

英文摘要

Structured pruning is a hardware-friendly way to compress LLMs, but it is mostly validated on multiple-choice recognition tasks, while the same compressed checkpoints can collapse on the free-form generation that deployment actually requires. Two observations trace this gap. First, greedy \textsc{pass}@$1$ nearly vanishes after compression, yet \textsc{pass}@$k$ recovers substantially under repeated sampling: useful generations are demoted, not erased. Second, the recoverable regime fails mainly through suffix repetition. Recovery should therefore train on the compressed model's own on-policy states with dense token-level supervision, which On-Policy Distillation (OPD) provides by reusing the pre-compression model as a frozen teacher. However, long on-policy rollouts spend early recovery budget on low-information repetitive suffixes, delaying loss descent. To mitigate this waste, we propose \textbf{\shortopd}, a short-to-long OPD schedule that detects teacher-confirmed repetitive suffixes, treats the surviving prefix as each rollout's effective length, and allocates future rollout budgets to the effective lengths the policy can currently use. Across math, code, and open-ended generation, \shortopd\ raises the compressed model's score to about $9\times$ its unrecovered value and $1.6$--$4.4\times$ standard recovery recipes (SFT w/o KD, KD, and SeqKD), and it matches a fixed $8192$-token rollout horizon within two points using a quarter of the training time ($8.5$ vs.\ $35.9$ hours) and $71\%$ fewer rollout tokens. We hope this recipe helps move structured pruning beyond marginal gains on perplexity and multiple-choice benchmarks, a step closer to deployment-ready generation quality.

URL PDF HTML
2603.00546 2026-07-16 cs.AI cs.CV 版本更新

Advancing Multimodal Judge Models through a Capability-Oriented Benchmark and MCTS-Driven Data Generation

通过能力导向的基准和MCTS驱动的数据生成推进多模态评判模型

Zeyu Chen, Huanjin Yao, Ziwang Zhao, Min Yang

发表机构 * Tsinghua University(清华大学) ByteDance(字节跳动)

AI总结 本文提出M-JudgeBench和Judge-MCTS框架,通过能力导向的基准和MCTS驱动的数据生成提升多模态评判模型的评估能力。

详情
AI中文摘要

利用多模态大语言模型(MLLMs)作为评判者以实现精确且一致的评估,已在各种领域逐渐成为一种新兴范式。评估MLLM作为评判系统的能力和可靠性对于确保可信的评估至关重要。现有的评判基准按任务类型对样本进行分类,但未能捕捉到可靠评估所需的基本判断能力。在本工作中,我们引入M-JudgeBench,一个十维的能力导向基准,旨在全面评估MLLM的判断能力。我们的基准将评估分解为成对的链式思维(CoT)比较、长度偏差避免和过程错误检测任务,共同涵盖十个细粒度子任务。这种设计使能够诊断模型在推理风格、响应长度和跨模型变化方面的可靠性。系统性评估揭示了现有MLLM作为评判系统中的系统性弱点。为了解决这个问题,我们进一步提出Judge-MCTS,一个数据构建框架,生成具有各种正确性和长度的成对推理轨迹。使用Judge-MCTS,我们构建了一个MCTS增强的数据集并训练了M-Judger,一系列强大的评判模型。广泛的实验表明,M-Judger在现有评判基准以及M-JudgeBench上都表现出优越性。总体而言,我们的工作通过M-JudgeBench和Judge-MCTS框架建立了更系统的基础来评估MLLM作为评判者,为未来评判模型评估和能力驱动的评判训练铺平了道路。

英文摘要

Using Multimodal Large Language Models (MLLMs) as judges to achieve precise and consistent evaluations has gradually become an emerging paradigm across various domains. Evaluating the capability and reliability of MLLM-as-a-judge systems is therefore essential for ensuring trustworthy assessment. Existing judge benchmarks categorize samples by task types but fail to capture the fundamental judgment capabilities required for reliable evaluation. In this work, we introduce M-JudgeBench, a ten-dimensional capability-oriented benchmark designed to comprehensively assess the judgment abilities of MLLMs. Our benchmark decomposes evaluation into pairwise Chain-of-Thought (CoT) comparison, length bias avoidance, and process error detection tasks, jointly covering ten fine-grained subtasks. This design enables diagnosis of model reliability across reasoning styles, response lengths, and cross-model variations. Systematic evaluation uncovers the systematic weaknesses in existing MLLM-as-a-judge systems. To address this issue, we further propose Judge-MCTS, a data construction framework generating pairwise reasoning trajectories with various correctness and length. Using Judge-MCTS, we construct an MCTS-augmented dataset and train M-Judger, a series of strong judge models. Extensive experiments demonstrate the superiority of M-Judger on existing judge benchmarks as well as M-JudgeBench. Overall, our work establishes a more principled foundation for evaluating MLLM-as-a-judge through M-JudgeBench and Judge-MCTS framework, paving the way for future research on judge model evaluation and capability-driven judge training.

URL PDF HTML