この論文では、変換ベースのノイズ除去フレームワークに触発された画像ノイズ除去のための可逆ニューラルネットワーク(DnINN)を提案します。提案されたDnINNは、LINNと呼ばれる可逆ニューラルネットワークで構成され、そのアーキテクチャは、ウェーブレット理論のリフティングスキームと、変換係数からノイズを除去するために使用されるスパース性駆動型ノイズ除去ネットワークに触発されています。ノイズ除去操作は、単一のソフトしきい値操作または学習された反復収縮しきい値ネットワークを使用して実行されます。 LINNのフォワードパスは、ノイズ除去により適した過度に完全な表現を生成します。ノイズ除去された画像は、ノイズ除去ネットワークの出力を使用したLINNのバックワードパスを使用して再構成されます。シミュレーション結果は、提案されたDnINN法が、学習可能なパラメータの1/4しか必要とせずに、DnCNN法に匹敵する結果を達成することを示しています。
In this paper, we propose an invertible neural network for image denoising (DnINN) inspired by the transform-based denoising framework. The proposed DnINN consists of an invertible neural network called LINN whose architecture is inspired by the lifting scheme in wavelet theory and a sparsity-driven denoising network which is used to remove noise from the transform coefficients. The denoising operation is performed with a single soft-thresholding operation or with a learned iterative shrinkage thresholding network. The forward pass of LINN produces an over-complete representation which is more suitable for denoising. The denoised image is reconstructed using the backward pass of LINN using the output of the denoising network. The simulation results show that the proposed DnINN method achieves results comparable to the DnCNN method while only requiring 1/4 of learnable parameters.