在隐私保护下用少量标注数据精准预测妊娠期糖尿病。
Federated Semi-Supervised Graph Neural Networks with Prototype-Guided Pseudo-Labeling for Privacy-Preserving Gestational Diabetes Mellitus Prediction

- 通过原型引导伪标签和动态图更新,提升无标签数据利用效率。
- 在80%数据缺失时仍保持高准确率(如早期阶段AUROC达0.9634)。
- 适合医疗数据跨机构协作,兼顾隐私与模型性能。
妊娠期糖尿病(GDM)是高发妊娠并发症,需早期精准风险分层以降低母婴风险。但真实临床部署受双重限制:一是标签稀缺,大量电子健康记录(EHR)缺乏确诊标签;二是数据隐私,医院间无法共享患者级数据。本文提出FedTGNN-SS,一种隐私保护的联邦半监督框架,用于临床表格型EHR分析。每家医院构建局部k近邻患者相似性图,并训练拓扑自适应图神经网络编码器。为有效利用无标签记录,该方法结合:(1)基于原型的伪标签与邻域一致性;(2)基于学习嵌入定期更新的自适应图重构;(3)仅对连续变量应用临床感知一致性增强;(4)仅交换类别级中心点的隐私安全原型共享。在三个糖尿病相关数据集(GDM: N=3,525;Pima: N=768;Early Stage: N=520)上,当各机构标签缺失率10%-80%时,FedTGNN-SS在11个联邦基线中取得56次显著胜出(p < 0.05),并在极端标签稀缺下表现优异(如Pima数据集80%缺失时AUROC达0.8037,Early Stage数据集80%缺失时达0.9634)。
原文摘要 · Abstract (English)
Gestational Diabetes Mellitus (GDM) is a high-prevalence pregnancy complication that requires accurate early risk stratification to reduce maternal and fetal morbidity. However, real-world clinical deployment of machine learning is hindered by two coupled constraints: (i) label scarcity, where a large fraction of electronic health records (EHR) lack confirmed diagnostic labels, and (ii) data privacy, which prevents sharing patient-level data across hospitals. This paper proposes FedTGNN-SS, a privacy-preserving federated semi-supervised framework for clinical tabular EHR. Each hospital builds a local k-nearest-neighbor patient similarity graph and trains a topology-adaptive GNN encoder. To robustly exploit unlabeled records, FedTGNN-SS combines (1) prototype-guided pseudo-labeling with neighborhood agreement, (2) adaptive graph refinement that periodically updates the k-NN graph using learned embeddings, (3) clinical-aware consistency augmentation applied only to continuous variables, and (4) privacy-safe prototype sharing that exchanges only class-level centroids. Across three diabetes-related datasets (GDM: N = 3,525; Pima: N = 768; Early Stage: N = 520) under 10\%-80\% missing labels per silo, FedTGNN-SS achieves 56 significant wins ($p < 0.05$) against 11 federated baselines and attains strong AUROC under extreme scarcity (Pima: 0.8037 at 80\% missing, Early Stage: 0.9634 at 80\% missing).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。