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

面向不确定性的多模态大语言模型知识蒸馏

Uncertainty-Aware Knowledge Distillation for Multimodal Large Language Models

NEC Laboratories America, Inc.(NEC美洲实验室);University at Buffalo, SUNY(布法罗大学);Rutgers University(罗格斯大学)

Jingchen Sun, Shaobo Han, Deep Patel, Wataru Kohno, Can Jin, Changyou Chen

AI总结:

本文提出Beta-KD框架,通过统一贝叶斯视角平衡数据与教师指导,提升多模态VQA任务性能。

摘要:

知识蒸馏建立了一种结合数据监督和教师指导的学习范式。然而,确定数据学习与教师指导之间的最佳平衡具有挑战性,因为某些样本可能噪声较大而其他样本受教师不确定性影响。为此,本文提出Beta-weighted Knowledge Distillation(Beta-KD),一种面向不确定性的蒸馏框架,能够自适应调节学生对教师指导的依赖程度。具体而言,我们从统一的贝叶斯视角出发,将教师指导解释为学生激活的Gibbs先验。这产生了一个闭式、面向不确定性的加权机制,并支持任意蒸馏目标及其组合。在多模态VQA基准上的大量实验表明,从大教师VLM蒸馏学生视觉-语言模型能一致提升性能。结果表明,Beta-KD优于现有知识蒸馏方法。代码可在https://github.com/Jingchensun/beta-kd获取。

英文摘要:

Knowledge distillation establishes a learning paradigm that leverages both data supervision and teacher guidance. However, determining the optimal balance between learning from data and learning from the teacher is challenging, as some samples may be noisy while others are subject to teacher uncertainty. This motivates the need for adaptively balancing data and teacher supervision. We propose Beta-weighted Knowledge Distillation (Beta-KD), an uncertainty-aware distillation framework that adaptively modulates how much the student relies on teacher guidance. Specifically, we formulate teacher--student learning from a unified Bayesian perspective and interpret teacher supervision as a Gibbs prior over student activations. This yields a closed-form, uncertainty-aware weighting mechanism and supports arbitrary distillation objectives and their combinations. Extensive experiments on multimodal VQA benchmarks demonstrate that distilling student Vision-Language Models from a large teacher VLM consistently improves performance. The results show that Beta-KD outperforms existing knowledge distillation methods. The code is available at https://github.com/Jingchensun/beta-kd.

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

↑