In this study, we proposed a new approach to reduce streaking artifacts in quantitative susceptibility mapping via deep learning. It combined two convolutional neural networks to reduce streaking artifacts from classic threshold-based k-space division (TKD) . The proposed method achieved impressive performance both visually and statistically.
Purpose
In recent years, quantitative susceptibility mapping (QSM) has been developed as a novel MR technique to assess brain iron in vivo. However, QSM is an ill-conditioned inverse problem in calculating underlying magnetic susceptibility distribution from phase image. Threshold-based k-space division (TKD)1 is a direct method to solve this problem without explicit a priori knowledge. However, it suffers from severe streaking artifacts. Inspired by the recent achievements of deep learning in medical imaging, we proposed a two-stage convolutional neural network (CNN) to reduce streaking artifacts in QSM images. Firstly, we used a simple CNN to remove major artifacts in images produced by TKD; then another network with U-net architecture2 was used to learn structure information of images.We took a 3D simulated model of the brain3 as the true susceptibility distribution. The phase image $$$\phi\left(r\right)$$$ was synthesized by forward model4, where TE = 26ms, B0 = 3T, voxel size = 0.5mm$$$\times$$$0.5mm$$$\times$$$0.5mm and matrix size = 640$$$\times$$$640$$$\times$$$640. An initial estimation of the susceptibility map with streaking artifacts was reconstructed from phase image by TKD where truncated value t = 0.1 was used1. Since deep learning requires a large amount of samples to achieve satisfactory results, we augmented the data by rotating the brain image for different angles in 3D before synthesizing phase images. 1400 and 200 QSM 2D images were used for training and testing respectively, each cropped to 400$$$\times$$$400.
The flowchart of the proposed two-stage method is shown in Fig. 1. In the first stage, QSM images with streaking artifacts were input into the trained network1 as shown in Fig. 2(a) to get the artifacts images. Then, we obtained artifacts-free images by subtracting the artifacts images from QSM images with streaking artifacts. Then the Fourier-transformed artifacts-free images $$$\chi^{'}_{recon.}\left(k\right)$$$ was merged with original k-space $$$\chi_{ori.}\left(k\right)$$$. Finally, the merged k-space was inverse Fourier transformed to get the reconstructed images of the first stage. In the second stage, we used a U-net architecture network as shown in Fig. 2(b) to learn the end-to-end mapping from reconstructed images of the first stage to ground truth. Then we merged the k-space data to obtain the final reconstructed images.
During the training, we used stochastic gradient descent (SGD) to minimize the loss function (Euclidean distance function). Our CNN model was built with TensorFlow, and run on a personal computer powered with two NVIDIA Titan X GPUs for about 24 hours on training. It took only tens of milliseconds to reconstruct single slice image.
In this study, we used two CNN networks of different structures to reconstruct QSM images. According to the experimental results, the proposed method can significantly reduce streaking artifacts and preserve image details at the same time, and thus can effectively improve the image quality.
The study was supported by the key project of the National Natural Science Foundation of China (61731009). The authors would like to thank Dr. Sagar Buch and Dr. E. Mark Haacke for their discussion about the 3D brain model.
1. Wharton S, Schafer A, Bowtell R. Susceptibility mapping in the human brain using threshold-based k-space division. Magn Reson Med 2010;63:1292–1304.
2. Ronneberger O, Fischer P, Brox T. U-Net: convolutional networks for biomedical image segmentation. International Conference on Medical Image Computing and Computer-Assisted Intervention. Springer, Cham, 2015:234-241.
3. Buch S, Liu S, Neelavalli J, et al. Simulated 3D brain model to study the phase behavior of brain structures. In: Proceedings of the International Society for Magnetic Resonance in Medicine 20th Scientific Meeting, 2012; Melbourne, Australia. p. 2332.
4. Marques J, Bowtell R. Application of a Fourier-based method for rapid calculation of field inhomogeneity due to spatial variation of magnetic susceptibility. Concept Magn Reson B 2005;25B:65–78.