大型模型學習能力強但運行速度慢,小型模型運行速度快但學習能力弱。知識蒸餾透過訓練一個緊湊的學生模型來模仿大型教師模型,並利用教師模型的軟預測而非硬標籤,從而兼顧兩者之間的差異。
軟預測包含更多資訊。當老師給貓分配 70% 的機率,給狗 25% 的機率時,學生就能學到貓和狗有一些共同特徵。而硬標籤只會說是貓。這種額外的訊號有助於學生比僅基於標籤進行訓練更好地進行泛化。
常見的蒸餾變體
- 基於響應的輸出logits蒸餾
- 基於特徵的中間層蒸餾
- 基於關係的層關係提煉
- 同一架構內的自蒸餾
這項技術為許多生產系統提供支援。移動助理、推薦模型和即時檢測器通常依賴精煉後的學生模型,這些模型以極低的成本保留了教師模型的大部分準確率。
Comments (2)
Leave a comment