Nothing Special   »   [go: up one dir, main page]

skip to main content
research-article

Unsupervised deformable image registration network for 3D medical images

Published: 01 January 2022 Publication History

Abstract

Image registration aims to establish an active correspondence between a pair of images. Such correspondence is critical for many significant applications, such as image fusion, tumor growth monitoring, and atlas generation. In this study, we propose an unsupervised deformable image registration network (UDIR-Net) for 3D medical images. The proposed UDIR-Net is designed in an encoder-decoder architecture and directly estimates the complex deformation field between input pairwise images without any supervised information. In particular, we recalibrate the feature slice of each feature map that is propagated between the encoder and the decoder in accordance with the importance of each feature slice and the correlation between feature slices. This method enhances the representational power of feature maps. To achieve efficient and robust training, we design a novel hierarchical loss function that evaluates multiscale similarity loss between registered image pairs. The proposed UDIR-Net is tested on different public magnetic resonance image datasets of the human brain. Experimental results show that UDIR-Net exhibits competitive performance against several state-of-the-art methods.

References

[1]
Abadi M, Agarwal A, Barham P, Brevdo E, Chen Z, Citro C, Corrado GS, Davis A, Dean J, Devin M et al (2016) Tensorflow: Large-scale machine learning on heterogeneous distributed systems. arXiv:160304467
[2]
Avants BB, Epstein CL, Grossman M, and Gee JC Symmetric diffeomorphic image registration with cross-correlation: evaluating automated labeling of elderly and neurodegenerative brain Med Image Anal 2008 12 1 26-41
[3]
Avants BB, Tustison NJ, Song G, Cook PA, Klein A, and Gee JC A reproducible evaluation of ants similarity metric performance in brain image registration Neuroimage 2011 54 3 2033-2044
[4]
Balakrishnan G, Zhao A, Sabuncu MR, Guttag J, Dalca AV (2018) An unsupervised learning model for deformable medical image registration. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp 9252–9260
[5]
Balakrishnan G, Zhao A, Sabuncu MR, Guttag J, and Dalca AV Voxelmorph: a learning framework for deformable medical image registration IEEE Trans Med Imaging 2019 38 8 1788-1800
[6]
Cao X, Yang J, Zhang J, Wang Q, Yap PT, and Shen D Deformable image registration using a cue-aware deep regression network IEEE Trans Biomed Eng 2018 65 9 1900-1911
[7]
Chee E, Wu J (2018) Airnet: Self-supervised affine registration for 3d medical images using neural networks. arXiv:181002583
[8]
Cheng X, Zhang L, and Zheng Y Deep similarity learning for multimodal medical images Comput Methods Biomech Biomed Eng Imaging Vis 2018 6 3 248-252
[9]
Chollet F (2015) Keras. https://keras.io
[10]
Dalca AV, Balakrishnan G, Guttag J, Sabuncu MR (2018) Unsupervised learning for fast probabilistic diffeomorphic registration. In: International conference on medical image computing and computer-assisted intervention, pp 729–738
[11]
Di Martino A, Yan CG, Li Q, Denio E, Castellanos FX, Alaerts K, Anderson JS, Assaf M, Bookheimer SY, Dapretto M, et al. The autism brain imaging data exchange: towards a large-scale evaluation of the intrinsic brain architecture in autism Mol Psychiatry 2014 19 6 659-667
[12]
Eppenhof KA and Pluim JP Error estimation of deformable image registration of pulmonary ct scans using convolutional neural networks J Med Imaging 2018 5 2 024003
[13]
Fan J, Cao X, Yap PT, and Shen D Birnet: Brain image registration using dual-supervised fully convolutional networks Med Image Anal 2019 54 193-206
[14]
Fischl B Freesurfer Neuroimage 2012 62 2 774-781
[15]
Haskins G, Kruger U, Yan P (2019) Deep learning in medical image registration: A survey. arXiv:190302026
[16]
Hu J, Shen L, Sun G (2018) Squeeze-and-excitation networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp 7132–7141
[17]
Jaderberg M, Simonyan K, Zisserman A et al (2015) Spatial transformer networks. Advances in neural information processing systems, 2017–2025
[18]
Khader M, Schiavi E, and Hamza AB A multicomponent approach to nonrigid registration of diffusion tensor images Appl Intell 2017 46 2 241-253
[19]
Kingma DP, Ba J (2014) Adam: A method for stochastic optimization. arXiv:14126980
[20]
Krebs J, Mansi T, Mailhé B, Ayache N, Delingette H (2018) Unsupervised probabilistic deformation modeling for robust diffeomorphic registration. Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, 101–109
[21]
Li H and Fan Y Non-rigid image registration using self-supervised fully convolutional networks without training data IEEE Int Symp Biomed Imaging 2018 2018 1075-1078
[22]
Li H, Xiong P, An J, Wang L (2018) Pyramid attention network for semantic segmentation. arXiv:180510180
[23]
Liao R, Miao S, de Tournemire P, Grbic S, Kamen A, Mansi t, Comaniciu D (2017) An artificial agent for robust image registration. Thirty-First AAAI Conference on Artificial Intelligence
[24]
Litjens G, Kooi T, Bejnordi BE, Setio AAA, Ciompi F, Ghafoorian M, Van Der Laak JA, Van Ginneken B, and Sánchez CI A survey on deep learning in medical image analysis Med Image Anal 2017 42 60-88
[25]
Maes F, Collignon A, Vandermeulen D, Marchal G, and Suetens P Multimodality image registration by maximization of mutual information IEEE Trans Med Imaging 1997 16 2 187-198
[26]
Marcus DS, Wang TH, Parker J, Csernansky JG, Morris JC, and Buckner RL Open access series of imaging studies (oasis): cross-sectional mri data in young, middle aged, nondemented, and demented older adults J Cogn Neurosci 2007 19 9 1498-1507
[27]
Miao S, Wang ZJ, and Liao R A cnn regression approach for real-time 2d/3d registration IEEE Trans Med Imaging 2016 35 5 1352-1363
[28]
Milham MP, Fair D, Mennes M, Mostofsky SH, et al. The adhd-200 consortium: a model to advance the translational potential of neuroimaging in clinical neuroscience Front Syst Neurosci 2012 6 62
[29]
Mueller SG, Weiner MW, Thal LJ, Petersen RC, Jack CR, Jagust W, Trojanowski JQ, Toga AW, and Beckett L Ways toward an early diagnosis in alzheimer’s disease: the alzheimer’s disease neuroimaging initiative (adni) Alzheimer’s & Dementia 2005 1 1 55-66
[30]
Ronneberger O, Fischer P, Brox T (2015) U-net: Convolutional networks for biomedical image segmentation. In: International conference on medical image computing and computer-assisted intervention, pp 234–241
[31]
Sheikhjafari A, Noga M, Punithakumar K, Ray N (2018) Unsupervised deformable image registration with fully connected generative neural network. In: International conference on Medical Imaging with Deep Learning
[32]
Shen D, Wu G, and Suk HI Deep learning in medical image analysis Annu Rev Biomed Eng 2017 19 221-248
[33]
Sokooti H, de Vos B, Berendsen F, Lelieveldt BP, Išgum I, Staring M (2017) Nonrigid image registration using multi-scale 3d convolutional neural networks. International Conference on Medical Image Computing and Computer-Assisted Intervention, 232–239
[34]
Sotiras A, Davatzikos C, and Paragios N Deformable medical image registration: a survey IEEE Trans Med Imaging 2013 32 7 1153-1190
[35]
de Vos BD, Berendsen FF, Viergever MA, Staring M, Išgum I (2017) End-to-end unsupervised deformable image registration with a convolutional neural network. Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, 204–212
[36]
Woo S, Park J, Lee JY, So Kweon I (2018) Cbam: Convolutional block attention module, 3–19
[37]
Wu G, Kim M, Wang Q, Gao Y, Liao S, Shen D (2013) Unsupervised deep feature learning for deformable registration of mr brain images. In: International conference on medical image computing and computer-assisted intervention, pp 649–656
[38]
Wu G, Kim M, Wang Q, Munsell BC, and Shen D Scalable high-performance image registration framework by unsupervised deep feature representations learning IEEE Trans Biomed Eng 2015 63 7 1505-1516
[39]
Yang J, Yu K, Gong Y, Huang T (2009) Linear spatial pyramid matching using sparse coding for image classification. In: 2009 IEEE Conference on computer vision and pattern recognition, pp 1794–1801
[40]
Zhang Z and Sejdić E Radiological images and machine learning: trends, perspectives, and prospects Comput Biol Med 2019 108 354-370

Cited By

View all
  • (2023)An Unsupervised End-to-End Recursive Cascaded Parallel Network for Image RegistrationNeural Processing Letters10.1007/s11063-023-11311-355:6(8255-8268)Online publication date: 1-Dec-2023
  • (2023)Closed-loop feedback registration for consecutive images of moving flexible targetsApplied Intelligence10.1007/s10489-022-04068-053:9(10647-10667)Online publication date: 1-May-2023

Index Terms

  1. Unsupervised deformable image registration network for 3D medical images
            Index terms have been assigned to the content through auto-classification.

            Recommendations

            Comments

            Please enable JavaScript to view thecomments powered by Disqus.

            Information & Contributors

            Information

            Published In

            cover image Applied Intelligence
            Applied Intelligence  Volume 52, Issue 1
            Jan 2022
            1143 pages

            Publisher

            Kluwer Academic Publishers

            United States

            Publication History

            Published: 01 January 2022
            Accepted: 05 January 2021

            Author Tags

            1. Image registration
            2. Convolutional neural network
            3. Unsupervised learning
            4. Brain MR images

            Qualifiers

            • Research-article

            Funding Sources

            Contributors

            Other Metrics

            Bibliometrics & Citations

            Bibliometrics

            Article Metrics

            • Downloads (Last 12 months)0
            • Downloads (Last 6 weeks)0
            Reflects downloads up to 14 Dec 2024

            Other Metrics

            Citations

            Cited By

            View all
            • (2023)An Unsupervised End-to-End Recursive Cascaded Parallel Network for Image RegistrationNeural Processing Letters10.1007/s11063-023-11311-355:6(8255-8268)Online publication date: 1-Dec-2023
            • (2023)Closed-loop feedback registration for consecutive images of moving flexible targetsApplied Intelligence10.1007/s10489-022-04068-053:9(10647-10667)Online publication date: 1-May-2023

            View Options

            View options

            Media

            Figures

            Other

            Tables

            Share

            Share

            Share this Publication link

            Share on social media