论文标题
表示临床因素的可解释性和分类准确性提高了脑电图
Representation learning for improved interpretability and classification accuracy of clinical factors from EEG
论文作者
论文摘要
尽管进行了广泛的标准化,但针对心理健康障碍的诊断访谈涵盖了实质性的主观判断。先前的研究表明,基于脑电图的神经测量可以作为抑郁症的可靠客观相关性,甚至可以作为抑郁症及其过程的预测指标。但是,由于1)缺乏应对与脑电图数据相关的固有噪声的方法,而2)缺乏对脑电信号的哪些方面可能是临床疾病的标志的知识。在这里,我们从最近的深层表示学习文献中调整了无监督的管道,以通过1)使用$β$ -VAE来学习信号来解决这些问题。我们证明,我们的方法能够在许多因素(包括参与者的年龄和抑郁诊断)上胜过规范手工设计的基线分类方法。此外,我们的方法还恢复了一种可用于自动从新颖,单个EEG轨迹中自动提取deNOCE的事件相关电位(ERP)的表示形式,并支持快速监督的重新映射到各种临床标签,从而使临床医生能够重新使用单个EEG表示,无论对标准化诊断系统的更新。最后,学到的分解表示形式的单个因素通常对应于有意义的临床因素标记,如扫描自动检测,从而可以对模型提出的建议进行人类的可解释性和事后专家分析。
Despite extensive standardization, diagnostic interviews for mental health disorders encompass substantial subjective judgment. Previous studies have demonstrated that EEG-based neural measures can function as reliable objective correlates of depression, or even predictors of depression and its course. However, their clinical utility has not been fully realized because of 1) the lack of automated ways to deal with the inherent noise associated with EEG data at scale, and 2) the lack of knowledge of which aspects of the EEG signal may be markers of a clinical disorder. Here we adapt an unsupervised pipeline from the recent deep representation learning literature to address these problems by 1) learning a disentangled representation using $β$-VAE to denoise the signal, and 2) extracting interpretable features associated with a sparse set of clinical labels using a Symbol-Concept Association Network (SCAN). We demonstrate that our method is able to outperform the canonical hand-engineered baseline classification method on a number of factors, including participant age and depression diagnosis. Furthermore, our method recovers a representation that can be used to automatically extract denoised Event Related Potentials (ERPs) from novel, single EEG trajectories, and supports fast supervised re-mapping to various clinical labels, allowing clinicians to re-use a single EEG representation regardless of updates to the standardized diagnostic system. Finally, single factors of the learned disentangled representations often correspond to meaningful markers of clinical factors, as automatically detected by SCAN, allowing for human interpretability and post-hoc expert analysis of the recommendations made by the model.