Sinkhorn蒸馏

批量分布对齐

一种用Sinkhorn距离做知识蒸馏的方法,能从跨样本的几何结构中捕捉教师与学生分布差异,避免传统散度在分布重叠时的退化。

崔晓 · Qin, Yulei · Gao, Yuting · Enwei, Zhang · Xu, Zihan · Tong, Wu · Li, Ke · Sun, Xing · Zhou W. · 李厚强

IEEE Transactions on Neural Networks and Learning Systems 2024

技术优势

避开三种退化

KL、RKL、JS在分布重叠时分别导致模式平均、模式坍缩和模式低估,Sinkhorn距离能精确度量分布差异,避免这些退化。

能看批量结构

摆脱逐样本蒸馏限制,改为批量重构,捕捉高维空间中跨样本分布的几何结构,获得更全局的对齐信号。

三类架构通吃

在仅编码器、编码器-解码器、仅解码器三类LLM上均超越SOTA,适应不同任务和模型结构。

应用场景