PedFed: A performance evaluation-driven federated learning framework for efficient communication

Niu, Ke1; Tai, Wenjuan1; Peng, Xueping2; Guo, Zhongmin1; Zhang, Can1; Li, Heng3

2024-08-01

10.1142/S1793962324400014

INTERNATIONAL JOURNAL OF MODELING SIMULATION AND SCIENTIFIC COMPUTING   影响因子和分区

1793-9623

1793-9615

Protecting healthcare data privacy and security is crucial in advanced manufacturing, which involves medical devices. It encompasses patient records and clinical trial data. Federated learning emerges as a solution that enables model training across different institutions without compromising data privacy and security. However, existing frameworks often exhibit a bias towards clients with larger data volumes, neglecting the connection between global and local model performance. This can result in suboptimal aggregation of the global model, thereby affecting the effectiveness and efficiency of the overall process. To address these limitations, we propose a performance evaluation-driven federated learning framework (PedFed). The primary objective of PedFed is to enhance global model aggregation and improve communication efficiency. Our approach involves a client selection strategy based on performance evaluation of local and global models. Specifically, we introduce the concept of local model improvement (LMI) using Intersection over Union (IoU) for client selection in medical image segmentation scenarios. Moreover, we introduce a dynamic aggregation framework incorporating validation IoU as a weighting factor to mitigate model divergence caused by not independent and identically distributed (non-IID) data. We focus on performing image segmentation tasks to simulate the analysis of sensitive data in the healthcare domain. Experimental results conducted on brain tumor and heart segmentation datasets demonstrate the superiority of the PedFed framework over the baseline framework, confirming its benefits in communication efficiency.

Federated learning data privacy non-IID data medical image segmentation communication efficiency

ESCI ; EI

英语

通讯

Promoting the Classification and Development of Colleges - Student Innovation and Entrepreneurship Training Program (School of Computer)[5112410852]

Computer Science

Computer Science, Theory & Methods

WOS:001292125900001

WORLD SCIENTIFIC PUBL CO PTE LTD

Web of Science

1.Beijing Informat Sci & Technol Univ, Comp Sch, Beijing, Peoples R China
2.Univ Technol Sydney, Australian Artificial Intelligence Inst, Fac Engn & Informat Technol, Sydney, Australia
3.Southern Univ Sci & Technol, Res Inst Trustworthy Autonomous Syst, Shenzhen, Peoples R China

Niu, Ke,Tai, Wenjuan,Peng, Xueping,et al. PedFed: A performance evaluation-driven federated learning framework for efficient communication[J]. INTERNATIONAL JOURNAL OF MODELING SIMULATION AND SCIENTIFIC COMPUTING,2024.

Niu, Ke,Tai, Wenjuan,Peng, Xueping,Guo, Zhongmin,Zhang, Can,&Li, Heng.(2024).PedFed: A performance evaluation-driven federated learning framework for efficient communication.INTERNATIONAL JOURNAL OF MODELING SIMULATION AND SCIENTIFIC COMPUTING.

Niu, Ke,et al."PedFed: A performance evaluation-driven federated learning framework for efficient communication".INTERNATIONAL JOURNAL OF MODELING SIMULATION AND SCIENTIFIC COMPUTING (2024).