解决跨架构知识蒸馏的训练不稳问题,提升模型迁移效果。
Heterogeneous Knowledge Distillation via Geometry Decoupling and Momentum-Aware Gradient Regulation

- 分离特征幅度与方向,稳定语义对齐过程
- 动态调整梯度冲突,实现高效收敛
- 轻量设计适合实际部署,性能领先
异构知识蒸馏(HKD)旨在将知识从不同架构间迁移(如从Transformer到CNN),但存在严重的训练不稳定性。我们发现这种不稳定性源于两个高度耦合的问题:显著的特征范数差异导致优化阻力,以及主任务与蒸馏目标间因不同归纳偏置引发的严重梯度冲突。为此,我们提出SPOFA框架,基于特征与梯度双重稳定机制。在特征层面,引入基于层归一化的解耦投影器,显式分离特征幅度与方向,构建有限且稳定的语义对齐空间;在梯度层面,提出受动量驱动的指数移动平均(MEMA)动态缩放器,通过建立优化轨迹的历史基准,主动评估瞬时梯度冲突并自适应惩罚有害蒸馏信号,确保稳定收敛。重要的是,SPOFA以极轻量参数实现双层稳定。在两个主流基准上的大量实验表明,SPOFA达到当前最优精度,在显著优于计算开销较大的方法的同时,相比标准基线仅引入极小额外计算开销。
原文摘要 · Abstract (English)
Heterogeneous Knowledge Distillation (HKD) aims to transfer knowledge across varying architectures (e.g., from Transformer to CNN) but inherently suffers from severe training instability. We reveal that this instability stems from two highly coupled challenges: massive feature norm discrepancies that cause optimization drag, and severe gradient conflicts between the primary and distillation objectives arising from distinct inductive biases. To achieve stable distillation, we propose SPOFA, a framework built upon a novel Feature and Gradient Dual Stabilization mechanism. Specifically, at the feature level, we introduce a LayerNorm-based decoupling projector that explicitly decouples feature magnitude from direction, creating a bounded and stable space for semantic alignment. At the gradient level, we propose a momentum-driven Exponential Moving Average (MEMA) dynamic scaler. By establishing a robust historical baseline of the optimization trajectory, MEMA actively evaluates instantaneous gradient conflicts and adaptively penalizes harmful distillation signals, guaranteeing stable convergence. Importantly, SPOFA achieves this dual stabilization with an extremely lightweight parameter footprint. Extensive experiments on two mainstream benchmarks demonstrate that SPOFA achieves state-of-the-art accuracy, significantly outperforming computationally expensive methods while introducing only minimal computational overhead compared to standard baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。