Optimised Feature Selection and Cervical Cancer Prediction Using Machine Learning Classification
Cervical Cancer Prediction
Abstract
Background/Aim: Screening and early detection play a key role in cervical cancer prevention. The present study predicts the outcome of various diagnostic tests used to diagnose cervical cancer using machine learning algorithms.
Methods: The present study ran various cervical cancer risk factors on a machine learning (ML) classifier to predict outcomes of Hinselmann, Schiller, cytology and biopsy. The dataset is publicly available on the Machine Learning Repository website of the University of California Irvine. The imbalanced dataset was pre-processed using oversampling methods. The significantly varied features between the two levels of a response variable were used to train the machine learning classifiers on MATLAB. The classifiers used were Decision Trees, Support Vector Machine, K-Nearest Neighbours and Ensemble learning classifiers. The performance metrics of the classifiers were expressed as accuracy, the area under the receiver operator characteristic (AU-ROC) curve, sensitivity and specificity.
Results: The Fine Gaussian SVM classifier was the best to classify Hinselmann, cytology and biopsy with the accuracy of 97.5 %, 62.5 % and 98 %, respectively. However, Boosted trees performed best in the classification of Schiller with 81.3 % accuracy.
Conclusion: The present study selected optimised features among multiple risk factors to train various ML classifiers to predict cervical cancer.
References
Bray F, Ferlay J, Soerjomataram I, Siegel RL, Torre LA, Jemal A. Global cancer statistics 2018: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries. CA Cancer J Clin 2018 Nov;68(6):394–424.
Zhang S, Xu H, Zhang L, Qiao Y. Cervical cancer: Epidemiology, risk factors and screening. Chinese J Cancer Res 2020 Dec 31;32(6):720-8.
Kisling LA, M Das J. Prevention strategies. [Updated 2022 May 8]. In: StatPearls [Internet]. Treasure Island (FL): StatPearls Publishing; 2022 Jan-. [cited 2022 Jun 6]. Available from: https://www.ncbi.nlm.nih.gov/books/NBK537222/
Momenimovahed Z, Momenimovahed Z, Salehiniya H. Incidence, mortality and risk factors of cervical cancer in the world. Biomed Res Ther 2017 Dec;4(12):1795–811.
Sagala NTM. A Comparative study of data mining methods to diagnose cervical cancer. J Phys Conf Ser 2019 Aug;1255(1):012022. doi:10.1088/1742-6596/1255/1/012022.
Parikh RB, Manz C, Chivers C, Regli SH, Braun J, Draugelis ME, et al. Machine learning approaches to predict 6-month mortality among patients with cancer. JAMA Netw Open 2019 Oct 2;2(10):e1915997. doi: 10.1001/jamanetworkopen.2019.15997.
Arezzo F, La Forgia D, Venerito V, Moschetta M, Tagliafico AS, Lombardi C, et al. A machine learning tool to predict the response to neoadjuvant chemotherapy in patients with locally advanced cervical cancer. Appl Sci 2021;11(2):823. doi:10.3390/app11020823
Kuruvilla A, Jayanthi B. Analysis and review on feature selection and classification methods on cervical cancer. Ictact J Soft Comput 2022;12(2):2551-8.
Jahan S, Islam MDS, Islam L, Rashme TY, Prova AA, Paul BK, et al. Automated invasive cervical cancer disease detection at early stage through suitable machine learning model. SN Appl Sci 2021 Oct;3(10):1–17.
R KK, Kanimozhi N. Prediction of cervical cancer using machine learning and deep learning algorithms the Creative Commons Attribution License (CC BY 4.0). Int J Trend Sci Res Dev 2020 Oct;4(6):426–30.
Mehmood M, Rizwan M, Gregus M, Abbas S. Machine learning assisted cervical cancer detection. Front Public Heal 2021 Dec 23;9:788376. doi: 10.3389/fpubh.2021.788376.
Alsmariy R, Healy G, Abdelhafez H. Predicting cervical cancer using machine learning methods. Int J Adv Comput Sci Appl 2020;11(7):173–84.
Asadi F, Salehnasab C AL. Supervised algorithms of machine learning for the prediction of cervical cancer. J Biomed Phys Eng 2020;10(4):513–22.
Moldovan D. Cervical cancer diagnosis using a chicken swarm optimization based machine learning method. Int Conf EHB 2020 Dec;10;1–4.
Fernandes K, Chicco D, Cardoso JS, Fernandes J. Supervised deep learning embeddings for the prediction of cervical cancer diagnosis. PeerJ Comput Sci. 2018 May 14;4:e154. doi: 10.7717/peerj-cs.154.
Wu W, Zhou H. Data-driven diagnosis of cervical cancer with support vector machine-based approaches. IEEE Access 2017;5:25189-95.
Team MATLAB. Statistics and Machine Learning Toolbox. Natick, Massachusetts: The Mathworks Inc; 2019.
JASP Team. JASP (Version 0.16.2.0)[Computer software] [Internet]. 2022. [cited 2022 Jun 6]. Available from: https://jasp-stats.org/.
Razali N, Mostafa SA, Mustapha A, Wahab MHA, Ibrahim NA. Risk factors of cervical cancer using classification in data mining. J Phys Conf Ser [Internet]. 2020 Apr 1 [cited 2022 Jun 6];1529(2):022102. Available from: https://iopscience.iop.org/article/10.1088/1742-6596/1529/2/022102.
Tanimu JJ, Hamada M, Hassan M, Kakudi HA, Abiodun JO. A machine learning method for classification of cervical cancer. Electronics 2022;11(3):463. doi:10.3390/electronics11030463.
CH N, Sai PP, Madhuri G, Reddy KS, Simha B, Reddy DV. Artificial Intelligence based Cervical Cancer Risk Prediction Using M1 Algorithms. 2022 Int Conf Emerg Smart Comput Informatics 2022 Mar;1–6. doi: 10.1109/ESCI53509.2022.9758241.
Ali MM, Ahmed K, Bui FM, Paul BK, Ibrahim SM, Quinn JMW, et al. Machine learning-based statistical analysis for early stage detection of cervical cancer. Comput Biol Med 2021 Dec;139:104985. doi: 10.1016/j.compbiomed.2021.104985.
Nithya B, Ilango V. Evaluation of machine learning based optimized feature selection approaches and classification methods for cervical cancer prediction. SN Appl Sci 2019 Jun;1(6):1–16.
Zahras D, Rustam Z. Cervical cancer risk classification based on deep convolutional neural network. Proc ICAITI 2018 - 1st Int Conf Appl Inf Technol Innov Towar A New Paradig Des Assist Technol Smart Home Care 2018 Jul 2;149–53.
Asaduzzaman S, Ahmed MR, Rehana H, Chakraborty S, Islam MS, Bhuiyan T. Machine learning to reveal an astute risk predictive framework for Gynecologic Cancer and its impact on women psychology: Bangladeshi perspective. BMC Bioinformatics 2021 Apr 24;22(1):213. doi: 10.1186/s12859-021-04131-6.
Chaudhuri AK, Ray A, Banerjee DK, Das A. A multi-stage approach combining feature selection with machine learning techniques for higher prediction reliability and accuracy in cervical cancer diagnosis. Int J Intell Syst Appl 2021 Oct 8;13(5):46–63.
Sobar, Machmud R, Wijaya A. Behavior determinant based cervical cancer early detection with machine learning algorithm. Adv Sci Lett 2016 Oct 1;22(10):3120–3.
Ceylan Z, Pekel E. Comparison of multi-label classification methods for prediagnosis of cervical cancer. Int J Intell Syst Appl Eng 2017 Dec;5(4):232–6.
Gupta A, Anand A, Hasija Y. Recall-based Machine Learning approach for early detection of Cervical Cancer. 2021 6th International Conference for Convergence in Technology (I2CT) 2021;1-5. doi:10.1109/I2CT51068.2021.9418099.
Ahishakiye E, Wario R, Mwangi W TD. Prediction of cervical cancer basing on risk factors using ensemble learning. 2020 IST-Africa Conference (IST-Africa) 2020;1-12.
Ijaz MF, Attique M, Son Y. Data-driven cervical cancer prediction model with outlier detection and over-sampling methods. Sensors (Basel) 2020 May 15;20(10):2809. doi: 10.3390/s20102809.
Chauhan NK, Singh K. Performance assessment of machine learning classifiers using selective feature approaches for cervical cancer detection. Wirel Pers Commun 2022 Jan;124(3):2335–66.
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).