Abstract
Deep learning is currently widely applied in medical image processing and has achieved good results. However, recognizing vertebrae via image processing remains a challenging problem due to their complex spatial structures. CapsNet is a newly proposed network whose characteristics compensate for some shortcomings of traditional CNNs, and it has been shown to perform well on many tasks, including medical image recognition. In this paper, we applied a modified CapsNet to recognise 3D vertebral images by introducing an RNN module into CapsNet to further enhance its learning ability. This new network is called RNNinCaps, and it achieves the highest recognition performance on 3D vertebral images (the average accuracy of RNNinCaps exceeds the accuracy of the original CapsNet by 46.2% and that of a traditional CNN by 12.6%). RNNinCaps also performs better than several mainstream networks. RNNinCaps can promotes CapsNet’s application in the field of 3D medical image recognition.
Similar content being viewed by others
Notes
The source website of this picture is : http://spineweb.digitalimaginggroup.ca/spineweb/uploads/Main/uw_database_overview.png
References
Burns JE (2015) Imaging of the spine: A medical and physical perspective. In: Spinal imaging and image analysis. Springer, pp 3–29
Zhao X, Wu Y, Song G, Li Z, Zhang Y, Fan Y (2018) A deep learning model integrating fcnns and crfs for brain tumor segmentation. Medical Image Analysis 43:98–111
Rathi VGP, Palani S (2015) Brain tumor detection and classification using deep learning classifier on mri images. Res J Appl Sci Eng Technol 10(2):177–187
Zhang Q, Xiao Y, Dai W, Suo J, Wang C, Shi J, Zheng H (2016) Deep learning based classification of breast tumors with shear-wave elastography. Ultrasonics 72:150–157
Bejnordi BE, Veta M, Van Diest PJ, Van Ginneken B, Karssemeijer N, Litjens G, Van Der Laak JA, Hermsen M, Manson QF, Balkenhol M, et al. (2017) Diagnostic assessment of deep learning algorithms for detection of lymph node metastases in women with breast cancer. Jama 318(22):2199–2210
Sirinukunwattana K, Raza SEA, Tsang YW, Snead D, Cree I, Rajpoot N (2015) A spatially constrained deep learning framework for detection of epithelial tumor nuclei in cancer histology images. In: International workshop on patch-based techniques in medical imaging. Springer, pp 154–162
Rajpurkar P, Irvin J, Zhu K, Yang B, Mehta H, Duan T, Ding D, Bagul A, Langlotz C, Shpanskaya K, et al. (2017) Chexnet: Radiologist-level pneumoni detection on chest x-rays with deep learning. arXiv:1711.05225
Suzani A, Seitel A, Liu Y, Fels S, Rohling R, Abolmaesumi P (2015) Fast automatic vertebrae detection and localization in pathological ct scans-a deep learning approach. In: International conference on medical image computing and computer-assisted intervention. Springer, pp 678–686
Suzani A, Rasoulian A, Seitel A, Fels SS, Rohling R, Abolmaesumi P (2015) Deep learning for automatic localization, identification, and segmentation of vertebral bodies in volumetric mr images. Proc SPIE 9415:941514
Chen H, Shen C, Qin J, Ni D, Shi L, Cheng JC, Heng PA (2015) Automatic localization and identification of vertebrae in spine ct via a joint learning model with deep neural networks. In: International conference on medical image computing and computer-assisted intervention. Springer, pp 515–522
Liao H, Mesfin A, Luo J (2018) Joint vertebrae identification and localization in spinal ct images by combining short-and long-range contextual information. IEEE Trans Med Imaging 37(5):1266–1275
Sabour S, Frosst N, Hinton GE (2017) Dynamic routing between capsules. In: Advances in neural information processing systems, pp 3856–3866
Xi E, Bing S, Jin Y (2017) Capsule network performance on complex data. arXiv:1712.03480
Hinton GE, Sabour S, Frosst N (2018) Matrix capsules with em routing international conference on learning representations
Jiménez-Sánchez A, Albarqouni S, Mateus D (2018) Capsule networks against medical imaging data challenges. In: Intravascular imaging and computer assisted stenting and large-scale annotation of biomedical data and expert label synthesis. Springer, pp 150–160
Mobiny A, Van Nguyen H (2018) Fast capsnet for lung cancer screening. In: International conference on medical image computing and computer-assisted intervention. Springer, pp 741–749
Wang Q, Xu C, Zhou Y (2018) An attention-based Bi-GRU-CapsNet model for hypernymy detection between compound entities. arXiv:1805.04827
Afshar P, Mohammadi A, Plataniotis KN (2018) Brain tumor type classification via capsule networks. In: 2018 25th IEEE International Conference on Image Processing (ICIP). IEEE, pp 3129–3133
Xiao L, Zhang H, Chen W, Wang Y, Jin Y (2018) Mcapsnet: Capsule network for text with multi-task learning. In: Proceedings of the 2018 conference on empirical methods in natural language processing, pp 4565–4574
Kumar AD (2018) Novel deep learning model for traffic sign detection using capsule networks. arXiv:1805.04424
Sak H, Senior AW, Beaufays F (2014) Long short-term memory based recurrent neural network architectures for large vocabulary speech recognition. arXiv: Neural and Evolutionary Computing
Xingjian S, Chen Z, Wang H, Yeung DY, Wong WK, Woo WC (2015) Convolutional lstm network: A machine learning approach for precipitation nowcasting. In: Advances in neural information processing systems, pp 802–810
Krizhevsky A, Sutskever I, Hinton GE (2012) Imagenet classification with deep convolutional neural networks. In: Advances in neural information processing systems, pp 1097–1105
Simonyan K, Zisserman A (2014) Very deep convolutional networks for large-scale image recognition. arXiv:1409.1556
Huang G, Liu Z, Van Der Maaten L, Weinberger KQ (2017) Densely connected convolutional networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp 4700–4708
Tran D, Bourdev L, Fergus R, Torresani L, Paluri M (2015) Learning spatiotemporal features with 3d convolutional networks. In: Proceedings of the IEEE international conference on computer vision, pp 4489–4497
Acknowledgments
This work was supported by National Natural Science Foundation of China under Grant 61872407, Grant 61572167 and Grant 61571176.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interests
The authors declared that they have no conflicts of interest in this work. We declare that we do not have any commercial or associative interest that represents a conflict of interest in connection with the work submitted.
Additional information
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Wang, H., Shao, K. & Huo, X. An improved CapsNet applied to recognition of 3D vertebral images. Appl Intell 50, 3276–3290 (2020). https://doi.org/10.1007/s10489-020-01695-3
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-020-01695-3