sgr-interview-300

第 71 题:知识蒸馏的温度系数,软标签熵与蒸馏效果的关系?

题目

知识蒸馏的温度系数,软标签熵与蒸馏效果的关系?


完整讲解

一、知识蒸馏与软标签

知识蒸馏用教师模型(大模型)的软输出作为学生模型(小模型)的监督。设教师 logits 为 $\boldsymbol{z}^T$,学生为 $\boldsymbol{z}^S$,软标签为 softmax 后的概率分布 $\boldsymbol{p}^T,\ \boldsymbol{p}^S$。蒸馏损失常用 KL 散度: \(\mathcal{L}_{KD} = \text{KL}(\boldsymbol{p}^S \| \boldsymbol{p}^T) = \sum_i p_i^T \ln \frac{p_i^T}{p_i^S}.\)

二、温度系数与软标签熵

引入温度 $T$,将 logits 缩放后再做 softmax: \(p_i = \frac{\exp(z_i / T)}{\sum_j \exp(z_j / T)}.\)

三、蒸馏效果与温度选择


面试要点


记忆要点

返回模块 返回总览