用频域特征融合提升脑功能连接分析,助力抑郁症精准诊断
Frequency Feature Fusion Graph Network For Depression Diagnosis Via fNIRS
- 引入离散傅里叶变换提取频域特征,构建新时序生物标志物
- 在1086人数据集上实现更高F1分数,PSM子集验证效果稳定
- 模型可解释性强,适合临床场景落地应用
基于数据驱动的抑郁症诊断方法在神经医学中日益重要,得益于相关数据集的发展。近年来,图神经网络(GNN)因其能从时空双重角度捕捉脑通道功能连接而被广泛应用,但其性能受限于缺乏强有力的时序生物标志物。本文通过离散傅里叶变换(DFT)提出一种新型有效生物标志物,并构建基于时序图卷积网络(TGCN)的定制化图网络架构。模型在包含1,086名受试者的数据集上训练,该数据规模超过以往抑郁症诊断研究的十倍以上。为符合医疗需求,我们采用倾向性评分匹配(PSM)生成精炼子集(即PSM数据集)。实验表明,引入新设计的生物标志物显著提升了脑通道时序表征能力,在真实世界数据集和PSM数据集上均获得更优的F1分数。此外,利用SHapley Additive exPlaination(SHAP)验证了模型的可解释性,确保其在医疗环境中的实际应用潜力。
原文摘要 · Abstract (English)
Data-driven approaches for depression diagnosis have emerged as a significant research focus in neuromedicine, driven by the development of relevant datasets. Recently, graph neural network (GNN)-based models have gained widespread adoption due to their ability to capture brain channel functional connectivity from both spatial and temporal perspectives. However, their effectiveness is hindered by the absence of a robust temporal biomarker. In this paper, we introduce a novel and effective biomarker for depression diagnosis by leveraging the discrete Fourier transform (DFT) and propose a customized graph network architecture based on Temporal Graph Convolutional Network (TGCN). Our model was trained on a dataset comprising 1,086 subjects, which is over 10 times larger than previous datasets in the field of depression diagnosis. Furthermore, to align with medical requirements, we performed propensity score matching (PSM) to create a refined subset, referred to as the PSM dataset. Experimental results demonstrate that incorporating our newly designed biomarker enhances the representation of temporal characteristics in brain channels, leading to improved F1 scores in both the real-world dataset and the PSM dataset. This advancement has the potential to contribute to the development of more effective depression diagnostic tools. In addition, we used SHapley Additive exPlaination (SHAP) to validate the interpretability of our model, ensuring its practical applicability in medical settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。