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.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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)
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
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
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Switzerland AG
About this paper
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)