Skip to main content
Log in

Dilated Volumetric Network: an Enhanced Fully Convolutional Network for Volumetric Prostate Segmentation from Magnetic Resonance Imaging

  • APPLICATION PROBLEMS
  • Published:
Pattern Recognition and Image Analysis Aims and scope Submit manuscript

Abstract

Early detection of prostate cancer is crucial for its successful treatment. However, it is not always an easy task because of the various image capturing configurations, like acquisition protocols, magnetic field strength, presence/absence of endorectal coil, and resolution. The major bottleneck in the process is the delineation of the prostate boundary for its localization, which is required for the detection of abnormalities and performing radiotherapy accurately. Phenomenal development in Artificial Intelligence and Deep Learning has been contributing significantly to medical diagnostics using Computer Vision and the self-learning capabilities of Deep Learning has been explored to present a viable solution to automate this repetitive task of prostate segmentation. The previous approaches of 2D segmentation do not capture volumetric information and are very time consuming too. Hence, we have developed a Deep Learning based automated solution called DV-Net (Dilated Volumetric Network) for volumetric segmentation of prostate cancer. The proposed method considers the full prostate volume in 3D and requires minimal post-processing, which makes it less dependent on the type of input. We also focus on increasing the receptive field of the network and use deep supervision for better segmentation accuracy. Owing to all these features, DV-Net has shown to outperform the accuracy of the baseline V-Net model on the Prostate MR Image Segmentation (PROMISE) data set.

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.

Similar content being viewed by others

Notes

  1. https://www.wcrf.org/dietandcancer/prostate-cancer

  2. https://promise12.grand-challenge.org

  3. https://www.slicer.org

  4. link to segmentation results in video http://bit.ly/37TjBKr

REFERENCES

  1. J. G. E. Awad, “Prostate segmentation and regions of interest detection in transrectal ultrasound images,” PhD Thesis (Univ. of Waterloo, 2007).

  2. Yiqiang Zhan and Dinggang Shen, “Deformable segmentation of 3-D ultrasound prostate images using statistical texture matching method,” IEEE Trans. Med. Imaging 25, 256–272 (2006).

    Article  Google Scholar 

  3. L. A. Eskew, R. L. Bare, and D. L. Mccullough, “Systematic 5 region prostate biopsy is superior to sextant method for diagnosing carcinoma of the prostate,” J. Urol. 157 (1), 199–202 (1997).

    Google Scholar 

  4. F. Milletari, N. Navab, and S.-A. Ahmadi, “V-Net: Fully convolutional neural networks for volumetric medical image segmentation,” in 2016 Fourth International Conference on 3D Vision (3DV) (2016), pp. 565–571.

  5. E. Shelhamer, J. Long, and T. Darrell, “Fully convolutional networks for semantic segmentation,” in 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) (2015), pp. 3431–3440.

  6. O. Ronneberger, P. Fischer, and T. Brox, “U-Net: Convolutional networks for biomedical image segmentation,” arXiv (2015). arXiv:1505.04597 [cs.CV]

  7. Qikui Zhu, Bo Du, B. Turkbey, P. L. Choyke, and Pingkun Yan, “Deeply-supervised CNN for prostate segmentation,” in Proceedings of the International Joint Conference on Neural Networks (2017), pp. 178–84. https://doi.org/10.1109/IJCNN.2017.7965852

  8. Wanli Chen, Yue Zhang, Junjun He, Yu Qiao, Yifan Chen, Hongjian Shi, and Xiaoying Tang, “W-net: Bridged U-net for 2D medical image segmentation,” arXiv (2018). arXiv:1807.04459v1

  9. Lequan Yu, Xin Yang, Hao Chen, Jing Qin, and Pheng-Ann Heng, “Volumetric ConvNets with mixed residual connections for automated prostate segmentation from 3D MR images,” in Thirty-First AAAI Conference on Artificial Intelligence (2017), pp. 66–72.

  10. Dou, Qi, Lequan Yu, Hao Chen, Yueming Jin, Xin Yang, Jing Qin, and Pheng Ann Heng, “3D deeply supervised network for automated segmentation of volumetric medical images,” Med. Image Anal. 41, 40–54 (2017). https://doi.org/10.1016/j.media.2017.05.001

    Article  Google Scholar 

  11. Manu Goyal, Moi Hoon Yap, and Saeed Hassanpour, “Multi-class semantic segmentation of skin lesions via fully convolutional networks,” arXiv (2017). arXiv:1711.10449 [cs.CV]

  12. Chen-Yu Lee, Saining Xie, P. W. Gallagher, Zhengyou Zhang, and Zhuowen Tu, “Deeply-supervised nets,” in Proceedings of the Eighteenth International Conference on Artificial Intelligence and Statistics, Ed. by Guy Lebanon and S. V. N. Vishwanathan (2015), Vol. 38, pp. 562–570.

  13. K. Kamnitsas, C. Ledig, V. F. J. Newcombe, J. P. Simpson, A. D. Kane, D. K. Menon, D. Rueckert, and B. Glocker, “Efficient multi-scale 3D CNN with fully connected CRF for accurate brain lesion segmentation,” Med. Image Anal. 36, 61–78 (2017). https://doi.org/10.1016/j.media.2016.10.004

    Article  Google Scholar 

  14. Xide Xia and Brian Kulis, “W-Net: A deep model for fully unsupervised image segmentation,” arXiv (2017). arXiv:1711.08506 [cs.CV]

  15. Xin Yang, Lequan Yu, Lingyun Wu, Yi Wang, Dong Ni, Jing Qin, and Pheng-Ann Heng, “Fine-grained recurrent neural networks for automatic prostate segmentation in ultrasound images,” arxiv (2016). http://arxiv.org/abs/1612.01655.

  16. Ke Yan, Changyang Li, Xiuying Wang, Ang Li, Yuchen Yuan, David Dagan Feng, Mohamed Khadra, and Jinman Kim, “Automatic prostate segmentation on MR images with deep network and graph model,” in 2016 38th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC) (2016), pp. 635–638.

  17. A. Krizhevsky, I. Sutskever, and G. E. Hinton, “ImageNet classification with deep convolutional neural networks,” Commun. ACM 60 (6), 84–90 (2017). https://doi.org/10.1145/3065386

    Article  Google Scholar 

  18. M. Kirschner, F. Jung, and S. Wesarg, “Automatic prostate segmentation in MR images with a probabilistic active shape model,” in International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI) (2012).

  19. A. Salimi, M. A. Pourmina, and M. S. Moin, “Fully automatic prostate segmentation in MR images using a new hybrid active contour-based approach,” Signal Image Video Process. 12 (8), 1629–1637 (2018). https://doi.org/10.1007/s11760-018-1320-y

    Article  Google Scholar 

  20. G. Vincent, G. Guillard, and M. Bowes, “Fully automatic segmentation of the prostate using active appearance models,” in MICCAI 2012 Grand Challenge: Prostate MR Image Segmentation (2012), pp. 35–43. https://doi.org/10.1371/journal.pone.0076645

    Book  Google Scholar 

  21. N. B. Rizvandi, A. Pizurica, and W. Philips, “Active appearance model (AAM) from theory to implementation,” in VISAPP 2008: Proceedings of the Third International Conference on Computer Vision Theory and Applications (Funchal, Madeira, Portugal, 2008), Vol. 1, pp. 539–542.

  22. F. Malmberg, R. Strand, J. Kullberg, R. Nordenskjöld, and E. Bengtsson, “Smart paint: A new interactive segmentation method applied to MR prostate segmentation,” in International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI) (2012).

  23. A. Gubern-Merida and R. Marti, “Atlas based segmentation of the prostate in mr images,” in MICCAI: Segmentation Challenge Workshop (2009).

    Google Scholar 

  24. J. Dowling, J. Fripp, P. Greer, S. Ourselin, and O. Salvado, “Automatic atlas-based segmentation of the prostate,” in A MICCAI 2009 Prostate Segmentation Challenge Entry. Workshop in Med. Image Comput. Comput. Assist. Interv. (2009), pp. 17–24.

    Google Scholar 

  25. S. Ghose, O. Arnau, R. Marti, X. Lladó, J. Freixenet, J. C. Vilanova, and F. Meriaudeau, “Prostate segmentation with texture enhanced active appearance model,” in 2010 Sixth International Conference on Signal-Image Technology and Internet Based Systems (IEEE, 2010), pp. 18–22.

  26. Fisher Yu and Vladlen Koltun, “Multi-scale context aggregation by dilated convolutions,” arXiv (2015). arXiv:1511.07122 [cs.CV]

  27. G. Litjens, R. Toth, W. van de Ven, C. Hoeks, S. Kerkstra, B. van Ginneken, G. Vincent, et al., “Evaluation of prostate segmentation algorithms for MRI: The PROMISE12 Challenge,” Med. Image Anal. 18 (2), 359–373 (2014).

    Article  Google Scholar 

  28. S. Ioffe and C. Szegedy, “Batch normalization: Accelerating deep network training by reducing internal covariate shift,” arXiv (2015). arXiv:1502.03167 [cs.LG]

  29. C. Szegedy, V. Vanhoucke, S. Ioffe, J. Shlens, and Z. Wojna, “Rethinking the inception architecture for computer vision,” in 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) (2016), pp. 2818–2826.

  30. J. T. Springenberg, A. Dosovitskiy, T. Brox, and M. A. Riedmiller, “Striving for simplicity: The all convolutional net,” arXiv (2014). arXiv:1412.6806 [cs.LG]

  31. M. Drozdzal, E. Vorontsov, G. Chartrand, S. Kadoury, and C. J. Pal, “The importance of skip connections in biomedical image segmentation,” in DLMIA 2016, LABELS 2016: Deep Learning and Data Labeling for Medical Applications (2016), pp. 179–187.

  32. 2009 Prostate Segmentation Challenge MICCAI (2009). http://wiki.na-mic.org/Wiki/index.php.

Download references

Author information

Authors and Affiliations

Authors

Corresponding authors

Correspondence to Aman Agarwal, Aditya Mishra, Madhushree Basavarajaiah, Priyanka Sharma or Sudeep Tanwar.

Ethics declarations

The authors declare that they have no conflicts of interest. This article does not contain any studies involving animals or human participants performed by any of the authors.

Additional information

Aman Agarwal was born in 1997 and graduated with a degree in Computer Engineering from the Institute of Technology, Nirma University, Ahmedabad, Gujarat, India in 2019. During the undergraduate period, his research interests included the application of machine learning and deep learning techniques in medical imaging, satellite imaging, signal processing, parallel computing, and natural language processing. He has several years of programming experience. Currently, he is working as a software engineer in a software company in India.

Aditya Mishra earned his Bachelor’s degree in Computer Engineering in the year 2019 from the Institute of Technology, Nirma University, Ahmedabad, Gujarat, India. His research interests include Computer Vision, Audio Processing, Natural Language Processing, High-Performance Computing and their applications in the field of Medical Imaging, Nuclear Physics, Satellite Imaging, Agricultural Science, Speech Synthesis and Media and Entertainment. Currently, he is working as a Product Engineer in a startup based out of Bengaluru, India.

Madhushree Basavarajaiah is a Ph.D. Research Scholar working under the supervision of Dr. Priyanka Sharma at the Institute of Technology, Nirma University, Ahmedabad. Her research areas include Video processing, Machine Learning, and Deep Learning. She received a Bachelor’s degree in Information Science and Engineering in 2011 and a Master’s degree in Web Technology in 2013 from University Visvesvaraya College of Engineering, Bengaluru. She has teaching experience of three years at undergraduate and postgraduate levels. She is an active member of ACM and IEEE.

Dr. Priyanka Sharma has received her BE in Computer Engineering from L D Engineering College (1999), MTech in Computer Science and Engineering and PhD from Nirma University (2012), Gujarat, India. She is NVIDIA Certified Deep Learning University Ambassador and has over 20 years of teaching and industrial experience. Her international exposure includes visits to Silicon Valley, USA and Halifax and Toronto in Canada for research-based collaborations/presentations. She has completed three research projects funded by Shastri Indo-Canadian Research Grant, IPR and GUJCOST and is currently working on 2 projects funded by BRNS-DAE. She has also worked as the Principal Investigator of NVIDIA Research Center (2014). She has received various national level awards for her work. She is also involved as an Adviser/Member at various academic and research bodies at Nirma University and is also a member of various professional bodies like IEEE, ACM, IEEE Sensors Council, IEEE Biometrics Council, IEEE Signal Processing Society and Life Member of ISTE and CSI.

Dr. Sudeep Tanwar is an Associate Professor in the Computer Science and Engineering Department at the Institute of Technology, Nirma University, Ahmedabad, Gujarat, India. He is visiting Professor at Jan Wyzykowski University in Polkowice, Poland and the University of Pitesti in Pitesti, Romania. He received B.Tech in 2002 from Kurukshetra University, India, M.Tech (Honor’s) in 2009 from Guru Gobind Singh Indraprastha University, Delhi, India, and Ph.D. in 2016 with specialization in Wireless Sensor Network. He has authored or coauthored more than 130 technical research papers published in leading journals and conferences from the IEEE, Elsevier, Springer, Wiley, etc. He has also published six edited/authored books with International/National Publishers like IET, Springer. He has guided many students leading to M.E./M.Tech and guiding students leading to Ph.D. He is Associate Editor of IJCS, Wiley and Security and Privacy Journal, Wiley. His current interest includes Wireless Sensor Networks, Fog Computing, Smart Grid, IoT, and Blockchain Technology. He was invited as Guest Editors/Editorial Board Members of many International Journals, invited for keynote Speaker in many International Conferences held in Asia and invited as Program Chair, Publications Chair, Publicity Chair, and Session Chair in many International Conferences held in North America, Europe, Asia, and Africa. He has been awarded the best research paper awards from IEEE GLOBECOM 2018, IEEE ICC 2019, and Springer ICRIC-2019.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Aman Agarwal, Mishra, A., Basavarajaiah, M. et al. Dilated Volumetric Network: an Enhanced Fully Convolutional Network for Volumetric Prostate Segmentation from Magnetic Resonance Imaging. Pattern Recognit. Image Anal. 31, 228–239 (2021). https://doi.org/10.1134/S1054661821020024

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1134/S1054661821020024

Keywords:

Navigation