arXiv reaDer
逆イメージング問題におけるニューラル ネットワークのサイクル整合性に基づく不確実性の定量化
Cycle Consistency-based Uncertainty Quantification of Neural Networks in Inverse Imaging Problems
不確実性の推定はディープ ニューラル ネットワークの多くのアプリケーションにとって重要であり、研究者からの注目が高まっています。ここでは、サイクル一貫性に基づいた逆問題で使用されるディープ ニューラル ネットワークの不確実性定量化アプローチを示します。利用可能な物理順モデルと、当面の逆問題を解決する訓練されたディープ ニューラル ネットワークを使用して順逆サイクルを構築し、それに応じて、これらの順逆サイクルの一貫性に関する回帰分析を通じて不確実性推定量を導き出します。私たちは、サイクル一貫性メトリクスを理論的に分析し、ニューラル ネットワーク推論の不確実性、バイアス、ロバスト性との関係を導き出します。これらのサイクル一貫性ベースの不確実性推定器の有効性を実証するために、広く使用されている画像のブレ除去および超解像度ニューラル ネットワークの一部をテストベッドとして使用して、破損した分布外の入力画像データを分類しました。私たちの方法のブラインド テストは、目に見えない入力データの破損や分布の変化を特定する点で、他のモデルよりも優れた性能を発揮しました。この研究は、逆問題を解くために使用されるさまざまなニューラル ネットワークに普遍的に適用できる、実装が簡単で迅速な不確実性の定量化手法を提供します。
Uncertainty estimation is critical for numerous applications of deep neural networks and draws growing attention from researchers. Here, we demonstrate an uncertainty quantification approach for deep neural networks used in inverse problems based on cycle consistency. We build forward-backward cycles using the physical forward model available and a trained deep neural network solving the inverse problem at hand, and accordingly derive uncertainty estimators through regression analysis on the consistency of these forward-backward cycles. We theoretically analyze cycle consistency metrics and derive their relationship with respect to uncertainty, bias, and robustness of the neural network inference. To demonstrate the effectiveness of these cycle consistency-based uncertainty estimators, we classified corrupted and out-of-distribution input image data using some of the widely used image deblurring and super-resolution neural networks as testbeds. The blind testing of our method outperformed other models in identifying unseen input data corruption and distribution shifts. This work provides a simple-to-implement and rapid uncertainty quantification method that can be universally applied to various neural networks used for solving inverse problems.
updated: Mon May 22 2023 09:23:18 GMT+0000 (UTC)
published: Mon May 22 2023 09:23:18 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アソシエイト