A residual U-Net network with image prior for 3D image denoising

Denoising algorithms via sparse representation are among the state-of-the art for image restoration. On previous work, we proposed SPADE - a sparse- and prior-based method for 3D-image denoising. In this work, we extend this idea to learning approaches and propose a novel residual-U-Net prior-based (ResPrU-Net) method that exploits a prior image. The proposed ResPrU-Net architecture has two inputs, the noisy image and the prior image, and a residual connection that connects the prior image to the output of the network. We compare ResPrU-Net to U-Net and SPADE on human knee data acquired on a spectral computerized tomography scanner. The prior image is built from the noisy image by combining information from neighbor slices and it is the same for both SPADE and ResPrU-Net. For deep learning approaches, we use four knee samples and data augmentation for training, one knee for validation and two for test. Results show that for high noise, U-Net leads to worst results, with images that are excessively blurred. Prior-based methods, SPADE and ResPrU-Net, outperformed U-Net, leading to restored images that present similar image quality than the target. ResPrU-Net provides slightly better results than SPADE. For low noise, methods present similar results.

[1]  Jong Chul Ye,et al.  A deep convolutional neural network using directional wavelets for low‐dose X‐ray CT reconstruction , 2016, Medical physics.

[2]  Alessandro Foi,et al.  Image Denoising by Sparse 3-D Transform-Domain Collaborative Filtering , 2007, IEEE Transactions on Image Processing.

[3]  Karl Kunisch,et al.  Total Generalized Variation , 2010, SIAM J. Imaging Sci..

[4]  M. Elad,et al.  $rm K$-SVD: An Algorithm for Designing Overcomplete Dictionaries for Sparse Representation , 2006, IEEE Transactions on Signal Processing.

[5]  Jonas Adler,et al.  Solving ill-posed inverse problems using iterative deep neural networks , 2017, ArXiv.

[6]  J. F. P. J. Abascal,et al.  A sparse and prior based method for 3D image denoising , 2019, 2019 27th European Signal Processing Conference (EUSIPCO).

[7]  Jinyi Qi,et al.  Low-dose CT image denoising without high-dose reference images , 2019, 15th International Meeting on Fully Three-Dimensional Image Reconstruction in Radiology and Nuclear Medicine.

[8]  David P. Cormode,et al.  Review of an initial experience with an experimental spectral photon-counting computed tomography system , 2017 .

[9]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[10]  Ali Mousavi,et al.  Learning to invert: Signal recovery via Deep Convolutional Networks , 2017, 2017 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[11]  Martín Abadi,et al.  TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems , 2016, ArXiv.

[12]  Manuel Desco,et al.  A Novel Prior- and Motion-Based Compressed Sensing Method for Small-Animal Respiratory Gated CT , 2016, PloS one.

[13]  Michael Unser,et al.  Convolutional Neural Networks for Inverse Problems in Imaging: A Review , 2017, IEEE Signal Processing Magazine.

[14]  Michael Unser,et al.  Deep Convolutional Neural Network for Inverse Problems in Imaging , 2016, IEEE Transactions on Image Processing.

[15]  L. Rudin,et al.  Nonlinear total variation based noise removal algorithms , 1992 .

[16]  Xuanqin Mou,et al.  Low-Dose CT Image Denoising Using a Generative Adversarial Network With Wasserstein Distance and Perceptual Loss , 2017, IEEE Transactions on Medical Imaging.

[17]  A. Bruckstein,et al.  K-SVD : An Algorithm for Designing of Overcomplete Dictionaries for Sparse Representation , 2005 .

[18]  Yoram Singer,et al.  Adaptive Subgradient Methods for Online Learning and Stochastic Optimization , 2011, J. Mach. Learn. Res..

[19]  Jiasong Wu,et al.  Improving Low-Dose CT Image Using Residual Convolutional Network , 2017, IEEE Access.

[20]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[21]  Xavier Bresson,et al.  Bregmanized Nonlocal Regularization for Deconvolution and Sparse Reconstruction , 2010, SIAM J. Imaging Sci..

[22]  Stanley Osher,et al.  A Low Patch-Rank Interpretation of Texture , 2013, SIAM J. Imaging Sci..