论文标题
基于变压器的规范建模,用于早期精神分裂症的异常检测
Transformer-based normative modelling for anomaly detection of early schizophrenia
论文作者
论文摘要
尽管精神疾病对临床健康有影响,但早期诊断仍然是一个挑战。机器学习研究表明,在诊断预测任务中,分类器往往过于狭窄。条件之间的重叠导致参与者之间的异质性很高,而分类模型无法充分捕获。为了解决这个问题,规范方法已成为另一种方法。通过使用生成模型来学习健康的大脑数据模式的分布,我们可以将病理的存在视为偏离模型分布的偏差或离群值。特别是,深层生成模型作为鉴定大脑中神经病变的规范模型显示出了很好的结果。但是,与大多数神经病变不同,精神疾病在几个大脑区域都广泛地展现出细微的变化,从而使这些改变具有挑战性。在这项工作中,我们评估了基于变压器的规范模型的性能,以检测青少年和年轻人表达的微妙大脑变化。我们对神经型个体的3D MRI扫描进行了训练(n = 1,765)。然后,我们从人类Connectome项目的独立数据集(n = 93)中获得了神经型控制和精神病患者的可能性。将扫描的可能性作为规范得分的代理,当评估对照组与患有早期精神分裂症的个体之间的差异时,我们获得了0.82的AUROC。我们的方法超过了基于脑年龄和高斯过程的最新规范方法,显示了有希望使用深层生成模型来帮助个性化分析的方法。
Despite the impact of psychiatric disorders on clinical health, early-stage diagnosis remains a challenge. Machine learning studies have shown that classifiers tend to be overly narrow in the diagnosis prediction task. The overlap between conditions leads to high heterogeneity among participants that is not adequately captured by classification models. To address this issue, normative approaches have surged as an alternative method. By using a generative model to learn the distribution of healthy brain data patterns, we can identify the presence of pathologies as deviations or outliers from the distribution learned by the model. In particular, deep generative models showed great results as normative models to identify neurological lesions in the brain. However, unlike most neurological lesions, psychiatric disorders present subtle changes widespread in several brain regions, making these alterations challenging to identify. In this work, we evaluate the performance of transformer-based normative models to detect subtle brain changes expressed in adolescents and young adults. We trained our model on 3D MRI scans of neurotypical individuals (N=1,765). Then, we obtained the likelihood of neurotypical controls and psychiatric patients with early-stage schizophrenia from an independent dataset (N=93) from the Human Connectome Project. Using the predicted likelihood of the scans as a proxy for a normative score, we obtained an AUROC of 0.82 when assessing the difference between controls and individuals with early-stage schizophrenia. Our approach surpassed recent normative methods based on brain age and Gaussian Process, showing the promising use of deep generative models to help in individualised analyses.