軽量化が必要な背景
高精度なモデルはパラメータ数と計算量が大きく、クラウドの大規模なGPUが必要です。しかし自動運転や工場の検査のように通信遅延が許されない場面、機密データを外部に送りたくない場面、スマートフォンや組み込み機器で動かしたい場面では、端末側で推論するエッジAIが求められます。そのためにモデルを小さく速くする技術がモデル圧縮です。
主な手法
| 手法 | 内容 |
|---|---|
| 蒸留 | 大きく高精度な教師モデルの出力(クラスごとの確率分布などのソフトターゲット)を目標に、小さな生徒モデルを学習させる。確率分布にはクラス間の類似性など正解ラベル以上の情報が含まれる |
| プルーニング | 寄与の小さい重みやノード、チャネルを削除してネットワークをまばらにする |
| 量子化 | 重みや計算を32ビット浮動小数点から8ビット整数などに変換し、メモリと計算量を減らす |
宝くじ仮説
密なネットワークには、単独で学習しても元のネットワークと同等の精度に達する小さな部分ネットワーク(当たりくじ)が含まれているという仮説です。プルーニングで残った構造を初期値から学習し直しても高精度になることから提唱され、軽量化の理論的な背景の一つになっています。
試験で問われる観点
- エッジAIが求められる理由(遅延、機密性、機器の制約)
- 蒸留・プルーニング・量子化の区別
- 宝くじ仮説の内容