FedSAF提升胃癌检测模型性能,保护患者隐私
FedSAF: A Federated Learning Framework for Enhanced Gastric Cancer Detection and Privacy Preservation
- 用注意力消息传递和费雪信息矩阵优化联邦学习
- 在非独立同分布数据下准确率优于现有方法
- 适合医疗数据隐私敏感场景的模型开发
胃癌是常见高死亡率癌症,受限于医疗资源,构建机器学习模型可高效辅助诊断。但模型训练需大量样本,易泄露患者隐私。联邦学习可在不共享数据的前提下跨机构协作训练。本文针对公开胃癌数据集样本量有限的问题,提出改进数据处理方法,并设计新算法FedSAF,通过注意力机制与费雪信息矩阵提升模型精度,结合模型拆分降低计算与通信开销。超参数调优与消融实验表明,该方法在多个胃癌数据集上测试准确率优于FedAMP、FedAvg和FedProx等现有联邦学习方法。其鲁棒性与泛化能力在SEED、BOT、FashionMNIST和CIFAR-10等额外数据集上也得到验证,展现了跨场景高性能表现。
原文摘要 · Abstract (English)
Gastric cancer is one of the most commonly diagnosed cancers and has a high mortality rate. Due to limited medical resources, developing machine learning models for gastric cancer recognition provides an efficient solution for medical institutions. However, such models typically require large sample sizes for training and testing, which can challenge patient privacy. Federated learning offers an effective alternative by enabling model training across multiple institutions without sharing sensitive patient data. This paper addresses the limited sample size of publicly available gastric cancer data with a modified data processing method. This paper introduces FedSAF, a novel federated learning algorithm designed to improve the performance of existing methods, particularly in non-independent and identically distributed (non-IID) data scenarios. FedSAF incorporates attention-based message passing and the Fisher Information Matrix to enhance model accuracy, while a model splitting function reduces computation and transmission costs. Hyperparameter tuning and ablation studies demonstrate the effectiveness of this new algorithm, showing improvements in test accuracy on gastric cancer datasets, with FedSAF outperforming existing federated learning methods like FedAMP, FedAvg, and FedProx. The framework's robustness and generalization ability were further validated across additional datasets (SEED, BOT, FashionMNIST, and CIFAR-10), achieving high performance in diverse environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。