Adaptive Federated Learning Based on Device Performance in a Heterogeneous Environment of Medical Computing Devices
摘要
Federated Learning (FL), as one of the effective methods to solve the problem of medical data silos, can promote mutual cooperation among medical institutions under the premise of safeguarding the privacy and security of medical data, and effectively promote the development of intelligent medical treatment. However, the characteristics of medical data with multiple sources and heterogeneity bring certain challenges to the application of federated learning in the medical field. Due to the wide range of medical data sources, some medical institutions are unable to complete the federated learning training task on time due to the limitation of device performance, which leads to the failure of federated learning training. To address the problem of heterogeneity of computing devices among medical institutions that may cause some clients to fall behind, we propose an Adaptive Federated Learning based on Device Performance (AFedDP) method. Specifically, we design an adaptive local model training method to reduce the computational burden of medical institutions with poor equipment performance in the local model training process by adaptively constructing a local training model. Additionally, we propose a hierarchical model aggregation algorithm based on dynamic weight. This allows the heterogeneous adaptive local model to actively participate in the global model aggregation, making full use of the local data of each medical institution and improving the performance of the global model. Finally, we design an aggregation weight update algorithm based on momentum optimization to enhance the stability of global model aggregation. This algorithm updates contribution and aggregation weight with the idea of momentum optimization. Simulation experiments demonstrate that AFedDP achieves higher accuracy compared to the baseline algorithm. AFedDP can better address the issue of some devices being unable to participate in federated training properly.