Abstract
Different deep learning-based architectures have been developed for medical image registration in the last few years. The architectures of these methods are complex and require a considerable amount of memory. The scalability of the architectures is limited due to their compatibility with low or moderate memory devices. The deformable image registration attained attention in the field of medical image registration. The uprising in deep learning has paved the path for more sophisticated solutions to the many medical imaging problems. We have proposed the resource-efficient and structure-preserving network (RESPNet) to solve medical image registration issues. RESPNet is convolutional neural networks (CNNs)-based architecture to predict the deformation vector field (DVF), which signifies the displacement of each pixel in all directions. We have developed the three CNNs modules (i.e., StraightCNN, UpCNN, and DownCNN) to reduce parameter size and preserve the structure of the images. In this paper, the architecture of RESPNet utilizes the structural properties of images for 2D registration of the retina images and 3D registration of brain magnetic resonance (MR) images. The proposed architecture requires less than 25% memory compared to the current state-of-the-art methods and can be trained in 6-8 hours on a 13 GB GPU to produce the results. The dice score of 2D and 3D images registration is 0.8784 and 0.7515, respectively. The suggested architecture reduces the memory requirements by more than 75% and achieves better performance in dice score, mutual information, and processing time. We have developed a memory-efficient deep learning architecture for medical image registration. This architecture can be employed to register 2D and 3D images efficiently.











Similar content being viewed by others
Data Availability
The datasets used in this work will be available from the corresponding author on reasonable request.
References
Krizhevsky A, Sutskever I, Hinton GE (2012) Imagenet classification with deep convolutional neural networks. In: F. Pereira, C. J. C. Burges, L. Bottou, and K. Q. Weinberger, editors, Advances in Neural Information Processing Systems 25, pp. 1097–1105. Curran Associates, Inc.,
Mamoshina P, Vieira A, Putin E, Zhavoronkov A (2016) Applications of deep learning in biomedicine. Mol Pharm 13(5):1445–1454
Shanker R, Bhattacharya M (2022) Classification of brain mr images using modified version of simplified pulse-coupled neural network and linear programming twin support vector machines. J Supercomput 78:13831–13863
Sotiras A, Davatzikos C, Paragios N (2013) Deformable medical image registration: A survey. IEEE Trans Med Imaging 32(7):1153
Cortés X, Serratosa F (2015) An interactive method for the image alignment problem based on partially supervised correspondence. Expert Syst Appl 42(1):179–192
Cao W (2021) Applying image registration algorithm combined with cnn model to video image stitching. J Supercomput 77(12):13879–13896
Wyawahare MV, Patil PM, Abhyankar HK et al (2009) Image registration techniques: an overview. Int J Signal Process, Image Process Pattern Recognition 2(3):11–28
Crum WR, Hartkens T, Hill DLG (2004) Non-rigid image registration: theory and practice. Br J Radiol 77(suppl–2):S140–S153
Seungjong O, Kim S (2017) Deformable image registration in radiation therapy. Radiat Oncol J 35(2):101
Lundervold AS, Lundervold Arvid (2019) An overview of deep learning in medical imaging focusing on mri. Z Med Phys 29(2):102–127
Zaharchuk G, Gong E, Wintermark M, Rubin D, Langlotz CP (2018) Deep learning in neuroradiology. Am J Neuroradiol 39(10):1776–1784
Johnson HJ, Christensen Gary E (2002) Consistent landmark and intensity-based image registration. IEEE Trans Med Imaging 21(5):450–461
Khader M, Hamza AB (2012) An information-theoretic method for multimodality medical image registration. Expert Syst Appl 39(5):5548–5556
Hill DLG, Batchelor PG, Holden M, Hawkes David J (2001) Medical image registration. Phys Med Biol 46(3):R1
Maes F, Collignon A, Vandermeulen D, Marchal G, Suetens P (1997) Multimodality image registration by maximization of mutual information. IEEE Trans Med Imaging 16(2):187–198
Bhattacharya M, Das A (2007) Multi resolution medical image registration using maximization of mutual information & optimization by genetic algorithm. In: 2007 IEEE Nuclear Science Symposium Conference Record, vol 4, pp. 2961–2964. IEEE,
Devarriya D, Gulati C, Mansharamani V, Sakalle A, Bhardwaj A (2019) Unbalanced breast cancer data classification using novel fitness functions in genetic programming. Expert Syst Appl, p. 112866
Das A, Bhattacharya M (2011) Affine-based registration of ct and mr modality images of human brain using multiresolution approaches: comparative study on genetic algorithm and particle swarm optimization. Neural Comput Appl 20(2):223–237
Zheng Z, Cao W, Ye D, Cao G, Lian D (2022) Multi-strategy mutual learning network for deformable medical image registration. Neurocomputing
An Q, Chen S, Liang J, Snyder M, Di Y (2022) Evaluation of dir schemes on tumor/organ with progressive shrinkage: accuracy of tumor/organ internal tissue tracking during the radiation treatment. Radiotherapy and Oncol
Yang T, Bai X, Cui X, Gong Y, Li L (2022) Transdir: Deformable imaging registration network based on transformer to improve the feature extraction ability. Med Phys 49(2):952–965
Amor BB, Arguillère S, Shao L (2021) Resnet-lddmm: advancing the lddmm framework using deep residual networks. arXiv preprint arXiv:2102.07951
Shanker R, Bhattacharya Mahua (2019) Brain tumor segmentation of normal and lesion tissues using hybrid clustering and hierarchical centroid shape descriptor. Comput Methods in Biomechan Biomed Eng: Imaging Vis 7(5–6):676–689
Shanker R, Bhattacharya M (2020) An automated computer-aided diagnosis system for classification of mr images using texture features and gbest-guided gravitational search algorithm. Biocybernet Biomed Eng 40(2):815–835
Rawat W, Wang Z (2017) Deep convolutional neural networks for image classification: a comprehensive review. Neural Comput 29(9):2352–2449
Kearney V, Haaf S, Sudhyadhom A, Valdes G, Solberg TD (2018) An unsupervised convolutional neural network-based algorithm for deformable image registration. Phys Med Biol 63(18):185017
Chen Z, Zekai X, Yi W, Yang X, Hou W, Ding M, Granichin O (2019) Real-time and multimodal brain slice-to-volume registration using cnn. Expert Syst Appl 133:86–96
Yang X, Kwitt R, Styner M, Niethammer Marc (2017) Quicksilver: Fast predictive image registration-a deep learning approach. Neuroimage 158:378–396
Cao Y, Miller M, Winslow RL, Younes L (2005) Large deformation diffeomorphic metric mapping of vector fields. IEEE Trans Med Imaging 24(9):1216–1230
Balakrishnan G, Zhao A, Sabuncu MR, Guttag J, Dalca AV (2019) Voxelmorph: a learning framework for deformable medical image registration. IEEE Trans Med Imag
Köpüklü O, Kose N, Gunduz A, Rigoll G (2019) Resource efficient 3d convolutional neural networks. arXiv preprint arXiv:1904.02422
Speiser A, Yan J, Archer EW, Buesing L, Turaga SC, Macke JH (2017) Fast amortized inference of neural activity from calcium imaging data with variational autoencoders. In: Advances in Neural Information Processing Systems, pp. 4024–4034
Hernandez-Matas C, Zabulis X, Triantafyllou A, Anyfanti P, Douma S, Argyros Antonis A (2017) Fire: fundus image registration dataset. J Model Ophthalmol 1(4):16–28
Gousias IS, Edwards AD, Rutherford MA, Counsell SJ, Hajnal JoV, Rueckert D, Hammers A (2012) Magnetic resonance imaging of the newborn brain: manual segmentation of labelled atlases in term-born and preterm infants. Neuroimage 62(3):1499–1509
Jaderberg M, Simonyan K, Zisserman A, et al. (2015) Spatial transformer networks. In: Advances in neural information processing systems, pp. 2017–2025
Hecht-Nielsen R (1992) Theory of the backpropagation neural network. In: Neural networks for perception, pp. 65–93. Elsevier
Kingma Diederik P, Ba Jimmy (2014) Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980
Hua B-S, Tran M-K, Yeung S-K (2018) Pointwise convolutional neural networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 984–993
Huang Gao, Liu S, Van der ML, Weinberger KQ (2018) Condensenet: An efficient densenet using learned group convolutions. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2752–2761
Szegedy C, Ioffe S, Vanhoucke V, Alemi AA (2017) Inception-v4, inception-resnet and the impact of residual connections on learning. In: Thirty-First AAAI Conference on Artificial Intelligence
Iandola FN, Han S, Moskewicz MW, Ashraf K, Dally WJ, Keutzer K (2016) Squeezenet: Alexnet-level accuracy with 50x fewer parameters and \(< 0.5 mb\) model size. arXiv preprint arXiv:1602.07360,
Ioffe S, Szegedy C (2015) Batch normalization: Accelerating deep network training by reducing internal covariate shift. arXiv preprint arXiv:1502.03167
Li Y, Yuan Y (2017) Convergence analysis of two-layer neural networks with relu activation. In: Advances in Neural Information Processing Systems, pp. 597–607
Zhang X, Zhou X, Lin M, Sun J (2018) Shufflenet: An extremely efficient convolutional neural network for mobile devices. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 6848–6856
Boureau Y-L, Ponce J, LeCun Y (2010) A theoretical analysis of feature pooling in visual recognition. In: Proceedings of the 27th international conference on machine learning (ICML-10), pp. 111–118
Sun B, Zhou S (2012) Study on the 3d interpolation models used in color conversion. Int J Eng Technol 4(1):57
Keane RD, Adrian RJ (1992) Theory of cross-correlation analysis of piv images. Appl Sci Res 49(3):191–215
Zhang L, Zhang L, Mou X, Zhang D (2011) Fsim: A feature similarity index for image quality assessment. IEEE Trans Image Process 20(8):2378–2386
Lowekamp BC, Chen DT, Ibáñez L, Blezek D (2013) The design of simpleitk. Front Neuroinform 7:45
Garyfallidis E, Brett M, Amirbekian B, Rokem A, Van Der Walt S, Descoteaux M, Nimmo-Smith I (2014) Dipy, a library for the analysis of diffusion mri data. Front Neuroinform 8:8
Rigaud B, Simon A, Castelli J, Gobeli M, Ospina Arango J-D, Cazoulat G, Henry O, Haigron P, De Crevoisier R (2015) Evaluation of deformable image registration methods for dose monitoring in head and neck radiotherapy. BioMed Res Int
Kirby N, Chuang C, Ueda U, Pouliot J (2013) The need for application-based adaptation of deformable image registration. Med Phys 40(1):011702
Haussler D, Opper Manfred et al (1997) Mutual information, metric entropy and cumulative relative entropy risk. Ann Stat 25(6):2451–2492
Baez JC, Fritz T, Leinster T (2011) A characterization of entropy in terms of information loss. Entropy 13(11):1945–1957
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that they have no conflict of interest.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Shanker, R., Sankesara, H., Nagar, S. et al. RESPNet: resource-efficient and structure-preserving network for deformable image registration. J Supercomput 79, 4713–4736 (2023). https://doi.org/10.1007/s11227-022-04840-6
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11227-022-04840-6