Abstract
Deep learning has drawn increasing attention in microstructure estimation with undersampled diffusion MRI (dMRI) data. A representative method is the hybrid graph transformer (HGT), which achieves promising performance by integrating q-space graph learning and x-space transformer learning into a unified framework. However, this method overlooks the 3D spatial information as it relies on training with 2D slices. To address this limitation, we propose 3D hybrid graph transformer (3D-HGT), an advanced microstructure estimation model capable of making full use of 3D spatial information and angular information. To tackle the large computation burden associated with 3D x-space learning, we propose an efficient q-space learning model based on simplified graph neural networks. Furthermore, we propose a 3D x-space learning module based on the transformer. Extensive experiments on data from the human connectome project show that our 3D-HGT outperforms state-of-the-art methods, including HGT, in both quantitative and qualitative evaluations.
J. Yang and H. Jiang–Contributed equally to this work. This work was supported in part by the National Natural Science Foundation of China through Grants 62201465 and 62171377, and the Natural Science Foundation of Heilongjiang Province through Grant LH2021F046. P.-T. Yap was supported in part by the United States National Institutes of Health (NIH) through Grants MH125479 and EB008374.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Ba, J.L., Kiros, J.R., Hinton, G.E.: Layer normalization. arXiv preprint arXiv:1607.06450 (2016)
Chen, G., Dong, B., Zhang, Y., Lin, W., Shen, D., Yap, P.T.: Denoising of diffusion MRI data via graph framelet matching in x-q space. IEEE Trans. Med. Imaging 38(12), 2838–2848 (2019)
Chen, G., Dong, B., Zhang, Y., Lin, W., Shen, D., Yap, P.T.: XQ-SR: joint x-q space super-resolution with application to infant diffusion MRI. Med. Image Anal. 57, 44–55 (2019)
Chen, G., et al.: Estimating tissue microstructure with undersampled diffusion data via graph convolutional neural networks. In: Martel, A.L., et al. (eds.) MICCAI 2020. LNCS, vol. 12267, pp. 280–290. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-59728-3_28
Chen, G., et al.: Hybrid graph transformer for tissue microstructure estimation with undersampled diffusion MRI data. In: International Conference on Medical Image Computing and Computer-Assisted Intervention, LNCS, vol. 13431, pp. 113–122. Springer, Cham (2022). https://doi.org/10.1007/978-3-031-16431-6_11
Chen, G., Wu, Y., Shen, D., Yap, P.T.: Noise reduction in diffusion MRI using non-local self-similar information in joint x-q space. Med. Image Anal. 53, 79–94 (2019)
Chen, H., Dou, Q., Yu, L., Qin, J., Heng, P.A.: VoxResNet: deep voxelwise residual networks for brain segmentation from 3D MR images. NeuroImage 170, 446–455 (2018)
Daducci, A., et al.: Accelerated microstructure imaging via convex optimization (AMICO) from diffusion MRI data. NeuroImage 105, 32–44 (2015)
Dou, Q., et al.: Automatic detection of cerebral microbleeds from MR images via 3D convolutional neural networks. IEEE Trans. Med. Imaging 35(5), 1182–1195 (2016)
Falk, T., et al.: U-net: deep learning for cell counting, detection, and morphometry. Nat. Methods 16(1), 67–70 (2019)
Gibbons, E.K., et al.: Simultaneous NODDI and GFA parameter map generation from subsampled q-space imaging using deep learning. Magnet. Resonan. Med. 81(4), 2399–2411 (2019)
Golkov, V., et al.: q-Space deep learning: twelve-fold shorter and model-free diffusion MRI scans. IEEE Trans. Med. Imaging 35(5), 1344–1351 (2016)
Hendrycks, D., Gimpel, K.: Gaussian error linear units (GELUs). arXiv preprint arXiv:1606.08415 (2016)
Hong, Y., Chen, G., Yap, P.-T., Shen, D.: Multifold acceleration of diffusion MRI via deep learning reconstruction from slice-undersampled data. In: Chung, A.C.S., Gee, J.C., Yushkevich, P.A., Bao, S. (eds.) IPMI 2019. LNCS, vol. 11492, pp. 530–541. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-20351-1_41
Jensen, J.H., Helpern, J.A., Ramani, A., Lu, H., Kaczynski, K.: Diffusional kurtosis imaging: the quantification of non-gaussian water diffusion by means of magnetic resonance imaging. Magnet. Resonan. Med. 53(6), 1432–1440 (2005)
Kaden, E., Kruggel, F., Alexander, D.C.: Quantitative mapping of the per-axon diffusion coefficients in brain white matter. Magnet. Resonan. Med. 75(4), 1752–1763 (2016)
Liu, Z., et al.: Swin transformer: hierarchical vision transformer using shifted windows. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 10012–10022 (2021)
Tian, Q., et al.: DeepDTI: high-fidelity six-direction diffusion tensor imaging using deep learning. NeuroImage 219, 117017 (2020)
Van Essen, D.C., et al.: The WU-Minn human connectome project: an overview. NeuroImage 80, 62–79 (2013)
Wu, F., Souza, A., Zhang, T., Fifty, C., Yu, T., Weinberger, K.: Simplifying graph convolutional networks. In: International Conference on Machine Learning, pp. 6861–6871. PMLR (2019)
Ye, C.: Estimation of tissue microstructure using a deep network inspired by a sparse reconstruction framework. In: Niethammer, M., et al. (eds.) IPMI 2017. LNCS, vol. 10265, pp. 466–477. Springer, Cham (2017). https://doi.org/10.1007/978-3-319-59050-9_37
Ye, C.: Tissue microstructure estimation using a deep network inspired by a dictionary-based framework. Med. Image Anal. 42, 288–299 (2017)
Zhang, H., Schneider, T., Wheeler-Kingshott, C.A., Alexander, D.C.: NODDI: practical in vivo neurite orientation dispersion and density imaging of the human brain. NeuroImage 61(4), 1000–1016 (2012)
Zheng, T., et al.: A microstructure estimation transformer inspired by sparse representation for diffusion MRI. Med. Image Anal. 86, 102788 (2023)
Zhou, H.Y., et al.: nnFormer: volumetric medical image segmentation via a 3D transformer. IEEE Trans. Image Process. (2023)
Zhou, Z., Rahman Siddiquee, M.M., Tajbakhsh, N., Liang, J.: UNet++: a nested U-Net architecture for medical image segmentation. In: Stoyanov, D., et al. (eds.) DLMIA/ML-CDS -2018. LNCS, vol. 11045, pp. 3–11. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-00889-5_1
Author information
Authors and Affiliations
Corresponding authors
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Yang, J. et al. (2023). Towards Accurate Microstructure Estimation via 3D Hybrid Graph Transformer. In: Greenspan, H., et al. Medical Image Computing and Computer Assisted Intervention – MICCAI 2023. MICCAI 2023. Lecture Notes in Computer Science, vol 14227. Springer, Cham. https://doi.org/10.1007/978-3-031-43993-3_3
Download citation
DOI: https://doi.org/10.1007/978-3-031-43993-3_3
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-43992-6
Online ISBN: 978-3-031-43993-3
eBook Packages: Computer ScienceComputer Science (R0)