Reliability-quantified multi-task learning: Dual-head architectures of risk-awareness for EEG devices
摘要
Neurological disorders globally affect millions of people, a major burden, both personally and financially for individuals, as well as for society. Although deep learning models provide accurate diagnoses in laboratory settings with curated data. The models do not perform when they are used on newer portable, user friendly, Portable and clinical EEG devices can produce varying amounts of channels of EEG recordings to produce a greater amount of real-world EEG recordings. In addition to there being fewer channels than with traditional EEG devices this reliability issue prevents the clinician from being able to use the output of the deep learning model to make an informed decision, because there is no way to provide the clinician with the uncertainty associated with the output using currently available standard modeling. Therefore, this work’s objectives are to present a framework called a reliability-based LSTM, a multi-task learning system of analysing the EEG; for use in a clinical setting. the analysis has been modified to using a dual-head LSTM with three qualitatively different options for estimating the outputs easy measure High Reliability Score (HRS), temperature scaling, and conformal prediction. This integration will allow a model to continue to provide a high degree of diagnostic confidence and produce appropriately calibrated confidence estimates for risk-based decision making. The proposed model achieved a seizure detection accuracy rate of 96.66%, while also providing appropriately calibrated results as evidenced by near perfect performance on reliability diagrams and the Risk-Coverage curve. The HRS allowed the authors to risk stratify using a mean HRS of 0.78 and a high confidence (