Memory-efficient Learning for Large-scale Computational Imaging
Critical aspects of computational imaging systems, such as experimental design and image priors, can be optimized through deep networks formed by the unrolled iterations of classical model-based reconstructions (termed physics-based networks). However, for real-world large-scale inverse problems, computing gradients via backpropagation is infeasible due to memory limitations of graphics processing units. In this work, we propose a memory-efficient learning procedure that exploits the reversibility of the network's layers to enable data-driven design for large-scale computational imaging systems. We demonstrate our method on a small-scale compressed sensing example, as well as two large-scale real-world systems: multi-channel magnetic resonance imaging and super-resolution optical microscopy.
Code (0)
등록된 구현이 없습니다.
Tasks
compressed sensingExperimental DesignSuper-ResolutionSimilar Papers 제목 키워드 기반
Memory-efficient Learning for Large-scale Computational Imaging -- NeurIPS deep inverse workshop
Computational imaging systems jointly design computation and hardware to retrieve information which is not traditionally accessible with standard imaging systems. Recently, critical aspects such as experimental design an…
Experimental DesignSuper-ResolutionHigh-dimensional Fast Convolutional Framework (HICU) for Calibrationless MRI
Purpose: To present a computational procedure for accelerated, calibrationless magnetic resonance image (Cl-MRI) reconstruction that is fast, memory efficient, and scales to high-dimensional imaging. Theory and Methods: …
compressed sensingMRI ReconstructionVocal Bursts Intensity PredictionSIMBA: Scalable Inversion in Optical Tomography using Deep Denoising Priors
Two features desired in a three-dimensional (3D) optical tomographic image reconstruction algorithm are the ability to reduce imaging artifacts and to do fast processing of large data volumes. Traditional iterative inver…
DenoisingImage ReconstructionScan-based Compressed Terahertz Imaging and Real-Time Reconstruction via the Complex-valued Fast Block Sparse Bayesian Learning Algorithm
Compressed Sensing based Terahertz imaging (CS-THz) is a computational imaging technique. It uses only one THz receiver to accumulate the random modulated image measurements where the original THz image is reconstruct fr…
compressed sensingSGD-Net: Efficient Model-Based Deep Learning with Theoretical Guarantees
Deep unfolding networks have recently gained popularity in the context of solving imaging inverse problems. However, the computational and memory complexity of data-consistency layers within traditional deep unfolding ne…
Deep Learning