arXiv reaDer
ドメインシフト下での不確実性キャリブレーションのための分布的にロバストな学習
Distributionally Robust Learning for Uncertainty Calibration under Domain Shift
ドメインシフトの下で較正された不確実性を学習するためのフレームワークを提案します。ソース(トレーニング)分布がターゲット(テスト)分布と異なる場合を考えます。バイナリドメイン分類器を使用してこのようなドメインシフトを検出し、タスクネットワークと統合して、エンドツーエンドで共同でトレーニングします。バイナリドメイン分類器は、ソース(トレーニング)分布に対するターゲット(テスト)サンプルの近さを反映する密度比を生成します。これを使用して、タスクネットワークの予測の不確実性を調整します。密度比を使用するというこのアイデアは、分散ロバスト学習(DRL)フレームワークに基づいています。これは、敵対的なリスクの最小化を通じてドメインシフトを説明します。私たちの方法は、教師なしドメイン適応(UDA)や半教師あり学習(SSL)など、多くのダウンストリームタスクに役立つキャリブレーションされた不確実性を生成することを示しています。これらのタスクでは、セルフトレーニングやFixMatchなどのメソッドは、不確実性を使用して、再トレーニング用の信頼できる疑似ラベルを選択します。私たちの実験は、DRLの導入がクロスドメインパフォーマンスの大幅な改善につながることを示しています。また、推定された密度比が人間の選択頻度と一致していることを示しており、人間が知覚する不確実性の代用との正の相関を示唆しています。
We propose a framework for learning calibrated uncertainties under domain shifts. We consider the case where the source (training) distribution differs from the target (test) distribution. We detect such domain shifts through the use of a binary domain classifier and integrate it with the task network and train them jointly end-to-end. The binary domain classifier yields a density ratio that reflects the closeness of a target (test) sample to the source (training) distribution. We employ it to adjust the uncertainty of prediction in the task network. This idea of using the density ratio is based on the distributionally robust learning (DRL) framework, which accounts for the domain shift through adversarial risk minimization. We demonstrate that our method generates calibrated uncertainties that benefit many downstream tasks, such as unsupervised domain adaptation (UDA) and semi-supervised learning (SSL). In these tasks, methods like self-training and FixMatch use uncertainties to select confident pseudo-labels for re-training. Our experiments show that the introduction of DRL leads to significant improvements in cross-domain performance. We also demonstrate that the estimated density ratios show agreement with the human selection frequencies, suggesting a positive correlation with a proxy of human perceived uncertainties.
updated: Sun Dec 26 2021 23:21:37 GMT+0000 (UTC)
published: Thu Oct 08 2020 02:10:54 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アソシエイト