Development of Machine-Learning Models for Tinnitus-Related Distress Classification Using Wavelet-Transformed Auditory Evoked Potential Signals and Clinical Data

Abstract

Tinnitus is a highly prevalent condition, affecting more than 1 in 7 adults in the EU and causing negative effects on sufferers’ quality of life. In this study, we utilised data collected within the “UNITI” project, the largest EU tinnitus-related research programme. Initially, we extracted characteristics from both auditory brainstem response (ABR) and auditory middle latency response (AMLR) signals, which were derived from tinnitus patients. We then combined these features with the patients’ clinical data, and integrated them to build machine learning models for the classification of individuals and their ears according to their level of tinnitus-related distress. Several models were developed and tested on different datasets to determine the most relevant features and achieve high performances. Specifically, seven widely used classifiers were utilised on all generated datasets: random forest (RF), linear, radial, and polynomial support vector machines (SVM), naive bayes (NB), neural networks (NN), and linear discriminant analysis (LDA). Results showed that features extracted from the wavelet-scattering transformed AMLR signals were the most informative data. In combination with the 15 LASSO-selected clinical features, the SVM classifier achieved optimal performance with an AUC value, sensitivity, and specificity of 92.53%, 84.84%, and 83.04%, respectively, indicating high discrimination performance between the two groups.