Automated wave labelling of the auditory brainstem response using machine learning

Abstract

Objective To compare the performance of a selection of machine learning algorithms, trained to label peaks I, III, and V of the auditory brainstem response (ABR) waveform. An additional algorithm was trained to provide a confidence measure related to the ABR wave latency estimates. Design Secondary data analysis of a previously published ABR dataset. Five types of machine learning algorithm were compared within a nested k-fold cross-validation procedure. Study sample A set of 482 suprathreshold ABR waveforms were used. These were recorded from 81 participants with audiometric thresholds within normal limits. Results A convolutional recurrent neural network (CRNN) outperformed the other algorithms evaluated. The algorithm labelled 95.9% of ABR waves within ±0.1 ms of the target. The mean absolute error was 0.025 ms, averaged across the outer validation folds of the nested cross-validation procedure. High confidence levels were generally associated with greater wave-labelling accuracy. Conclusions Machine learning algorithms have the potential to assist clinicians with ABR interpretation. The present work identifies a promising machine learning approach, but any algorithm to be used in clinical practice would need to be trained on a large, accurately labelled, heterogeneous dataset and evaluated in clinical settings in follow-on work.