arXiv reaDer
Distinction Maximization Loss: 損失の置き換えとキャリブレーションによる分布外検出と不確実性推定の効率的な改善
Distinction Maximization Loss: Efficiently Improving Out-of-Distribution Detection and Uncertainty Estimation by Replacing the Loss and Calibrating
堅牢な決定論的ニューラル ネットワークの構築は、依然として課題です。一方で、一部のアプローチでは、状況によっては分類精度が低下しますが、分布外検出が改善されます。一方、一部の方法では、推論効率が低下しますが、分類精度、不確実性推定、および分布外検出が同時に向上します。この論文では、通常の SoftMax 損失 (つまり、線形出力層、SoftMax アクティベーション、クロスエントロピー損失の組み合わせ) のドロップイン代替として機能する DisMax 損失を使用して、決定論的ニューラル ネットワークをトレーニングすることを提案します。 . IsoMax+ 損失から始めて、正しいクラスに関連付けられたものだけでなく、すべてのプロトタイプまでの距離に基づいて各ロジットを作成します。また、画像を組み合わせて分数確率正則化と呼ばれるものを構築するメカニズムも紹介します。さらに、トレーニング後にネットワークを調整するための迅速な方法を提示します。最後に、分布外検出を実行するための複合スコアを提案します。私たちの実験では、DisMax は通常、決定論的ニューラル ネットワークの推論効率を維持しながら、分類精度、不確実性推定、および分布外検出において現在のアプローチよりも優れていることが示されています。結果を再現するコードは、https://github.com/dlmacedo/distinction-maximization-loss で入手できます。
Building robust deterministic neural networks remains a challenge. On the one hand, some approaches improve out-of-distribution detection at the cost of reducing classification accuracy in some situations. On the other hand, some methods simultaneously increase classification accuracy, uncertainty estimation, and out-of-distribution detection at the expense of reducing the inference efficiency. In this paper, we propose training deterministic neural networks using our DisMax loss, which works as a drop-in replacement for the usual SoftMax loss (i.e., the combination of the linear output layer, the SoftMax activation, and the cross-entropy loss). Starting from the IsoMax+ loss, we create each logit based on the distances to all prototypes, rather than just the one associated with the correct class. We also introduce a mechanism to combine images to construct what we call fractional probability regularization. Moreover, we present a fast way to calibrate the network after training. Finally, we propose a composite score to perform out-of-distribution detection. Our experiments show that DisMax usually outperforms current approaches simultaneously in classification accuracy, uncertainty estimation, and out-of-distribution detection while maintaining deterministic neural network inference efficiency. The code to reproduce the results is available at https://github.com/dlmacedo/distinction-maximization-loss.
updated: Mon Aug 01 2022 05:04:23 GMT+0000 (UTC)
published: Thu May 12 2022 04:37:35 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アソシエイト