arXiv reaDer
分布外検出のために超球面埋め込みを利用する方法は?
How to exploit hyperspherical embeddings for out-of-distribution detection?
分布外 (OOD) の検出は、信頼性の高い機械学習にとって重要なタスクです。表現学習の最近の進歩により、距離ベースの OOD 検出が可能になりました。この場合、テスト サンプルは、分散 (ID) クラスの重心またはプロトタイプから比較的離れている場合に OOD として検出されます。ただし、従来の方法は、ID サンプルを分類するのに十分な市販のコントラスト損失を直接利用しますが、テスト入力に OOD サンプルが含まれる場合は最適に設計されていません。この作業では、OOD 検出に超球面埋め込みを利用する新しい表現学習フレームワークである CIDER を提案します。 CIDER は、強力な ID-OOD 分離性を促進するために 2 つの損失を共同で最適化します。異なるクラス プロトタイプ間で大きな角度距離を促進する分散損失と、サンプルがクラス プロトタイプに近づくようにするコンパクトネス損失です。超球空間における OOD 検出性能と埋め込み特性との間の未知の関係を分析して確立し、分散とコンパクト性の重要性を実証します。 CIDER は、FPR95 で最新のライバルを 19.36% 上回る優れたパフォーマンスを確立しています。コードは https://github.com/deeplearning-wisc/cider で入手できます。
Out-of-distribution (OOD) detection is a critical task for reliable machine learning. Recent advances in representation learning give rise to distance-based OOD detection, where testing samples are detected as OOD if they are relatively far away from the centroids or prototypes of in-distribution (ID) classes. However, prior methods directly take off-the-shelf contrastive losses that suffice for classifying ID samples, but are not optimally designed when test inputs contain OOD samples. In this work, we propose CIDER, a novel representation learning framework that exploits hyperspherical embeddings for OOD detection. CIDER jointly optimizes two losses to promote strong ID-OOD separability: a dispersion loss that promotes large angular distances among different class prototypes, and a compactness loss that encourages samples to be close to their class prototypes. We analyze and establish the unexplored relationship between OOD detection performance and the embedding properties in the hyperspherical space, and demonstrate the importance of dispersion and compactness. CIDER establishes superior performance, outperforming the latest rival by 19.36% in FPR95. Code is available at https://github.com/deeplearning-wisc/cider.
updated: Tue Feb 28 2023 20:21:57 GMT+0000 (UTC)
published: Tue Mar 08 2022 23:44:01 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アソシエイト