Image reconstruction by domain-transform manifold learning
Bo ZhuJeremiah Z. LiuBruce R. RosenMatthew S. Rosen
Introduces AUTOMAP, a unified deep learning framework that learns direct transforms from raw sensor data to images across diverse acquisition strategies, eliminating ad hoc reconstruction pipelines while reducing noise and artifacts.
Modern imaging systems across medicine, astronomy, and materials science rely on image reconstruction to transform raw sensor measurements into viewable images. Standard reconstruction approaches typically depend on rigid, handcrafted signal processing pipelines that are tailored to specific hardware and often require manual parameter tuning. These conventional methods are especially vulnerable when scans are accelerated or radiation doses are lowered, conditions that reduce the signal-to-noise ratio and introduce severe image artifacts.
The article demonstrates and evaluates a unified deep learning framework named AUTOMAP (Automated Transform by Manifold Approximation), designed to automatically learn the end-to-end mathematical mapping from raw sensor data directly to output images across diverse acquisition strategies using a single neural network architecture.
To demonstrate this capability, the researchers implemented a deep neural network consisting of fully connected layers coupled with a sparse convolutional autoencoder. They evaluated the framework on human brain magnetic resonance imaging (MRI) benchmarks across four demanding acquisition scenarios: Radon projection data, spiral non-Cartesian sampling, 40% Poisson-disc undersampling, and hardware-misaligned sampling. Notably, for most tasks the model was trained entirely on generic photographs of natural scenes from ImageNet rather than medical images, testing its ability to generalize to unseen biological structures under varying levels of noise.
The analysis produced several key findings. First, AUTOMAP successfully learned accurate reconstruction mappings across all four acquisition strategies without altering network hyperparameters or architectures between tasks. Second, the framework demonstrated superior immunity to noise and sampling artifacts compared with standard techniques, eliminating white noise amplification, ringing, compressed sensing distortions, and aliasing. Third, internal layer evaluations revealed that training on structured image data naturally induced sparse hidden-layer activations and organized spatial weight correlations, explaining its robust noise suppression. Finally, by incorporating synthetic phase modulations during training, the network accurately reconstructed both complex magnitude and phase images from in vivo MRI scans.
These findings indicate that image reconstruction can be unified into a single data-driven framework, eliminating the need for custom, human-engineered processing chains. By providing robust reconstruction at low signal-to-noise levels, this approach enables faster scan times and lower radiation doses in clinical settings without degrading diagnostic quality. It also allows standard public photographic and medical repositories to serve as effective training data.
Organizations and research teams developing imaging systems should consider piloting data-driven reconstruction models to enhance existing hardware performance and explore novel, non-traditional sensor sampling patterns. Next steps should focus on scaling the framework to higher image resolutions and testing performance across other clinical modalities such as low-dose computed tomography, ultrasound, and optical coherence tomography.
Readers should note that the evaluations were primarily conducted at an image matrix size of 128x128 pixels, and fully connected deep learning architectures can impose significant memory and computational demands when scaling to high-resolution volumetric datasets. While confidence in the demonstrated noise immunity and cross-domain generalization is high under the tested conditions, clinical deployment will require validation on full-scale clinical workflows and larger patient cohorts.
- Paper: Deep Convolutional Neural Network for Inverse Problems in Imaging, Kyong Hwan Jin et al. (2016). This foundational paper establishes the framework of using deep convolutional neural networks to solve imaging inverse problems, providing essential context for deep learned reconstruction.
- Paper: Understanding deep image representations by inverting them, Aravindh Mahendran et al. (2014). Understanding how deep visual representations can be inverted to reconstruct images provides foundational insight into learning inverse mappings from encoded representations.
- Paper: Image Super-Resolution Using Deep Convolutional Networks, Chao Dong et al. (2014). This seminal work demonstrates end-to-end mapping from degraded measurements to high-resolution images using neural networks, underpinning deep learning-based image reconstruction.
- Paper: Image Restoration Using Very Deep Convolutional Encoder-Decoder Networks with Symmetric Skip Connections, Xiao-Jiao Mao et al. (2016). This paper establishes the use of deep encoder-decoder networks with skip connections for visual restoration, informing architecture designs in deep learned inverse problem solvers.
- Paper: Learning a Variational Network for Reconstruction of Accelerated MRI Data, Kerstin Hammernik et al. (2017). This work extends deep learned MRI reconstruction by explicitly incorporating physical variational network formulations to accelerate multi-channel MRI data acquisition.
- Paper: An overview of deep learning in medical imaging focusing on MRI, Alexander Selvikvåg Lundervold et al. (2018). This comprehensive overview reviews the broader landscape and clinical integration of deep learning across magnetic resonance imaging workflows, contextualizing manifold-learning reconstruction.
- Paper: Noise2Noise: Learning Image Restoration without Clean Data, Jaakko Lehtinen et al. (2018). This work advances learned image restoration and undersampled MRI reconstruction by demonstrating how neural networks can train directly on corrupted paired data without requiring clean ground truth.
- Paper: Diffusion Posterior Sampling for General Noisy Inverse Problems, Hyungjin Chung et al. (2022). This study pushes learned inverse problem solvers further by using generative diffusion models and posterior sampling to handle general noisy and nonlinear imaging transforms.
- Paper: Deep Image Prior, Dmitry Ulyanov et al. (2017). This paper investigates how network architectures implicitly regularize inverse problems, offering a complementary perspective to supervised manifold transform learning.
