• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2024, Vol. 46 ›› Issue (09): 1606-1615.

• Computer Network and Znformation Security • Previous Articles     Next Articles

A secure federated learning scheme based on differential privacy and model clustering

XIAO Di,YU Zhu-yang,LI Min,WANG Lian   

  1. (College of Computer Science,Chongqing University,Chongqing 401331,China)
  • Received:2023-11-14 Revised:2023-12-29 Accepted:2024-09-25 Online:2024-09-25 Published:2024-09-23

Abstract: Model security and clients privacy are urgent challenges to be addressed in federated learning. In order to simultaneously tackle these challenges, a federated learning scheme based on differential privacy and model clustering is proposed. Local differential privacy is introduced in clients updates to protect clients privacy by disrupting the parameters. To ensure precise clustering of noisy model updates, cosine gradient is defined for the first time to cluster noisy model updates. Based on the clustering results, malicious models are accurately identified and filtered. Finally, global differential privacy is introduced to resist potential backdoor attacks. The noise boundary of global noise is obtained by theoretical analysis and it is proved that the total noise introduced by our scheme is lower than that introduced by the classical model security scheme. The experimental results demonstrate that our scheme can achieve the expected goals in terms of accuracy, robustness and privacy.

Key words: federated learning, model security, backdoor attack, differential privacy, privacy protection