arXiv:2504.08584cs.CVcs.AI2025-04被引 14

用通用自监督表征提升多人群胸片分析的联邦学习效果

Boosting multi-demographic federated learning for chest radiograph analysis using general-purpose self-supervised representations

  • 用通用自监督模型做迁移,缓解医疗数据非独立同分布问题
  • 在儿童和大样本成人数据上,传统联邦学习表现下降,自监督后显著改善
  • 特别适合数据少、差异大的儿科医疗场景,易部署且有效

可靠的医学图像分析人工智能模型通常依赖大规模多样化的标注数据。联邦学习(FL)提供了一种去中心化且保护隐私的训练方式,但在高度非独立同分布(non-IID)环境下表现不佳,尤其当机构数据更具代表性时性能反而下降。现有大规模联邦学习研究主要集中在成人数据,忽视了儿科数据带来的额外非独立同分布挑战。我们分析了来自多国398,523张成人胸片和9,125张儿童胸片,利用通用自监督图像表示进行迁移学习,以分类肺炎与正常病例。采用先进视觉变压器模型发现,传统联邦学习仅在小规模成人数据集上提升性能(P<0.001),而在大规模成人数据(P=0.064)和儿童数据(P=0.242)上表现更差。但引入自监督权重后,显著改善了儿童数据表现(P=0.031)及大多数成人数据集(P<0.008),仅最大数据集例外(P=0.052)。结果表明,可轻松部署的通用自监督表征能有效应对临床联邦学习中的非独立同分布挑战,有望提升患者预后,推动儿科医疗发展。

原文摘要 · Abstract (English)

Reliable artificial intelligence (AI) models for medical image analysis often depend on large and diverse labeled datasets. Federated learning (FL) offers a decentralized and privacy-preserving approach to training but struggles in highly non-independent and identically distributed (non-IID) settings, where institutions with more representative data may experience degraded performance. Moreover, existing large-scale FL studies have been limited to adult datasets, neglecting the unique challenges posed by pediatric data, which introduces additional non-IID variability. To address these limitations, we analyzed n=398,523 adult chest radiographs from diverse institutions across multiple countries and n=9,125 pediatric images, leveraging transfer learning from general-purpose self-supervised image representations to classify pneumonia and cases with no abnormality. Using state-of-the-art vision transformers, we found that FL improved performance only for smaller adult datasets (P<0.001) but degraded performance for larger datasets (P<0.064) and pediatric cases (P=0.242). However, equipping FL with self-supervised weights significantly enhanced outcomes across pediatric cases (P=0.031) and most adult datasets (P<0.008), except the largest dataset (P=0.052). These findings underscore the potential of easily deployable general-purpose self-supervised image representations to address non-IID challenges in clinical FL applications and highlight their promise for enhancing patient outcomes and advancing pediatric healthcare, where data scarcity and variability remain persistent obstacles.

联邦学习医学影像自监督儿科医疗

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