mlsanityData to 14 Oct 2026

Heterogeneous Knowledge Distillation via Geometry Decoupling and Momentum-Aware Gradient Regulation

arXiv 2606.24557 ↗First posted 23 Jun 2026cs.CVcited by 0

On the map: Knowledge distillation → General knowledge distillation methods

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in General knowledge distillation methods, one of 12 directions in Knowledge distillation. Builds on 28 of its 40 references that are in the map. First posted on arXiv 23 Jun 2026.

Builds on

28 of 40 references.

Closest ancestors · 8

Same direction, newest first

  1. 2025Perspective-Aware Teaching: Adapting Knowledge for Heterogeneous Distillation15 Jan
  2. 2024CKD: Contrastive Knowledge Distillation from A Sample-wise Perspective22 Apr
  3. 2024VkD: Improving Knowledge Distillation using Orthogonal Projections10 Mar
  4. 2023One-for-All: Bridge the Gap Between Heterogeneous Architectures in Knowledge Distillation30 Oct
  5. 2023Heterogeneous Generative Knowledge Distillation with Masked Image Modeling18 Sep
  6. 2022Knowledge Distillation from A Stronger Teacher21 May
  7. 2022Aligning Logits Generatively for Principled Black-Box Knowledge Distillation21 May
  8. 2022Decoupled Knowledge Distillation16 Mar

Other references · 20

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2022A ConvNet for the 2020scited by 3,579
  3. 2014FitNets: Hints for Thin Deep Netscited by 1,653
  4. 2020Gradient Surgery for Multi-Task Learningcited by 1,256
  5. 2021Swin Transformer V2: Scaling Up Capacity and Resolutioncited by 1,228
  6. 2016Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfercited by 1,077
  7. 2021Transformers in Vision: A Surveycited by 1,013
  8. 2019Relational Knowledge Distillationcited by 889
  9. 2019Contrastive Representation Distillationcited by 748
  10. 2021Do Vision Transformers See Like Convolutional Neural Networks?cited by 675
  11. 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 459
  12. 2021Data-Free Knowledge Distillation for Heterogeneous Federated Learningcited by 383
8 more
  1. 2023CrossKD: Cross-Head Knowledge Distillation for Object Detectioncited by 52
  2. 2025Learning to Unlearn while Retaining: Combating Gradient Conflicts in Machine Unlearningcited by 19
  3. 2024Style Blind Domain Generalized Semantic Segmentation via Covariance Alignment and Semantic Consistence Contrastive Learningcited by 18
  4. 2024Safe and Balanced: A Framework for Constrained Multi-Objective Reinforcement Learningcited by 9
  5. 2024A Teacher-Free Graph Knowledge Distillation Framework with Dual Self-Distillationcited by 8
  6. 2025The Unseen Bias: How Norm Discrepancy in Pre-Norm MLLMs Leads to Visual Information Losscited by 5
  7. 2025Reflectance Prediction-based Knowledge Distillation for Robust 3D Object Detection in Compressed Point Cloudscited by 3
  8. 2025Decoupled Doubly Contrastive Learning for Cross Domain Facial Action Unit Detectioncited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025DeepKD: A Deeply Decoupled and Denoised Knowledge Distillation Trainer
  2. 2025Multi-Level Decoupled Relational Distillation for Heterogeneous Architectures
  3. 2025Heterogeneous Complementary Distillation
  4. 2024Fuse Before Transfer: Knowledge Fusion for Heterogeneous Distillation
  5. 2025Distilling Knowledge from Heterogeneous Architectures for Semantic Segmentation
  6. 2025UHKD: A Unified Framework for Heterogeneous Knowledge Distillation via Frequency-Domain Representations
  7. 2026SFKD: Spatial--Frequency Joint-Aware Heterogeneous Knowledge Distillation via Multi-Level Wavelet Spectral Interaction
  8. 2025Contrastive Representation Distillation via Multi-Scale Feature Decoupling