arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.18565cs.CV

用统一蒸馏协调神经网络中的知识迁移

Harmonizing knowledge Transfer in Neural Network with Unified Distillation

School of Computer Science, East China Normal University(华东师范大学计算机科学与技术学院)

Yaomin Huang, Zaomin Yan, Chaomin Shen, Faming Fang, Guixu Zhang

AI总结:

本文提出统一蒸馏框架,将中间层特征聚合为综合表示并预测分布参数,通过统一分布约束实现不同阶段的知识蒸馏,提升知识迁移的全面性与连贯性。

摘要:

知识蒸馏(KD)因其能够在不改变网络架构的情况下,将知识从庞大网络(教师)迁移到轻量网络(学生)而受到越来越多的关注。KD方法主要分为两类:基于特征的,侧重于中间层的特征;基于logits的,针对最终层的logits。本文通过在一个统一的KD框架内利用多样化的知识来源,引入了一种新颖的视角。具体来说,我们将中间层的特征聚合成一个综合表示,有效地收集来自不同阶段和尺度的语义信息。随后,我们从该表示中预测分布参数。这些步骤将中间层的知识转化为相应的分布形式,从而允许在网络的不同阶段通过统一的分布约束进行知识蒸馏,确保知识迁移的全面性和连贯性。进行了大量实验以验证所提出方法的有效性。

英文摘要:

Knowledge distillation (KD), known for its ability to transfer knowledge from a cumbersome network (teacher) to a lightweight one (student) without altering the architecture, has been garnering increasing attention. Two primary categories emerge within KD methods: feature-based, focusing on intermediate layers' features, and logits-based, targeting the final layer's logits. This paper introduces a novel perspective by leveraging diverse knowledge sources within a unified KD framework. Specifically, we aggregate features from intermediate layers into a comprehensive representation, effectively gathering semantic information from different stages and scales. Subsequently, we predict the distribution parameters from this representation. These steps transform knowledge from the intermediate layers into corresponding distributive forms, thereby allowing for knowledge distillation through a unified distribution constraint at different stages of the network, ensuring the comprehensiveness and coherence of knowledge transfer. Numerous experiments were conducted to validate the effectiveness of the proposed method.

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

↑