🤖 AI Summary
This work addresses the challenge of eddy current–induced distortions in diffusion MRI, which cause misalignment across multi-shell images and compromise the accuracy of microstructural analysis. The authors propose the first end-to-end deep learning framework for joint correction of eddy currents and subject motion. In the first stage, a supervised image translation network harmonizes image contrast across shells; in the second stage, an unsupervised registration module incorporating physical constraints simultaneously estimates distortion and motion parameters, enabling full correction in a single forward pass. By circumventing conventional iterative optimization, the method achieves correction accuracy comparable to FSL Eddy while offering substantially faster inference. Trained on UK Biobank data, the approach is well-suited for large-scale studies and clinical deployment.
📝 Abstract
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.