Variational volume reconstruction with the Deep Ritz Method
This addresses computational and accuracy challenges in biomedical imaging for applications like slice-to-volume reconstruction, though it is incremental as it builds on existing variational and neural network methods.
The paper tackles the problem of reconstructing 3D volumes from sparse, noisy slice data, such as in MRI-based imaging, by using the Deep Ritz method to avoid segmentation and achieve high-quality reconstructions in seconds.
We present a novel approach to variational volume reconstruction from sparse, noisy slice data using the Deep Ritz method. Motivated by biomedical imaging applications such as MRI-based slice-to-volume reconstruction (SVR), our approach addresses three key challenges: (i) the reliance on image segmentation to extract boundaries from noisy grayscale slice images, (ii) the need to reconstruct volumes from a limited number of slice planes, and (iii) the computational expense of traditional mesh-based methods. We formulate a variational objective that combines a regression loss designed to avoid image segmentation by operating on noisy slice data directly with a modified Cahn-Hilliard energy incorporating anisotropic diffusion to regularize the reconstructed geometry. We discretize the phase field with a neural network, approximate the objective at each optimization step with Monte Carlo integration, and use ADAM to find the minimum of the approximated variational objective. While the stochastic integration may not yield the true solution to the variational problem, we demonstrate that our method reliably produces high-quality reconstructed volumes in a matter of seconds, even when the slice data is sparse and noisy.