用深度隐因子模型压缩集成模型的不确定性,让小模型也能准确判断自身可信度。
Knowledge Distillation of Uncertainty using Deep Latent Factor Model
- 将教师集成看作随机过程,用高斯过程建模其输出分布
- 在多个数据集上优于现有方法,保持了集成模型的不确定性精度
- 适合轻量化部署、语言模型微调和分布偏移场景
深度集成可实现最先进的可靠不确定性量化,但其高昂的计算与内存开销限制了实际应用,如设备端AI。知识蒸馏可将集成模型压缩为小型学生模型,但现有方法难以保留不确定性,因减少DNN规模通常导致方差降低。为此,我们提出一种新的分布蒸馏方法——高斯蒸馏,通过特殊的高斯过程——深度隐因子模型(DLF),将教师集成的输出分布建模为一个随机过程,将每个教师成员视为该过程的实现。利用期望最大化(EM)算法稳定估计DLF中的均值与协方差函数。在多个基准数据集上,实验表明所提高斯蒸馏优于现有基线。此外,我们还验证其在语言模型微调及分布偏移问题上的有效性。
原文摘要 · Abstract (English)
Deep ensembles deliver state-of-the-art, reliable uncertainty quantification, but their heavy computational and memory requirements hinder their practical deployments to real applications such as on-device AI. Knowledge distillation compresses an ensemble into small student models, but existing techniques struggle to preserve uncertainty partly because reducing the size of DNNs typically results in variation reduction. To resolve this limitation, we introduce a new method of distribution distillation (i.e. compressing a teacher ensemble into a student distribution instead of a student ensemble) called Gaussian distillation, which estimates the distribution of a teacher ensemble through a special Gaussian process called the deep latent factor model (DLF) by treating each member of the teacher ensemble as a realization of a certain stochastic process. The mean and covariance functions in the DLF model are estimated stably by using the expectation-maximization (EM) algorithm. By using multiple benchmark datasets, we demonstrate that the proposed Gaussian distillation outperforms existing baselines. In addition, we illustrate that Gaussian distillation works well for fine-tuning of language models and distribution shift problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。