用注意力U-Net与联邦近端算法,提升超声乳腺癌分割精度并保护隐私
A Novel Approach to Breast Cancer Segmentation using U-Net Model with Attention Mechanisms and FedProx
- 结合注意力机制的U-Net模型增强肿瘤边界识别能力
- 在非独立同分布数据上实现96%分割准确率
- 适合医疗图像分割与隐私保护场景的研究者
乳腺癌是全球女性主要致死原因,早期检测与精准诊断至关重要。超声成像因其可靠且成本低被广泛使用,但医学数据敏感性使得构建准确且私密的人工智能模型面临挑战。联邦学习可在保护患者隐私的前提下进行分布式机器学习,但在非独立同分布(non-IID)本地数据上训练时,会影响模型精度与泛化能力,进而影响乳腺癌分割中肿瘤边界的精确划分。本研究通过引入联邦近端(FedProx)方法处理非IID超声乳腺癌影像数据,并采用改进的带注意力机制的U-Net模型提升分割性能。实验结果表明,该方法获得的全局模型达到96%准确率,验证了其在保障患者隐私的同时提升肿瘤分割精度的有效性。研究提示FedProx在非IID医疗数据上训练高精度模型方面具有潜力。
原文摘要 · Abstract (English)
Breast cancer is a leading cause of death among women worldwide, emphasizing the need for early detection and accurate diagnosis. As such Ultrasound Imaging, a reliable and cost-effective tool, is used for this purpose, however the sensitive nature of medical data makes it challenging to develop accurate and private artificial intelligence models. A solution is Federated Learning as it is a promising technique for distributed machine learning on sensitive medical data while preserving patient privacy. However, training on non-Independent and non-Identically Distributed (non-IID) local datasets can impact the accuracy and generalization of the trained model, which is crucial for accurate tumour boundary delineation in BC segmentation. This study aims to tackle this challenge by applying the Federated Proximal (FedProx) method to non-IID Ultrasonic Breast Cancer Imaging datasets. Moreover, we focus on enhancing tumour segmentation accuracy by incorporating a modified U-Net model with attention mechanisms. Our approach resulted in a global model with 96% accuracy, demonstrating the effectiveness of our method in enhancing tumour segmentation accuracy while preserving patient privacy. Our findings suggest that FedProx has the potential to be a promising approach for training precise machine learning models on non-IID local medical datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。