Comparison between logistic regression and machine learning algorithms on prediction of noise-induced hearing loss and investigation of SNP loci.

Journal: Scientific reports
PMID:

Abstract

To compare the comprehensive performance of conventional logistic regression (LR) and seven machine learning (ML) algorithms in Noise-Induced Hearing Loss (NIHL) prediction, and to investigate the single nucleotide polymorphism (SNP) loci significantly associated with the occurrence and progression of NIHL. A total of 1,338 noise-exposed workers from 52 enterprises in Jiangsu Province were included in this study. 88 SNP loci involving multiple genes related to noise exposure and hearing loss were detected. LR and multiple ML algorithms were employed to establish the NIHL prediction model with accuracy, recall, precision, F-score, R and AUC as performance indicators. Compared to conventional LR, the evaluated ML models Generalized Regression Neural Network (GRNN), Probabilistic Neural Network (PNN), Genetic Algorithm-Random Forests (GA-RF) demonstrate superior performance and were considered to be the optimal models for processing large-scale SNP loci dataset. The SNP loci screened by these models are pivotal in the process of NIHL prediction, which further improves the prediction accuracy of the model. These findings open new possibilities for accurate prediction of NIHL based on SNP locus screening in the future, and provide a more scientific basis for decision-making in occupational health management.

Authors

  • Jie Lu
    Department of Endocrinology and Metabolism, the Affiliated Hospital of Qingdao University, Qingdao, People's Republic of China.
  • Xinhao Lu
    School of Cyber Science and Engineering, Southeast University, Nanjing, 211189, China.
  • Yixiao Wang
    Department of Gastroenterology, The First Affiliated Hospital of Wenzhou Medical University, Nanbaixiang, Ouhai District, Wenzhou, Zhejiang, P.R. China.
  • Hengdong Zhang
    Department of Occupational Disease Prevention, Jiangsu Provincial Center for Disease Control and Prevention, Nanjing, China.
  • Lei Han
    Jiangsu Province Center for Disease Prevention and Control, Institute of Occupational Disease Prevention, Nanjing, Jiangsu, China.
  • Baoli Zhu
    School of Public Health, Nanjing Medical University, Nanjing, Jiangsu, China.
  • Boshen Wang
    Jiangsu Province Center for Disease Prevention and Control, Institute of Occupational Disease Prevention, Nanjing, Jiangsu, China.