Skip to main content
Log in

An improved CapsNet applied to recognition of 3D vertebral images

  • Published:
Applied Intelligence Aims and scope Submit manuscript

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.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11
Fig. 12
Fig. 13

Similar content being viewed by others

Notes

  1. The source website of this picture is : http://spineweb.digitalimaginggroup.ca/spineweb/uploads/Main/uw_database_overview.png

References

  1. Burns JE (2015) Imaging of the spine: A medical and physical perspective. In: Spinal imaging and image analysis. Springer, pp 3–29

  2. 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

    Article  Google Scholar 

  3. 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

    Google Scholar 

  4. 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

    Article  Google Scholar 

  5. 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

    Article  Google Scholar 

  6. 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

  7. 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

  8. 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

  9. 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

    Article  Google Scholar 

  10. 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

  11. 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

    Article  Google Scholar 

  12. Sabour S, Frosst N, Hinton GE (2017) Dynamic routing between capsules. In: Advances in neural information processing systems, pp 3856–3866

  13. Xi E, Bing S, Jin Y (2017) Capsule network performance on complex data. arXiv:1712.03480

  14. Hinton GE, Sabour S, Frosst N (2018) Matrix capsules with em routing international conference on learning representations

  15. 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

  16. 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

  17. Wang Q, Xu C, Zhou Y (2018) An attention-based Bi-GRU-CapsNet model for hypernymy detection between compound entities. arXiv:1805.04827

  18. 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

  19. 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

  20. Kumar AD (2018) Novel deep learning model for traffic sign detection using capsule networks. arXiv:1805.04424

  21. 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

  22. 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

  23. Krizhevsky A, Sutskever I, Hinton GE (2012) Imagenet classification with deep convolutional neural networks. In: Advances in neural information processing systems, pp 1097–1105

  24. Simonyan K, Zisserman A (2014) Very deep convolutional networks for large-scale image recognition. arXiv:1409.1556

  25. 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

  26. 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

Download references

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

Authors

Corresponding author

Correspondence to Xing Huo.

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

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-020-01695-3

Keywords

Navigation