Deep Convolutional Neural Network for Inverse Problems in Imaging
Kyong Hwan JinMichael T. McCannEmmanuel FrousteyMichael Unser
Proposes a framework combining direct physical inversion with a residual convolutional neural network to solve ill-posed imaging inverse problems, achieving superior quality over standard iterative reconstruction while recovering sparse-view computed tomography images in sub-second speeds.
Medical imaging modalities such as X-ray computed tomography face an inherent trade-off between acquisition time and image quality. Short acquisition times or sparse-view scans reduce harmful radiation exposure and patient motion artifacts, but direct mathematical reconstruction yields severe streaking artifacts. While iterative reconstruction methods produce high-quality images, they suffer from high computational demands and long reconstruction runtimes that limit their routine clinical adoption.
The article sets out to develop and evaluate a deep learning reconstruction framework that combines direct physical inversion with a deep convolutional neural network to solve imaging inverse problems rapidly while preserving image quality.
The researchers established a mathematical framework showing that a broad class of imaging problems can be formulated as convolutional operations, enabling the use of convolutional neural networks. Based on this, they designed a system named FBPConvNet, which performs a fast, standard filtered back projection to provide an initial physical estimate and then applies a multiresolution neural network based on the U-Net architecture with residual learning to remove artifacts. The approach was evaluated on sparse-view computed tomography using three datasets: a synthetic geometric ellipsoid dataset, 500 clinical in-vivo scans from the Mayo Clinic, and 377 experimental scans of a rat brain collected from a synchrotron light source.
The evaluation yielded several key findings. First, FBPConvNet consistently outperformed state-of-the-art total variation iterative reconstruction on realistic clinical and biological datasets, achieving higher signal-to-noise ratios (such as 36.15 dB versus 31.92 dB on clinical data at 143 views). Second, the neural network effectively preserved fine anatomical textures and detailed structures, avoiding the unnatural, oversmoothed cartoon-like artifacts typical of total variation methods. Third, the proposed method reconstructed a 512-by-512 image in under one second on a standard graphics processing unit, representing a massive speedup compared to iterative methods that required approximately seven minutes per image. Finally, the iterative method maintained an advantage only on idealized, piecewise-constant synthetic phantoms where total variation regularization is mathematically optimal.
These findings indicate that deep learning combined with physical domain models can substantially reduce patient radiation exposure by enabling reconstructions from sparse views (subsampled by up to twenty times) without sacrificing diagnostic texture quality. The dramatic reduction in processing time makes high-quality sparse reconstruction practical for real-time and high-throughput clinical workflows.
Organizations evaluating this approach should consider piloting direct-inversion neural networks in imaging pipelines where rapid processing and low-dose scans are critical. Before clinical deployment, future work should develop unified strategies for heterogeneous datasets and expand network architectures to support complex-valued data required for modalities such as magnetic resonance imaging.
A primary limitation of the method is its lack of transferability across varying scan configurations; a model trained on a specific downsampling factor must be retrained to process scans acquired with different numbers of views. While confidence in the experimental performance on computed tomography is high, testing remains constrained to two-dimensional slices, and careful validation on larger, varied clinical cohorts is necessary prior to production deployment.
- Paper: U-Net: Convolutional Networks for Biomedical Image Segmentation, Olaf Ronneberger et al. (2015). Introduces the U-Net multiresolution encoder-decoder architecture with skip connections that forms the foundational neural network backbone adapted by FBPConvNet for post-processing artifact removal.
- Paper: Image Super-Resolution Using Deep Convolutional Networks, Chao Dong et al. (2014). Establishes the foundational framework of using deep convolutional neural networks to solve ill-posed spatial inverse problems directly via end-to-end mapping.
- Paper: Accurate Image Super-Resolution Using Very Deep Convolutional Networks, Jiwon Kim et al. (2016). Demonstrates the effectiveness of deep residual learning in convolutional networks for image restoration, which informs FBPConvNet's strategy of learning residual artifact corrections.
- Paper: Deep Image Prior, Dmitry Ulyanov et al. (2017). Extends the application of convolutional neural networks to inverse problems by showing that network architectures intrinsically act as powerful regularizers without requiring training datasets.
- Paper: A survey on deep learning in medical image analysis, Geert Litjens et al. (2017). Provides a comprehensive survey synthesizing how deep convolutional networks, including U-Net and direct physical inversion approaches, transformed medical imaging tasks.
- Paper: FFDNet: Toward a Fast and Flexible Solution for CNN-Based Image Denoising, Kai Zhang et al. (2017). Builds upon learned image restoration by introducing a flexible CNN that handles spatially non-uniform and varying noise levels applicable to clinical imaging artifacts.
- Paper: Attention U-Net: Learning Where to Look for the Pancreas, Ozan Oktay et al. (2018). Improves standard U-Net-based medical image processing architectures by integrating attention gates to selectively focus on anatomical structures.
- Paper: SwinIR: Image Restoration Using Swin Transformer, Jingyun Liang et al. (2021). Advances deep learning for image restoration by replacing conventional CNN architectures with shifted-window self-attention transformers.
- Paper: Restormer: Efficient Transformer for High-Resolution Image Restoration, Syed Waqas Zamir et al. (2022). Presents a high-resolution transformer-based image restoration model that efficiently captures long-range spatial dependencies to resolve complex restoration inverse problems.
