回帰問題の不確実性推定を可能にするベイジアン ディープ ラーニング アプローチは、多くの場合収束が遅く、定量化に効果的に使用できない不十分に調整された不確実性推定を生成します。最近提案された事後キャリブレーション手法は、回帰問題に適用できることはめったになく、多くの場合、すでに遅いモデル トレーニング フェーズにオーバーヘッドが追加されます。この作業は、ポストアニーリングと呼ばれる回帰タスクの高速校正不確実性推定方法を提示します。これは、深い回帰モデルの収束を一貫して改善し、事後校正フェーズなしで校正不確実性をもたらします。低次元の回帰問題のみに焦点を当てた、回帰における較正された不確実性のための以前の方法とは異なり、私たちの方法は、広範囲の回帰問題でうまく機能します。私たちの経験的分析は、私たちのアプローチが多層パーセプトロン、1D/2D 畳み込みネットワーク、グラフ ニューラル ネットワークなどのさまざまなネットワーク アーキテクチャに一般化できることを示しています。 MRI画像を使用した表現、自然画像の超解像、および医用画像の翻訳。
Bayesian deep learning approaches that allow uncertainty estimation for regression problems often converge slowly and yield poorly calibrated uncertainty estimates that can not be effectively used for quantification. Recently proposed post hoc calibration techniques are seldom applicable to regression problems and often add overhead to an already slow model training phase. This work presents a fast calibrated uncertainty estimation method for regression tasks, called posterior annealing, that consistently improves the convergence of deep regression models and yields calibrated uncertainty without any post hoc calibration phase. Unlike previous methods for calibrated uncertainty in regression that focus only on low-dimensional regression problems, our method works well on a wide spectrum of regression problems. Our empirical analysis shows that our approach is generalizable to various network architectures including, multilayer perceptrons, 1D/2D convolutional networks, and graph neural networks, on five vastly diverse tasks, i.e., chaotic particle trajectory denoising, physical property prediction of molecules using 3D atomistic representation, natural image super-resolution, and medical image translation using MRI images.