arXiv reaDer
変分分類
Variational Classification
変分分類 (VC) と呼ばれる、クロスエントロピー損失でトレーニングされたニューラル ネットワーク ソフトマックス分類の潜在変数一般化を提示します。私たちのアプローチは、変分および従来のオートエンコーダの関係と同様に、非常によく知られたソフトマックス分類モデルに新しい確率論的な観点を提供します。私たちは、最適化が自明ではない証拠下限 (ELBO) に基づいてトレーニング目標を導き出し、それを最大化するための敵対的アプローチを提案します。我々は、VC がソフトマックス分類内の固有の不一致に対処すると同時に、既製のソフトマックス分類器内で明らかになった暗黙的な仮定の代わりに、潜在空間内の事前分布のより柔軟な選択を可能にすることを示します。画像およびテキスト分類データセットの経験的評価により、変分分類は、特に分布シフトや低データ設定の下で、キャリブレーションや敵対的堅牢性などの他の望ましい特性を向上させながら、予測精度を維持することが実証されています。
We present a latent variable generalisation of neural network softmax classification trained with cross-entropy loss, referred to as variational classification (VC). Our approach offers a novel probabilistic perspective on the highly familiar softmax classification model, to which it relates similarly to how variational and traditional autoencoders relate. We derive a training objective based on the evidence lower bound (ELBO) that is non-trivial to optimize, and therefore propose an adversarial approach to maximise it. We show that VC addresses an inherent inconsistency within softmax classification, whilst also allowing more flexible choices of prior distributions in the latent space in place of implicit assumptions revealed within off-the-shelf softmax classifiers. Empirical evaluation on image and text classification datasets demonstrates that variational classification maintains prediction accuracy while improving other desirable properties such as calibration and adversarial robustness, particularly under distribution shift and low data settings.
updated: Thu Aug 03 2023 15:22:05 GMT+0000 (UTC)
published: Wed May 17 2023 17:47:19 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト