Fedmrl: Data Heterogeneity Aware Federated Multi-agent Deep Reinforcement Learning For Medical Imaging

Abstract

Despite recent advancements in federated learning (FL) for medical image diagnosis, addressing data heterogeneity among clients remains a significant challenge for practical implementation. A primary hurdle in FL arises from the non-IID nature of data samples across clients, which typically results in a decline in the performance of the aggregated global model. In this study, we introduce FedMRL, a novel federated multi-agent deep reinforcement learning framework designed to address data heterogeneity. FedMRL incorporates a novel loss function to facilitate fairness among clients, preventing bias in the final global model. Additionally, it employs a multi-agent reinforcement learning (MARL) approach to calculate the proximal term \((\mu)\) for the personalized local objective function, ensuring convergence to the global optimum. Furthermore, FedMRL integrates an adaptive weight adjustment method using a Self-organizing map (SOM) on the server side to counteract distribution shifts amon

Stats

• citations0
• S2 citations—
• github stars0
• HF likes0
• heat score0.00
• arxiv keysahoo2024fedmrl

Related papers

• Heterogeneity-aware Personalized Federated Learning Via Adaptive Dual-agent Reinforcement Learning (2025)0.00
• A Multi-agent Reinforcement Learning Approach For Efficient Client Selection In Federated Learning (2022)15.00
• Optimized Local Updates In Federated Learning Via Reinforcement Learning (2025)0.00
• Adaptive Incentive For Cross-silo Federated Learning: A Multi-agent Reinforcement Learning Approach (2023)10.74
• A Fair Federated Learning Framework With Reinforcement Learning (2022)0.00
• Federated Offline Reinforcement Learning (2022)0.00
• On The Linear Speedup Of Personalized Federated Reinforcement Learning With Shared Representations (2024)0.00
• Dearfsac: An Approach To Optimizing Unreliable Federated Learning Via Deep Reinforcement Learning (2022)0.00