arXiv:2411.02286cs.LGcs.AI2024-11被引 4

用联邦图神经网络分析脑电数据,无须共享病历就能预测中风严重程度。

Federated GNNs for EEG-Based Stroke Assessment

  • 多机构协作训练图神经网络,本地保留数据不外传。
  • 预测中风量表得分误差仅3.23,接近专家水平。
  • 引入注意力机制和可解释性工具,看清大脑连接变化。

机器学习有望成为临床决策的重要工具,提升诊断能力和个性化治疗。但将医疗数据外包训练模型会引发法律、隐私与安全问题。联邦学习为多方协作建模提供了新范式,在不共享敏感数据的前提下保护患者隐私。本文提出一种结合联邦学习与图神经网络的新方法,用于跨多家医疗机构的脑电图(EEG)信号分析,以预测中风严重程度。研究将任务建模为回归问题,目标是预测国家卫生研究院中风量表(NIHSS)得分。所提模型采用掩码自注意力机制捕捉关键脑区连接模式,并使用EdgeSHAP提供中风后神经状态的后验解释。在四个医疗机构的EEG数据上评估,预测平均绝对误差(MAE)为3.23,接近人类专家的平均误差(约3.0),验证了该方法在保证数据隐私的同时,实现高精度且可解释的预测。

原文摘要 · Abstract (English)

Machine learning (ML) has the potential to become an essential tool in supporting clinical decision-making processes, offering enhanced diagnostic capabilities and personalized treatment plans. However, outsourcing medical records to train ML models using patient data raises legal, privacy, and security concerns. Federated learning has emerged as a promising paradigm for collaborative ML, meeting healthcare institutions' requirements for robust models without sharing sensitive data and compromising patient privacy. This study proposes a novel method that combines federated learning (FL) and Graph Neural Networks (GNNs) to predict stroke severity using electroencephalography (EEG) signals across multiple medical institutions. Our approach enables multiple hospitals to jointly train a shared GNN model on their local EEG data without exchanging patient information. Specifically, we address a regression problem by predicting the National Institutes of Health Stroke Scale (NIHSS), a key indicator of stroke severity. The proposed model leverages a masked self-attention mechanism to capture salient brain connectivity patterns and employs EdgeSHAP to provide post-hoc explanations of the neurological states after a stroke. We evaluated our method on EEG recordings from four institutions, achieving a mean absolute error (MAE) of 3.23 in predicting NIHSS, close to the average error made by human experts (MAE $\approx$ 3.0). This demonstrates the method's effectiveness in providing accurate and explainable predictions while maintaining data privacy.

联邦学习脑电分析图神经网络可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。