arXiv:2411.01547cs.LGcs.CV2024-11被引 11

通过分块对齐日志,让小模型更精准模仿大模型的特征表现。

Decoupling Dark Knowledge via Block-wise Logit Distillation for Feature-level Alignment

  • 用分块方式逐步替换教师模型,实现隐式特征对齐。
  • 在多个数据集上达到或超越当前最优知识蒸馏效果。
  • 适合想提升轻量化模型性能的研究者参考。

知识蒸馏(KD)通过大模型(教师)指导小模型(学生)学习,以生成高性能的轻量级模型。尽管基于中间特征的蒸馏方法曾超越早期基于输出日志的方法,但近期研究重新发现了日志蒸馏的潜力。本文提出一种分块日志蒸馏框架,通过逐步用教师模型的子模块作为中间桥梁,实现隐式的特征对齐,从而统一了日志与特征蒸馏的视角。该方法在多个基准数据集上取得可比或更优的结果,验证了结合日志与特征蒸馏的巨大潜力,为未来研究提供了新思路。

原文摘要 · Abstract (English)

Knowledge Distillation (KD), a learning manner with a larger teacher network guiding a smaller student network, transfers dark knowledge from the teacher to the student via logits or intermediate features, with the aim of producing a well-performed lightweight model. Notably, many subsequent feature-based KD methods outperformed the earliest logit-based KD method and iteratively generated numerous state-of-the-art distillation methods. Nevertheless, recent work has uncovered the potential of the logit-based method, bringing the simple KD form based on logits back into the limelight. Features or logits? They partially implement the KD with entirely distinct perspectives; therefore, choosing between logits and features is not straightforward. This paper provides a unified perspective of feature alignment in order to obtain a better comprehension of their fundamental distinction. Inheriting the design philosophy and insights of feature-based and logit-based methods, we introduce a block-wise logit distillation framework to apply implicit logit-based feature alignment by gradually replacing teacher's blocks as intermediate stepping-stone models to bridge the gap between the student and the teacher. Our method obtains comparable or superior results to state-of-the-art distillation methods. This paper demonstrates the great potential of combining logit and features, and we hope it will inspire future research to revisit KD from a higher vantage point.

知识蒸馏特征对齐轻量化模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。