让小模型在极端压缩下超越大模型,通过对抗性去相关提升性能。
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。