arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.14035cs.LGcs.AI

委员会智慧:来自大型基础模型和领域专家的多样化蒸馏

Wisdom of Committee: Diverse Distillation from Large Foundation Models and Domain Experts

Rice University(Rice大学);Google DeepMind(谷歌DeepMind);Google Inc(谷歌公司);University of California, Davis(加州大学戴维斯分校)

Zichang Liu, Qingyun Liu, Yuening Li, Liang Liu, Anshumali Shrivastava, Shuchao Bi, Lichan Hong, Ed H. Chi, Zhe Zhao

AI总结:

针对基础模型向紧凑领域模型蒸馏时能力、架构和模态差异大的问题,提出DiverseDistill框架,通过可学习的问答机制和对齐异构教师输出,在推荐和视觉任务上恢复73-114%的性能差距。

摘要:

从基础模型向紧凑领域模型进行知识蒸馏因能力、架构和模态的巨大差异而具有挑战性。例如,在我们的实验中,从7600万参数的语言模型蒸馏到200万参数的推荐模型仅能弥补未蒸馏学生与教师之间不到40%的性能差距。我们表明,引入与基础模型共享学生架构特征的领域专家作为多样化教师委员会,能显著改善迁移效果。然而,标准的多教师方法未能利用这种多样性:简单组合异构教师可能使性能低于单教师蒸馏。为此,我们提出DiverseDistill,一种交互式蒸馏框架,采用可学习的问答机制生成教师条件查询,并将异构教师输出对齐到学生的表示空间。与需要基于梯度的协同优化或修改教师架构的方法不同,DiverseDistill在冻结教师的情况下仅通过其中间层的前向推理运行:无需参数更新、无需协同训练、无需架构修改。动态教师重要性机制通过过滤每个样本中低相关性的教师(例如,在推荐任务中减少约30%的前向传播且无质量损失)进一步降低训练成本,而整个蒸馏模块在训练后被丢弃,推理时零开销。在推荐(38倍压缩)和视觉(3.6倍压缩)任务上的评估表明,DiverseDistill恢复了73-114%的师生性能差距,持续优于所有单教师和多教师基线方法。

英文摘要:

Knowledge distillation from foundation models to compact domain models is challenging due to substantial gaps in capacity, architecture, and modality. For example, in our experiments, distilling from a 76M-parameter language model to a 2M-parameter recommender closes less than 40% of the performance gap between the undistilled student and the teacher. We show that introducing domain-specific experts -- which share the student's architectural characteristics -- alongside the foundation model as a diverse teacher committee significantly improves transfer. However, standard multi-teacher methods fail to exploit this diversity: naively combining heterogeneous teachers can degrade performance below single-teacher distillation. To address this, we propose DiverseDistill, an interactive distillation framework that employs a learnable Question-Answer mechanism to generate teacher-conditioned queries and align heterogeneous teacher outputs into the student's representation space. Unlike methods requiring gradient-based co-optimization or architectural modification of teachers, DiverseDistill operates with frozen teachers using only forward-pass inference through their intermediate layers: no parameter updates, no co-training, and no architectural surgery. A dynamic teacher importance mechanism further reduces training cost by filtering low-relevance teachers per sample (e.g., ~30% fewer forward passes with no quality loss for recommendation tasks), while the entire Distillation Module is discarded after training, adding zero inference overhead. Evaluations on recommendation (38x compression) and vision (3.6x compression) tasks demonstrate that DiverseDistill recovers 73-114% of the teacher-student performance gap, consistently outperforming all single- and multi-teacher baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑