arXiv reaDer
最適な等角分類器の学習
Learning Optimal Conformal Classifiers
最新の深層学習ベースの分類器は、テストデータに対して非常に高い精度を示しますが、これは、特に医療診断などのハイステークAIアプリケーションで、安全な展開を十分に保証するものではありません。通常、予測は、信頼できる不確実性の見積もりや正式な保証なしに取得されます。コンフォーマル予測(CP)は、分類器の予測(確率推定など)を使用してこれらの問題に対処し、ユーザー指定の確率で真のクラスを含む信頼区間を予測します。ただし、トレーニング後にCPを個別の処理ステップとして使用すると、基になるモデルが信頼セットの予測に適応できなくなります。したがって、このペーパーでは、コンフォーマルラッパーをエンドツーエンドでモデルをトレーニングすることを目的として、トレーニング中にCPを介して差別化する戦略について説明します。私たちのアプローチであるコンフォーマルトレーニング(ConfTr)では、トレーニング中にミニバッチのコンフォーマル化を具体的に「シミュレート」します。標準のトレーニングと比較して、ConfTrは、トレーニング後に適用される最先端の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 predictions, e.g., its 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. Compared to standard training, ConfTr reduces the average confidence set size (inefficiency) of state-of-the-art CP methods applied after training. 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 Mar 07 2022 17:05:26 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アソシエイト