Optimized Machine Learning for the Early Detection of Polycystic Ovary Syndrome in Women.

Journal: Sensors (Basel, Switzerland)
PMID:

Abstract

Polycystic ovary syndrome (PCOS) is a medical condition that impacts millions of women worldwide; however, due to a lack of public awareness, as well as the expensive testing involved in the identification of PCOS, 70% of cases go undiagnosed. Therefore, the primary objective of this study is to design an expert machine learning (ML) model for the early diagnosis of PCOS based on initial symptoms and health indicators; two datasets were amalgamated and preprocessed to accomplish this goal, resulting in a new symptomatic dataset with 12 attributes. An ensemble learning (EL) model, with seven base classifiers, and a deep learning (DL) model, as the meta-level classifier, are proposed. The hyperparameters of the EL model were optimized through the nature-inspired walrus optimization (WaO), cuckoo search optimization (CSO), and random search optimization (RSO) algorithms, leading to the WaOEL, CSOEL, and RSOEL models, respectively. The results obtained prove the supremacy of the designed WaOEL model over the other models, with a PCOS prediction accuracy of 92.8% and an area under the receiver operating characteristic curve (AUC) of 0.93; moreover, feature importance analysis, presented with random forest (RF) and Shapley additive values (SHAP) for positive PCOS predictions, highlights crucial clinical insights and the need for early intervention. Our findings suggest that patients with features related to obesity and high cholesterol are more likely to be diagnosed as PCOS positive. Most importantly, it is inferred from this study that early PCOS identification without expensive tests is possible with the proposed WaOEL, which helps clinicians and patients make better informed decisions, identify comorbidities, and reduce the harmful long-term effects of PCOS.

Authors

  • Bharti Panjwani
    Department of Computer Science & Engineering, Shri Madhwa Vadiraja Institute of Technology and Management, Bantakal 574115, Karnataka, India.
  • Jyoti Yadav
    Instrumentation and Control Engineering Division, NSIT, Sec-3, Dwarka, New Delhi, India. bmjyoti@gmail.com.
  • Vijay Mohan
    Department of Mechatronics, Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal 576104, Karnataka, India.
  • Neha Agarwal
    School of Chemical Engineering, Yeungnam University, Gyeongsan 38541, Republic of Korea.
  • Saurabh Agarwal
    Department of Diagnostic Imaging, Rhode Island Hospital, 593 Eddy St, Main, Floor 3, Providence, RI, 02903, USA.