この論文では、ヘルスケア問題の予測信頼度を定量化するための不確実性を意識したINVASEを提示します。学習可能なガウス分布を導入することにより、それらの分散を活用して不確実性の程度を測定します。バニラINVASEに基づいて、2つの追加モジュールが提案されます。つまり、予測子の不確実性定量化モジュールと、セレクターの報酬形成モジュールです。 UCI-WDBCデータセットで広範な実験を実施します。特に、私たちの方法は、約20%のクエリでほぼすべての予測バイアスを排除しますが、不確実性にとらわれない対応物は、ほぼ100%のクエリを必要とします。詳細なチュートリアルを含むオープンソースの実装は、https://github.com/jx-zhong-for-academic-purpose/Uncertainty-aware-INVASE/blob/main/tutorialinvase%2B.ipynbで入手できます。
In this paper, we present an uncertainty-aware INVASE to quantify predictive confidence of healthcare problem. By introducing learnable Gaussian distributions, we lever-age their variances to measure the degree of uncertainty. Based on the vanilla INVASE, two additional modules are proposed, i.e., an uncertainty quantification module in the predictor, and a reward shaping module in the selector. We conduct extensive experiments on UCI-WDBC dataset. Notably, our method eliminates almost all predictive bias with only about 20% queries, while the uncertainty-agnostic counterpart requires nearly 100% queries. The open-source implementation with a detailed tutorial is available at https://github.com/jx-zhong-for-academic-purpose/Uncertainty-aware-INVASE/blob/main/tutorialinvase%2B.ipynb.