FINDER: Zero-Shot Field-Integrated Network for Distortion-free EPI Reconstruction in Diffusion MRI
Authors: Namgyu Han, Seong Dae Yun, Chaeeun Lim, Sunghyun Seok, Sunju Kim, Yoonhwan Kim, Yohan Jun, Tae Hyung Kim, +2 more
Organizations: Department of Artificial Intelligence and Robotics, Sejong University, Seoul 05006, South Korea · Institute of Neuroscience and Medicine 4, INM-4, Forschungszentrum Jülich, Jülich, Germany · Athinoula A. Martinos Center for Biomedical Imaging, Charlestown, MA 02129, USA · Harvard Medical School, Boston, MA 02115, USA · Department of Computer Engineering, Hongik University, Seoul 04066, South Korea
Echo-planar imaging (EPI) remains the cornerstone of diffusion MRI, but it is prone to severe geometric distortions due to its rapid sampling scheme that renders the sequence highly sensitive to B0 field inhomogeneities. While deep learning has helped improve MRI reconstruction, integrating robust geometric distortion correction into a self-supervised framework remains an unmet need. To address this, we present FINDER (Field-Integrated Network for Distortion-free EPI Reconstruction), a novel zero-shot, scan-specific framework that reformulates reconstruction as a joint optimization of the underlying image and the B0 field map. Specifically, we employ a physics-guided unrolled network that integrates dual-domain denoisers and virtual coil extensions to enforce robust data consistency. This is coupled with an Implicit Neural Representation (INR) conditioned on spatial coordinates and latent image features to model the off-resonance field as a continuous, differentiable function. Employing an alternating minimization strategy, FINDER synergistically updates the reconstruction network and the field map, effectively disentangling susceptibility-induced geometric distortions from anatomical structures. Experimental results demonstrate that FINDER achieves superior geometric fidelity and image quality compared to state-of-the-art baselines, offering a robust solution for high-quality diffusion imaging.
Figures & tables
Figure 1: Overview of the proposed FINDER framework. The upper part presents the INR network for field map updates, while the lower part details the physics-guided unrolled image reconstruction network.
Figure 2: Visual comparison of the diffusion-weighted images at 5-fold acceleration (single direction). Yellow insets highlight baseline artifacts effectively suppressed by FINDER.
Metric
TOPUP
FD-net
BUDA-LORAKS
FINDER
NRMSE ↓
0.3426
0.3471
0.2066
0.1311
LCC ↑
0.9520
0.9410
0.9778
0.9904
SSIM ↑
0.6037
0.5579
0.7949
0.8451
HFEN ↓
0.5334
0.6305
0.3702
0.2669
LPIPS ↓
0.2934
0.3485
0.1646
0.0925
Table 1: Quantitative metrics averaged across all slices and diffusion directions
Figure 3: Visual comparison of the average DWIs across 32 directions and the corresponding FA maps. Yellow insets highlight severe noise amplification and artifacts in baseline methods, which are effectively suppressed by FINDER.
Figure 4: Visual comparison of the estimated field maps. TOPUP and FD-net estimate the field maps based on the highly aliased initial SENSE reconstructions. In contrast, FINDER dynamically refines the initial TOPUP field map using the proposed INR network. Red arrows indicate severe estimation errors, which are effectively mitigated by FINDER. The RMSEs of field maps in the brain region are 19.43 Hz, 31.07 Hz, and 13.57 Hz for TOPUP, FD-net, and FINDER, respectively.
Echo Planar Imaging (EPI) is the standard acquisition technique for diffusion and functional neuroimaging, enabling rapid imaging but suffering from geometric distortions caused by B0 field inhomogeneities. Existing correction methods first reconstruct distorted images using parallel imaging, then estimate the B0 field and correct the distortion in the image domain. In this sequential process, reconstruction artifacts at high acceleration factors and low SNR at high diffusion b-values degrade B0 estimation and limit the overall correction quality. We propose a physics-informed framework that jointly estimates the B0 field and distortion-free image directly from k-space data, without depending on an intermediate parallel-imaging reconstruction for the correction. The image and the B0 field are each represented as a superposition of Gaussian primitives embedded within an MRI physics forward model. The explicit, continuous parameterization captures both smooth regions and tissue boundaries and supports rotated-view EPI acquisitions without interpolation. The diffusion-weighted image is modeled as real and non-negative, with the image phase absorbed into a per-shot phase factor. Rotated views distribute distortions across multiple phase-encoding orientations, improving point spread function isotropy and providing stronger constraints for B0 estimation. On in vivo brain diffusion EPI, the proposed method attains the closest brain-boundary agreement with a distortion-free structural reference, with the largest improvement over sequential methods at high b-value and high acceleration. Extensive visual comparisons further show improved detail fidelity and noise suppression.
Wenqi Huang, Zhitao Li, Nan Wang +8
Chair for AI in Healthcare and Medicine, Technical University of Munich (TUM) and TUM University Hospital, and the School of Computation, Information and Technology, TUM, Munich, Germany · Innovation Academy for Precision Measurement Science and Technology, Chinese Academy of Sciences, Wuhan, China · Department of Radiology, Stanford University, Stanford, CA, USA +2
Functional magnetic resonance imaging (fMRI) utilizes echo-planar imaging (EPI) to capture blood-oxygen-level-dependent (BOLD) signals with high temporal resolution. However, EPI is inherently sensitive to magnetic field inhomogeneities, resulting in susceptibility-induced geometric distortions along the phase-encoding (PE) direction. To correct these distortions, conventional approaches rely on additional calibration scans, such as field maps or reverse PE acquisitions, which are not always available in practice. To overcome this limitation, we propose SACRED, a calibration scan-free susceptibility distortion correction framework that corrects geometric distortions via image translation-based registration using only a routinely acquired anatomical T1-weighted (T1w) image and a unidirectional PE BOLD image. SACRED employs an invertible neural network as the image translation backbone to bridge the contrast gap between BOLD and T1w images while enforcing structural consistency through a modality independent neighborhood descriptor. This design enables the use of a mono-contrast similarity objective to train the registration network in an unsupervised manner without requiring distortion-corrected BOLD images. In addition, we incorporate test-time adaptation (TTA) to further enhance performance on out-of-distribution (OOD) data at inference time. SACRED was evaluated on one in-distribution (ID) dataset and two OOD datasets, and was compared with representative fMRI distortion correction methods. The results demonstrate that SACRED significantly outperforms competing methods on both ID and OOD datasets, exhibiting robustness to scanner and population shifts, partly enabled by TTA. The code will be made publicly available upon acceptance.
Wooseung Kim, Sung-Hong Park
Department of Bio and Brain Engineering, Korea Advanced Institute of Science and Technology, Daejeon, Republic of Korea
Diffusion MRI (dMRI) relies on diffusion-weighted echo-planar imaging, which is highly susceptible to eddy-current-induced geometric distortions. These distortions vary across diffusion volumes according to gradient strength and direction, causing between-volume misalignment that can bias downstream microstructural analyses. Current state-of-the-art correction methods, such as FSL Eddy, achieve high-quality correction through iterative prediction-correction schemes but are computationally expensive. We propose Eddeep, a deep-learning framework for fast eddy-current distortion correction in dMRI. Eddeep decomposes the problem into two stages. First, a supervised image translation network standardises the appearance of diffusion-weighted and b=0 images, removing contrast differences that hinder reliable registration. Second, an unsupervised registration network estimates both eddy-current distortion and between-volume head motion parameters under a physics-constrained quadratic distortion model, enabling correction in a single forward pass. The method was trained on UK Biobank data and evaluated on both in-domain (UK Biobank) and out-of-domain (Memodyn) datasets. Across a range of complementary metrics, including between-volume jitter, diffusion kurtosis imaging residuals, signal irregularity, and mutual information, Eddeep achieved correction quality comparable to that of FSL Eddy while substantially reducing inference time. These results demonstrate that deep learning can provide accurate and efficient eddy-current distortion correction without relying on iterative optimisation, supporting the development of faster diffusion MRI processing pipelines for large-scale studies and clinical deployment. The code is available at: https://github.com/CIG-UCL/eddeep.
Antoine Legouhy, Ross Callaghan, Yuchuan Qiao +4
Hawkes Institute & Department of Computer Science, University College London, London, UK · Institut Pasteur, Université Paris Cité, Unité de Neuroanatomie Appliquée et Théorique, France · AINOSTICS ltd., Manchester, UK +3