Malicious Node Detection and Robust Federated Aggregation Algorithm Based on TD3
摘要
Malicious client nodes in federated learning can corrupt the global model of central server by modifying dataset or submitted model, making it less accurate or distorted. A TD3-based malicious node detection and robust federated aggregation algorithm is proposed. The algorithm excludes malicious clients based on model distance, and uses reinforcement learning to assign model weight to client nodes to reduce the influence of malicious nodes gradually. The experimental results show that the proposed algorithm is effective against data and model modification attacks and is superior to traditional aggregation algorithms such as FedAvg, Krum and MKrum.