ベイズ自己蒸留による画像分類の改善

公開:2025年12月30日 11:48
1分で読める
ArXiv

分析

本論文は、画像分類のための深層ニューラルネットワークの訓練に、ベイズ自己蒸留(BSD)という新しいアプローチを提案しています。従来の教師あり学習と既存の自己蒸留法の限界に対処するため、ベイズ推論を用いてサンプル固有のターゲット分布を作成します。主な利点は、BSDが初期化後、ハードターゲットに依存しないことであり、これにより精度、キャリブレーション、ロバスト性、およびラベルノイズ下での性能が向上します。結果は、さまざまなアーキテクチャとデータセットにおいて、既存の方法よりも大幅な改善を示しています。

参照

BSDは、既存のアーキテクチャを保持する自己蒸留法よりも、一貫して高いテスト精度(例:CIFAR-100のResNet-50で+1.4%)と、有意に低い期待キャリブレーションエラー(ECE)(CIFAR-100のResNet-50で-40%)を達成しています。