arXiv reaDer
最適な等角分類器の学習
Learning Optimal Conformal Classifiers
最新の深層学習ベースの分類器は、テストデータで非常に高い精度を示しますが、これは、特に医療診断などのハイステークAIアプリケーションで、安全な展開を十分に保証するものではありません。通常、予測は、信頼できる不確実性の見積もりや正式な保証なしに取得されます。コンフォーマル予測(CP)は、分類器の確率推定を使用して、ユーザー指定の確率で真のクラスを含む信頼セットを予測することにより、これらの問題に対処します。ただし、トレーニング後にCPを個別の処理ステップとして使用すると、基になるモデルが信頼区間の予測に適応できなくなります。したがって、このペーパーでは、コンフォーマルラッパーをエンドツーエンドで使用してモデルをトレーニングすることを目的として、トレーニング中にCPを介して差別化する戦略について説明します。私たちのアプローチであるコンフォーマルトレーニング(ConfTr)では、トレーニング中にミニバッチのコンフォーマル化を具体的に「シミュレート」します。 CTは、平均信頼区間サイズ(非効率)を減らすことにより、分類のための最先端のCP法よりも優れていることを示します。さらに、標準CPでは困難なテスト時に予測された信頼セットを「形成」することができます。いくつかのデータセットを使用した実験で、CPが提供する保証を維持しながら、ConfTrが非効率性をクラス間で分散する方法に影響を与えたり、含まれるクラスの観点から信頼セットの構成をガイドしたりできることを示します。
Modern deep learning based classifiers show very high accuracy on test data but this does not provide sufficient guarantees for safe deployment, especially in high-stake AI applications such as medical diagnosis. Usually, predictions are obtained without a reliable uncertainty estimate or a formal guarantee. Conformal prediction (CP) addresses these issues by using the classifier's probability estimates to predict confidence sets containing the true class with a user-specified probability. However, using CP as a separate processing step after training prevents the underlying model from adapting to the prediction of confidence sets. Thus, this paper explores strategies to differentiate through CP during training with the goal of training model with the conformal wrapper end-to-end. In our approach, conformal training (ConfTr), we specifically "simulate" conformalization on mini-batches during training. We show that CT outperforms state-of-the-art CP methods for classification by reducing the average confidence set size (inefficiency). Moreover, it allows to "shape" the confidence sets predicted at test time, which is difficult for standard CP. On experiments with several datasets, we show ConfTr can influence how inefficiency is distributed across classes, or guide the composition of confidence sets in terms of the included classes, while retaining the guarantees offered by CP.
updated: Mon Oct 18 2021 11:25:33 GMT+0000 (UTC)
published: Mon Oct 18 2021 11:25:33 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アソシエイト