arXiv:2603.05421cs.CVcs.AI2026-03

让小模型在极端压缩下超越大模型,通过对抗性去相关提升性能。

DARK: Diagonal-Anchored Repulsive Knowledge Distillation for Vision-Language Models under Extreme Compression

  • 用对角锚定+反向权重优化知识蒸馏,避免盲目模仿教师。
  • 压缩26倍后的小模型在3个零样本任务上超过教师,最高准确率提升5.1%。
  • 适合医疗视觉语言模型的轻量化部署,尤其在设备端运行场景。

为临床场景中的设备端部署,压缩视觉语言模型日益重要,但当教师-学生模型容量差距超过一个数量级时,传统知识蒸馏性能急剧下降。我们认为,在此类情况下,严格模仿教师并非最优目标:教师的成对相似性结构很大程度上反映其自身架构偏好,而非紧凑学生能高效表示的信息。本文提出一种对比式知识蒸馏框架DARK(Diagonal-Anchored Repulsive Knowledge Distillation),将蒸馏损失分解为对角项(匹配图像-文本对)与非对角项(非目标相似性)。对角项始终锚定匹配对的一致性;非对角项从正权重逐步退火为负权重,使学生从模仿转向“排斥”教师的非目标相似结构。我们以427M参数的胎儿超声视觉语言模型FetalCLIP为教师,蒸馏出75M参数的学生模型MobileFetalCLIP,其视觉编码器缩小26倍,在iPhone 16 Pro上仅需1.6毫秒。该学生模型在三个零样本基准上达到或超过教师表现,包括HC18生物测量有效性(88.6% vs. 83.5%)和脑部子平面F1值(0.784 vs. 0.702)。嵌入几何与logit分析表明,DARK实现“结构化去相关”:学生保留教师级别的每图置信度,同时脱离继承的类别混淆,表明在极端压缩下,可控排斥比单纯模仿更高效。

原文摘要 · Abstract (English)

Compressing vision-language models for on-device deployment is increasingly important in clinical settings, but knowledge distillation (KD) degrades sharply when the teacher-student capacity gap spans an order of magnitude or more. We argue that, under such gaps, strict imitation of the teacher is a poor objective: much of the teacher's pairwise similarity structure reflects its own architectural biases rather than information a compact student can efficiently represent. We propose \textbf{Diagonal-Anchored Repulsive Knowledge Distillation (DARK)}, a contrastive KD framework that decomposes the distillation loss into a diagonal term (matched image-text pairs) and an off-diagonal term (non-target similarities). The diagonal term anchors matched-pair alignment throughout training; the off-diagonal term is annealed from positive to negative weighting, transitioning the student from imitating to \emph{repelling} the teacher's non-target similarity structure. We instantiate DARK by distilling FetalCLIP, a 427M-parameter fetal ultrasound vision-language model, into \textbf{MobileFetalCLIP}, a 75M-parameter student model with a $26\times$ smaller visual encoder, running in 1.6\,ms on an iPhone~16~Pro. The student matches or exceeds its teacher on three zero-shot benchmarks, including HC18 biometry validity (88.6\% vs.\ 83.5\%) and brain sub-plane F1 (0.784 vs.\ 0.702). Embedding-geometry and logit analyses show that DARK induces \emph{structured decorrelation}: the student preserves teacher-aligned per-image confidence while diverging from inherited inter-class confusion, suggesting that controlled repulsion can be more efficient than imitation under extreme compression.

知识蒸馏视觉语言模型模型压缩医疗AI

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