Proceedings

A Multi-Granularity Clustering Approach for Federated Backdoor Defense with the Adam Optimizer

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 6931-6939.

https://doi.org/10.24963/ijcai.2025/771

Abstract

Federated learning is vulnerable to backdoor attacks due to its distributed nature and the inability to access local datasets. Meanwhile, the heterogeneity of distributed data further complicates the detection of such attacks. However, existing defense strategies often overlook the presence of non-stationary objectives and noisy gradients across multiple clients, making it challenging to accurately and efficiently identify malicious participants. To address these challenges, we propose a backdoor defense method for Federated Learning with Adam optimizer and multi-granularity Clustering (FLAC), incorporating both coarse-grained and fine-grained clustering mechanisms to neutralize backdoor attacks. First, the Adam optimizer accelerates the learning process by mitigating the impact of noisy gradients and addressing the non-stationary objectives posed by different clients under attack. Second, a multi-granularity clustering process is considered to differentiate between benign clients and potential attackers. This is followed by an adaptive clipping strategy to further alleviate the influence of malicious attackers. Our theoretical analysis demonstrates the consistent convergence of Adam in a federated backdoor defense environment. Extensive experimental results validate the effectiveness of our defense approach.

  1. [Azulay et al., 2020] Shahar Azulay, Lior Raz, Amir Glober- son, Tomer Koren, and Yehuda Afek. Holdout sgd: Byzan- tine tolerant federated learning. arXiv preprint arXiv: 2008.04612, 2020.
  2. [Bagdasaryan et al., 2020] Eugene Bagdasaryan, Andreas Veit, Yiqing Hua, Deborah Estrin, and Vitaly Shmatikov. How to backdoor federated learning. In International Conference on Artificial Intelligence and Statistics, pages 2938-2948. PMLR, 2020.
  3. [Bhagoji et al., 2019] Arjun Nitin Bhagoji, Supriyo Chakraborty, Prateek Mittal, and Seraphin Calo. Ana- lyzing federated learning through an adversarial lens. In International Conference on Machine Learning, pages 634-643. PMLR, 2019.
  4. [Biggio et al., 2012] Battista Biggio, Blaine Nelson, and Pavel Laskov. Poisoning attacks against support vector machines. arXiv preprint arXiv:1206.6389, 2012.
  5. [Blanchard et al., 2017] Peva Blanchard, El Mahdi El Mhamdi, Rachid Guerraoui, and Julien Stainer. Machine learning with adversaries: Byzantine toler- ant gradient descent. Advances in neural information processing systems, 30, 2017.
  6. [Cao et al., 2021] Xiaoyu Cao, Jinyuan Jia, and Neil Zhen- qiang Gong. Data poisoning attacks to local differential privacy protocols. In 30th USENIX Security Symposium, pages 947-964. USENIX Association, 2021.
  7. [Chen et al., 2023] Zekai Chen, Shengxing Yu, Mingyuan Fan, Ximeng Liu, and Robert H Deng. Privacy-enhancing and robust backdoor defense for federated learning on heterogeneous data. IEEE Transactions on Information Forensics and Security, 2023.
  8. [Chen et al., 2024] Shengbo Chen, Jidong Yuan, Zhihai Wang, and Yongqi Sun. Local perturbation-based black- box federated learning attack for time series classifica- tion. Future Generation Computer Systems, 158:488-500, 2024.
  9. [Fung et al., 2018] Clement Fung, Chris JM Yoon, and Ivan Beschastnikh. Mitigating sybils in federated learning poi- soning. arXiv preprint arXiv:1808.04866, 2018.
  10. [Geyer et al., 2017] Robin C Geyer, Tassilo Klein, and Moin Nabi. Differentially private federated learning: A client level perspective. arXiv preprint arXiv:1712.07557, 2017.
  11. [Gu and Bai, 2023] Yu-Hao Gu and Yue-Bin Bai. Survey on security and privacy of federated learning models. Ruan Jian Xue Bao/Journal of Software (in Chinese), 34(6):2833-2864, 2023.
  12. [Gu et al., 2024] Hongyan Gu, Xinyi Zhang, Jiang Li, Hui Wei, Baiqi Li, and Xinli Huang. Federated learning vul- nerabilities: Privacy attacks with denoising diffusion prob- abilistic models. In Proceedings of the ACM on Web Con- ference 2024, pages 1149-1157, 2024.
  13. [Jebreel et al., 2023] Najeeb Moharram Jebreel, Josep Domingo-Ferrer, David Sánchez, and Alberto Blanco- Justicia. Lfighter: Defending against the label-flipping attack in federated learning. Neural Networks, 170:111- 126, 2023.
  14. [Jiang et al., 2023] Yujing Jiang, Xingjun Ma, Sarah Mon- azam Erfani, and James Bailey. Backdoor attacks on time series: A generative approach. In 2023 IEEE Conference on Secure and Trustworthy Machine Learning (SaTML), pages 392-403. IEEE, 2023.
  15. [Kasyap and Tripathy, 2024] Harsh Kasyap and Somanath Tripathy. Beyond data poisoning in federated learning. Ex- pert Systems with Applications, 235:121192, 2024.
  16. [Kingma, 2014] Diederik P Kingma. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980, 2014.
  17. [Krizhevsky et al., 2009] Alex Krizhevsky, Geoffrey Hinton, et al. Learning multiple layers of features from tiny im- ages. 2009.
  18. [Kruskal, 1956] Joseph B Kruskal. On the shortest span- ning subtree of a graph and the traveling salesman prob- lem. Proceedings of the American Mathematical society, 7(1):48-50, 1956.
  19. [LeCun et al., 1998] Yann LeCun, Léon Bottou, Yoshua Bengio, and Patrick Haffner. Gradient-based learning ap- plied to document recognition. Proceedings of the IEEE, 86(11):2278-2324, 1998.
  20. [Li et al., 2022] Chengxi Li, Gang Li, and Pramod K Varsh- ney. Federated learning with soft clustering. IEEE Internet of Things Journal, 9(10):7773-7782, 2022.
  21. [Li et al., 2023a] Haoyang Li, Qingqing Ye, Haibo Hu, Jin Li, Leixia Wang, Chengfang Fang, and Jie Shi. 3dfed: Adaptive and extensible framework for covert backdoor attack in federated learning. In 2023 IEEE Symposium on Security and Privacy (SP), pages 1893-1907. IEEE, 2023.
  22. [Li et al., 2023b] Henger Li, Chen Wu, Senchun Zhu, and Zizhan Zheng. Learning to backdoor federated learning. arXiv preprint arXiv:2303.03320, 2023.
  23. [Lyu et al., 2023] Xiaoting Lyu, Yufei Han, Wei Wang, Jingkai Liu, Bin Wang, Jiqiang Liu, and Xiangliang Zhang. Poisoning with cerberus: Stealthy and colluded backdoor attack against federated learning. In Proceed- ings of the AAAI Conference on Artificial Intelligence, vol- ume 37, pages 9020-9028, 2023.
  24. [McMahan et al., 2017] Brendan McMahan, Eider Moore, Daniel Ramage, Seth Hampson, and Blaise Aguera y Ar- cas. Communication-efficient learning of deep networks from decentralized data. In Artificial intelligence and statistics, pages 1273-1282. PMLR, 2017.
  25. [Nguyen et al., 2021] Thien Duc Nguyen, Phillip Rieger, Huili Chen, Hossein Yalame, Helen Möllering, Hossein Fereidooni, Samuel Marchal, Markus Miettinen, Azalia Mirhoseini, Shaza Zeitouni, et al. Flame: Taming back- doors in federated learning (extended version 1). arXiv preprint arXiv: 2101.02281, 2021.
  26. [Nguyen et al., 2022] Dinh C Nguyen, Quoc-Viet Pham, Pubudu N Pathirana, Ming Ding, Aruna Seneviratne, Zi- huai Lin, Octavia Dobre, and Won-Joo Hwang. Federated learning for smart healthcare: A survey. ACM Computing Surveys (Csur), 55(3):1-37, 2022.
  27. [Pillutla et al., 2022] Krishna Pillutla, Sham M Kakade, and Zaid Harchaoui. Robust aggregation for federated learn- ing. IEEE Transactions on Signal Processing, 70:1142- 1154, 2022.
  28. [Portnoy et al., 2022] Amit Portnoy, Yoav Tirosh, and Danny Hendler. Towards federated learning with byzantine-robust client weighting. Applied Sciences, 12(17), 2022.
  29. [Reddi et al., 2019] Sashank J Reddi, Satyen Kale, and San- jiv Kumar. On the convergence of adam and beyond. arXiv preprint arXiv: 1904.09237, 2019.
  30. [Rodríguez-Barroso et al., 2023] Nuria Rodríguez-Barroso, Daniel Jiménez-López, M. Victoria Luzón, Francisco Her- rera, and Eugenio Martínez-Cámara. Survey on federated learning threats: Concepts, taxonomy on attacks and de- fences, experimental study and challenges. Information Fusion, 90:148-173, 2023.
  31. [Sattler et al., 2020] Felix Sattler, Klaus-Robert Müller, Thomas Wiegand, and Wojciech Samek. On the byzan- tine robustness of clustered federated learning. In ICASSP 2020 -2020 IEEE International Conference on Acoustics, Speech and Signal Processing, pages 8861-8865, 2020.
  32. [Schubert et al., 2017] Erich Schubert, Jörg Sander, Martin Ester, Hans Peter Kriegel, and Xiaowei Xu. Dbscan re- visited, revisited: why and how you should (still) use db- scan. ACM Transactions on Database Systems (TODS), 42(3):1-21, 2017.
  33. [Shazeer and Stern, 2018] Noam Shazeer and Mitchell Stern. Adafactor: Adaptive learning rates with sublinear memory cost. In Jennifer Dy and Andreas Krause, editors, Proceedings of the 35th International Conference on Machine Learning, volume 80 of Proceedings of Machine Learning Research, pages 4596-4604. PMLR, 10-15 Jul 2018.
  34. [Sun et al., 2019] Ziteng Sun, Peter Kairouz, Ananda Theertha Suresh, and H Brendan McMahan. Can you really backdoor federated learning? arXiv preprint arXiv:1911.07963, 2019.
  35. [Tolpegin et al., 2020] Vale Tolpegin, Stacey Truex, Mehmet Emre Gursoy, and Ling Liu. Data poisoning attacks against federated learning systems. In Computer Security-ESORICS 2020: 25th European Symposium on Research in Computer Security, ESORICS 2020, Guildford, UK, September 14-18, 2020, Proceedings, Part I 25, pages 480-501. Springer, 2020.
  36. [Wang et al., 2020a] Hongyi Wang, Kartik Sreenivasan, Shashank Rajput, Harit Vishwakarma, Saurabh Agarwal, Jy-yong Sohn, Kangwook Lee, and Dimitris Papailiopou- los. Attack of the tails: Yes, you really can backdoor feder- ated learning. Advances in Neural Information Processing Systems, 33:16070-16084, 2020.
  37. [Wang et al., 2020b] Hongyi Wang, Mikhail Yurochkin, Yuekai Sun, Dimitris Papailiopoulos, and Yasaman Khaz- aeni. Federated learning with matched averaging. arXiv preprint arXiv:2002.06440, 2020.
  38. [Wang et al., 2023] Yongkang Wang, Di-Hua Zhai, Yong- ping He, and Yuanqing Xia. An adaptive robust defend- ing algorithm against backdoor attacks in federated learn- ing. Future Generation Computer Systems, 143:118-131, 2023.
  39. [Xiao et al., 2017] Han Xiao, Kashif Rasul, and Roland Vollgraf. Fashion-mnist: a novel image dataset for bench- marking machine learning algorithms. arXiv preprint arXiv:1708.07747, 2017.
  40. [Yang et al., 2022] Wenbo Yang, Jidong Yuan, Xiaokang Wang, and Peixiang Zhao. Tsadv: Black-box adversarial attack on time series with local perturbations. Engineering Applications of Artificial Intelligence, 114:105218, 2022.
  41. [Yin et al., 2018] Dong Yin, Yudong Chen, Ramchandran Kannan, and Peter Bartlett. Byzantine-robust distributed learning: Towards optimal statistical rates. In Interna- tional Conference on Machine Learning, pages 5650- 5659. PMLR, 2018.
  42. [Zhang et al., 2019] Jiale Zhang, Junjun Chen, Di Wu, Bing Chen, and Shui Yu. Poisoning attack in federated learn- ing using generative adversarial nets. In 2019 18th IEEE international conference on trust, security and privacy in computing and communications/13th IEEE international conference on big data science and engineering (Trust- Com/BigDataSE), pages 374-380. IEEE, 2019.
  43. [Zhang et al., 2020] Jiale Zhang, Bing Chen, Xiang Cheng, Huynh Thi Thanh Binh, and Shui Yu. Poisongan: Genera- tive poisoning attacks against federated learning in edge computing systems. IEEE Internet of Things Journal, 8(5):3310-3322, 2020.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards