arXiv reaDer
Fast Bayesian Uncertainty Estimation and Reduction of Batch Normalized Single Image Super-Resolution Network
畳み込みニューラルネットワーク(CNN)は、近年、画像の超解像タスクで前例のない成功を収めています。ただし、ネットワークのパフォーマンスはトレーニングセットの分布に依存し、分布外のサンプルでは低下します。この論文では、出力に関連する不確実性を推定するためにベイズアプローチを採用し、それを深画像超解像モデルに適用して、上記の懸念に対処します。バッチ平均と分散の確率論がモンテカルロ(MC)サンプルを生成する、バッチ正規化レイヤーを使用した不確実性推定手法を使用します。異なる確率的パラメータを使用した異なる超解像画像に他ならないMCサンプルは、画像を再構成し、再構成の信頼性または不確実性マップを提供します。 MCサンプル生成のより高速なアプローチを提案します。これにより、テスト中に画像サイズを変更できます。したがって、画像再構成領域に役立ちます。私たちの実験結果は、この不確実性マップがディープCNNモデルによって生成された再構成の品質に強く関連していることを示し、その限界を説明しています。さらに、この論文は、入力画像に対するモデルの不確実性を低減するためのアプローチを提案し、画像の超解像モデルに対する敵対的な攻撃を防御するのに役立ちます。提案された不確実性低減手法は、分布外のテスト画像のモデルのパフォーマンスも向上させます。私たちの知る限りでは、私たちはあらゆる画像再構成領域で敵対的な防衛機構を提案した最初の人物です。
Convolutional neural network (CNN) has achieved unprecedented success in image super-resolution tasks in recent years. However, the network's performance depends on the distribution of the training sets and degrades on out-of-distribution samples. This paper adopts a Bayesian approach for estimating uncertainty associated with output and applies it in a deep image super-resolution model to address the concern mentioned above. We use the uncertainty estimation technique using the batch-normalization layer, where stochasticity of the batch mean and variance generate Monte-Carlo (MC) samples. The MC samples, which are nothing but different super-resolved images using different stochastic parameters, reconstruct the image, and provide a confidence or uncertainty map of the reconstruction. We propose a faster approach for MC sample generation, and it allows the variable image size during testing. Therefore, it will be useful for image reconstruction domain. Our experimental findings show that this uncertainty map strongly relates to the quality of reconstruction generated by the deep CNN model and explains its limitation. Furthermore, this paper proposes an approach to reduce the model's uncertainty for an input image, and it helps to defend the adversarial attacks on the image super-resolution model. The proposed uncertainty reduction technique also improves the performance of the model for out-of-distribution test images. To the best of our knowledge, we are the first to propose an adversarial defense mechanism in any image reconstruction domain.
updated: Wed May 19 2021 04:53:43 GMT+0000 (UTC)
published: Fri Mar 22 2019 09:05:32 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト