Skip to main content

Modified MobileNet for Patient Survival Prediction

  • Conference paper
  • First Online:
Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries (BrainLes 2020)

Abstract

Glioblastoma is a type of malignant tumor that varies significantly in size, shape, and location. The study of this type of tumor, one of which is about predicting the patient’s survival ability, is beneficial for the treatment of patients. However, the supporting data for the survival prediction model are minimal, so the best methods are needed for handling it. In this study, we propose an architecture for predicting patient survival using MobileNet combined with a linear survival prediction model (SPM). Several variations of MobileNet are tested to obtain the best results. Variations tested include modification of MobileNet V1 with freeze or unfreeze layers, and modification of MobileNet V2 with freeze or unfreeze layers connected to SPM. The dataset used for the trial came from BraTS 2020. A modification based on the MobileNet V2 architecture with the freezing layer was selected from the test results. The results of testing this proposed architecture with 95 training data and 23 validation data resulted in an MSE Loss of 78374.17. The online test results with the validation dataset 29 resulted in an MSE loss value of 149764.866 with an accuracy of 0.345. Testing with the testing dataset resulted in increased accuracy of 0.402. These results are promising for better architectural development.

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

Access this chapter

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD 89.00
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD 119.99
Price excludes VAT (USA)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free shipping worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

References

  1. Agravat, R.R., Raval, M.S.: Prediction of overall survival of brain tumor patients. In: IEEE Region 10 Annual International Conference, Proceedings/TENCON, vol. 2019, pp. 31–35. Institute of Electrical and Electronics Engineers Inc., Oct 2019. https://doi.org/10.1109/TENCON.2019.8929497

  2. Bakas, S., Akbari, H., Sotiras, A., et al.: Segmentation labels for the pre-operative scans of the tcga-gbm collection (2017). https://doi.org/10.7937/K9/TCIA.2017.KLXWJJ1Q

  3. Bakas, S., Akbari, H., Sotiras, A., et al.: Segmentation labels for the pre-operative scans of the tcga-lgg collection (2017). https://doi.org/10.7937/K9/TCIA.2017.GJQ7R0EF

  4. Bakas, S., Akbari, A., Sotiras, H., et al.: Advancing the cancer genome atlas glioma MRI collections with expert segmentation labels and radiomic features. Sci. Data 4, p. 170117 (2017). https://doi.org/10.1038/sdata.2017.117, http://www.nature.com/articles/sdata2017117

  5. Bakas, S., Reyes, M., Jakab, A., et al.: Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challenge. Technical report, November 2018. https://doi.org/10.17863/CAM.38755

  6. Carver, E., et al.: Automatic brain tumor segmentation and overall survival prediction using machine learning algorithms. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 406–418. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_36

    Chapter  Google Scholar 

  7. Chato, L., Latifi, S.: Machine learning and deep learning techniques to predict overall survival of brain tumor patients using MRI images. In: Proceedings - 2017 IEEE 17th International Conference on Bioinformatics and Bioengineering, BIBE 2017, vol. 2018, no. 1, pp. 9–14. Institute of Electrical and Electronics Engineers Inc., July 2017. https://doi.org/10.1109/BIBE.2017.00-86

  8. Dai, L., Li, T., Shu, H., Zhong, L., Shen, H., Zhu, H.: Automatic brain tumor segmentation with domain adaptation. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 380–392. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_34

    Chapter  Google Scholar 

  9. Feng, X., Tustison, N., Meyer, C.: Brain tumor segmentation using an ensemble of 3D U-Nets and overall survival prediction using radiomic features. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 279–288. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_25

    Chapter  Google Scholar 

  10. Howard, A.G., Zhu, M., Chen, B., et al.: MobileNets: efficient convolutional neural networks for mobile vision applications, April 2017. http://arxiv.org/abs/1704.04861

  11. Lefkovits, S., SzilĂ¡gyi, L., Lefkovits, L.: Brain tumor segmentation and survival prediction using a cascade of random forests. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 334–345. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_30

    Chapter  Google Scholar 

  12. Louis, D.N., Perry, A., Reifenberger, G., et al.: The 2016 world health organization classification of tumors of the central nervous system: a summary. Acta Neuropathol. 131(6), 803–820 (2016). https://doi.org/10.1007/s00401-016-1545-1

  13. Menze, B.H., Jakab, A., Bauer, S., et al.: The multimodal brain tumor image segmentation benchmark (BRATS). IEEE Trans. Med. Imaging 34(10), 1993–2024 (2015). https://doi.org/10.1109/TMI.2014.2377694

  14. Sandler, M., Howard, A., Zhu, M., et al.: MobileNetV2: inverted residuals and linear bottlenecks. In: Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, pp. 4510–4520. IEEE Computer Society (2018). http://arxiv.org/abs/1801.04381

  15. Shboul, Z.A., Alam, M., Vidyaratne, L., Pei, L., Iftekharuddin, K.M.: Glioblastoma survival prediction. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 508–515. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_45

    Chapter  Google Scholar 

  16. Srivastava, N., Hinton, G., Krizhevsky, A., et al.: Dropout: a simple way to prevent neural networks from overfitting. J. Mach. Learn. Res. 15(1), 1929–1958 (2014)

    MathSciNet  MATH  Google Scholar 

  17. Yang, H.-Y., Yang, J.: Automatic brain tumor segmentation with contour aware residual network and adversarial training. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11384, pp. 267–278. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-11726-9_24

    Chapter  Google Scholar 

Download references

Acknowledgements

This research was supported by the Ministry of Education and Culture and the Ministry of Research and Technology/BRIN, Indonesia. We are deeply grateful for both the Beasiswa Pendidikan Pascasarjana Dalam Negeri(BPPDN) and Penelitian Disertasi Doktor(PDD) 2020–2021 grant, which enabled this research to be carried out.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Agus Subhan Akbar .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2021 Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Akbar, A.S., Fatichah, C., Suciati, N. (2021). Modified MobileNet for Patient Survival Prediction. In: Crimi, A., Bakas, S. (eds) Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries. BrainLes 2020. Lecture Notes in Computer Science(), vol 12659. Springer, Cham. https://doi.org/10.1007/978-3-030-72087-2_33

Download citation

  • DOI: https://doi.org/10.1007/978-3-030-72087-2_33

  • Published:

  • Publisher Name: Springer, Cham

  • Print ISBN: 978-3-030-72086-5

  • Online ISBN: 978-3-030-72087-2

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics