arXiv:2412.08939cs.CV2024-12AAAI被引 12

动态调整知识蒸馏,让小模型更好学大模型的修复技巧。

Dynamic Contrastive Knowledge Distillation for Efficient Image Restoration

  • 根据学生模型学习状态动态调整知识传递方式
  • 在多种图像修复任务中超越现有蒸馏方法性能
  • 适配不同网络结构,可与优化约束方法结合使用

知识蒸馏(KD)是一种通过高性能但复杂的教师模型来提升紧凑型学生网络能力的有效方法。然而,现有的图像修复领域知识蒸馏方法忽视了学生模型的学习状态,采用固定解空间,限制了蒸馏效果。此外,仅依赖L1类损失难以利用图像的分布信息。本文提出一种新型动态对比知识蒸馏(DCKD)框架用于图像修复:引入动态对比正则化以感知学生学习状态,并通过对比学习动态调整蒸馏解空间;同时设计分布映射模块,提取并对齐教师与学生模型在像素级类别分布上的特征。所提DCKD为无结构依赖的蒸馏框架,可适配不同主干网络,并可与优化上界约束的方法结合以进一步提升性能。大量实验证明,DCKD在多种图像修复任务和网络结构下显著优于当前最优蒸馏方法。

原文摘要 · Abstract (English)

Knowledge distillation (KD) is a valuable yet challenging approach that enhances a compact student network by learning from a high-performance but cumbersome teacher model. However, previous KD methods for image restoration overlook the state of the student during the distillation, adopting a fixed solution space that limits the capability of KD. Additionally, relying solely on L1-type loss struggles to leverage the distribution information of images. In this work, we propose a novel dynamic contrastive knowledge distillation (DCKD) framework for image restoration. Specifically, we introduce dynamic contrastive regularization to perceive the student's learning state and dynamically adjust the distilled solution space using contrastive learning. Additionally, we also propose a distribution mapping module to extract and align the pixel-level category distribution of the teacher and student models. Note that the proposed DCKD is a structure-agnostic distillation framework, which can adapt to different backbones and can be combined with methods that optimize upper-bound constraints to further enhance model performance. Extensive experiments demonstrate that DCKD significantly outperforms the state-of-the-art KD methods across various image restoration tasks and backbones.

知识蒸馏图像修复对比学习

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