Deep flow-net for EPI distortion estimation

Abstract

Introduction: Geometric distortions along the phase encoding direction caused by off-resonant spins are a major issue in EPI based functional and diffusion imaging. The widely used blip up/down approach estimates the underlying distortion field from a pair of images with inverted phase encoding direction. Typically, iterative methods are used to find a solution to the ill-posed problem of finding the displacement field that maps up/down acquisitions onto each other. Here, we explore the use of a deep convolutional network to estimate the displacement map from a pair of input images. Methods: We trained a deep convolutional U-net architecture that was previously used to estimate optic flow between moving images to learn to predict the distortion map from an input pair of distorted EPI acquisitions. During the training step, the network minimizes a loss function (similarity metric) that is calculated from corrected input image pairs. This approach does not require the explicit knowledge of the ground truth distortion map, which is difficult to get for real life data. Results: We used data from a total of Ntrain ​= ​22 healthy subjects to train our network. A separate dataset of Ntest ​= ​12 patients including some with abnormal findings and unseen acquisition modes, e.g. LR-encoding, coronal orientation) was reserved for testing and evaluation purposes. We compared our results to FSL’s topup function with default parameters that served as the gold standard. We found that our approach results in a correction accuracy that is virtually identical to the optimum found by an iterative search, but with reduced computational time. Conclusion: By using a deep convolutional network, we can reduce the processing time to a few seconds per volume, which is significantly faster than iterative approaches like FSL’s topup which takes around 10min on the same machine (but using only 1 CPU). This facilitates the use of a blip up/down scheme for all diffusion-weighted acquisitions and potential real-time EPI distortion correction without sacrificing accuracy.