Proceedings

Unlearning during Learning: An Efficient Federated Machine Unlearning Method

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Main Track. Pages 4035-4043.

https://doi.org/10.24963/ijcai.2024/446

Abstract

In recent years, Federated Learning (FL) has garnered significant attention as a distributed machine learning paradigm. To facilitate the implementation of the "right to be forgotten," the concept of federated machine unlearning (FMU) has also emerged. However, current FMU approaches often involve additional time-consuming steps and may not offer comprehensive unlearning capabilities, which renders them less practical in real FL scenarios. In this paper, we introduce FedAU, an innovative and efficient FMU framework aimed at overcoming these limitations. Specifically, FedAU incorporates a lightweight auxiliary unlearning module into the learning process and employs a straightforward linear operation to facilitate unlearning. This approach eliminates the requirement for extra time-consuming steps, rendering it well-suited for FL.

Furthermore, FedAU exhibits remarkable versatility. It not only enables multiple clients to carry out unlearning tasks concurrently but also supports unlearning at various levels of granularity, including individual data samples, specific classes, and even at the client level. We conducted extensive experiments on MNIST, CIFAR10, and CIFAR100 datasets to evaluate the performance of FedAU. The results demonstrate that FedAU effectively achieves the desired unlearning effect while maintaining model accuracy.

  1. [Bourtoule et al., 2021] Lucas Bourtoule, Varun Chan- drasekaran, Christopher A Choquette-Choo, Hengrui Jia, Adelin Travers, Baiwu Zhang, David Lie, and Nicolas Papernot. Machine unlearning. In 2021 IEEE Symposium on Security and Privacy (SP), pages 141-159. IEEE, 2021.
  2. [Che et al., 2023] Tianshi Che, Yang Zhou, Zijie Zhang, Lingjuan Lyu, Ji Liu, Da Yan, Dejing Dou, and Jun Huan. Fast federated machine unlearning with nonlinear func- tional theory. In International conference on machine learning, pages 4241-4268. PMLR, 2023.
  3. [Chen et al., 2023] Min Chen, Weizhuo Gao, Gaoyang Liu, Kai Peng, and Chen Wang. Boundary unlearning: Rapid forgetting of deep networks via shifting the decision boundary. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 7766- 7775, 2023.
  4. [Cheng et al., 2020] Yong Cheng, Yang Liu, Tianjian Chen, and Qiang Yang. Federated learning for privacy- preserving ai. Communications of the ACM, 63(12):33-36, 2020.
  5. [Deng et al., 2020] Yuyang Deng, Mohammad Mahdi Ka- mani, and Mehrdad Mahdavi. Distributionally robust fed- erated averaging. Advances in neural information process- ing systems, 33:15111-15122, 2020.
  6. [Dwork, 2006] Cynthia Dwork. Differential privacy. In In- ternational colloquium on automata, languages, and pro- gramming, pages 1-12. Springer, 2006.
  7. [Eisenhofer et al., 2022] Thorsten Eisenhofer, Doreen Rie- pel, Varun Chandrasekaran, Esha Ghosh, O. Ohrimenko, and Nicolas Papernot. Verifiable and provably secure ma- chine unlearning. ArXiv, abs/2210.09126, 2022.
  8. [Gao et al., 2022] Xiangshan Gao, Xingjun Ma, Jingyi Wang, Youcheng Sun, Bo Li, Shouling Ji, Peng Cheng, and Jiming Chen. Verifi: Towards verifiable federated un- learning. arXiv preprint arXiv:2205.12709, 2022.
  9. [Goldreich, 1998] Oded Goldreich. Secure multi-party com- putation. Manuscript. Preliminary version, 78(110):1- 108, 1998.
  10. [Graves et al., 2021] Laura Graves, Vineel Nagisetty, and Vijay Ganesh. Amnesiac machine learning. In Proceed- ings of the AAAI Conference on Artificial Intelligence, vol- ume 35, pages 11516-11524, 2021.
  11. [Halimi et al., 2022] Anisa Halimi, Swanand Ravindra Kadhe, Ambrish Rawat, and Nathalie Baracaldo Angel. Federated unlearning: How to efficiently erase a client in fl? In International Conference on Machine Learning, 2022.
  12. [Harding et al., 2019] Elizabeth Liz Harding, Jarno J Vanto, Reece Clark, L Hannah Ji, and Sara C Ainsworth. Under- standing the scope and impact of the california consumer privacy act of 2018. Journal of Data Protection & Privacy, 2(3):234-253, 2019.
  13. [He et al., 2016] Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning for image recog- nition. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 770-778, 2016.
  14. [He et al., 2019] Zecheng He, Tianwei Zhang, and Ruby B Lee. Model inversion attacks against collaborative infer- ence. In Proceedings of the 35th Annual Computer Secu- rity Applications Conference, pages 148-162, 2019.
  15. [Jose and Simeone, 2021] Sharu Theresa Jose and O. Sime- one. A unified pac-bayesian framework for machine un- learning via information risk minimization. 2021 IEEE 31st International Workshop on Machine Learning for Sig- nal Processing (MLSP), pages 1-6, 2021.
  16. [Konečnỳ et al., 2015] Jakub Konečnỳ, Brendan McMahan, and Daniel Ramage. Federated optimization: Distributed optimization beyond the datacenter. arXiv preprint arXiv:1511.03575, 2015.
  17. [Krizhevsky et al., 2012] Alex Krizhevsky, Ilya Sutskever, and Geoffrey E Hinton. Imagenet classification with deep convolutional neural networks. In F. Pereira, C.J. Burges, L. Bottou, and K.Q. Weinberger, editors, Advances in Neu- ral Information Processing Systems, volume 25.
  18. Curran Associates, Inc., 2012.
  19. [Krizhevsky et al., 2014] Alex Krizhevsky, Vinod Nair, and Geoffrey Hinton. The cifar-10 dataset. online: http://www. cs. toronto. edu/kriz/cifar. html, 55(5), 2014. Link
  20. [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.
  21. [LeCun et al., 2010] Yann LeCun, Corinna Cortes, and CJ Burges. Mnist handwritten digit database. ATT Labs [Online]. Available: http://yann.lecun.com/exdb/mnist, 2, 2010. Link
  22. [Li et al., 2022] Qinbin Li, Yiqun Diao, Quan Chen, and Bingsheng He. Federated learning on non-iid data silos: An experimental study. In 2022 IEEE 38th International Conference on Data Engineering (ICDE), pages 965-978. IEEE, 2022.
  23. [Liu et al., 2021] Gaoyang Liu, Xiaoqiang Ma, Yang Yang, Chen Wang, and Jiangchuan Liu. Federaser: Enabling ef- ficient client-level data removal from federated learning models. In 2021 IEEE/ACM 29th International Sympo- sium on Quality of Service (IWQOS), pages 1-10. IEEE, 2021.
  24. [Liu et al., 2022a] Yi Liu, Lei Xu, Xingliang Yuan, Cong Wang, and Bo Li. The right to be forgotten in feder- ated learning: An efficient realization with rapid retrain- ing. In IEEE INFOCOM 2022-IEEE Conference on Com- puter Communications, pages 1749-1758. IEEE, 2022.
  25. [Liu et al., 2022b] Yi Liu, Lei Xu, Xingliang Yuan, Cong Wang, and Bo Li. The right to be forgotten in federated learning: An efficient realization with rapid retraining. IEEE INFOCOM 2022 -IEEE Conference on Computer Communications, pages 1749-1758, 2022.
  26. [Liu et al., 2023] Ziyao Liu, Yu Jiang, Jiyuan Shen, Minyi Peng, Kwok-Yan Lam, and Xingliang Yuan. A survey on federated unlearning: Challenges, methods, and future di- rections. arXiv preprint arXiv:2310.20448, 2023.
  27. [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.
  28. [Mercuri et al., 2022] Salvatore Mercuri, Raad Khraishi, Ramin Okhrati, Devesh Batra, Conor Hamill, Taha Ghasempour, and Andrew Nowlan. An introduction to ma- chine unlearning. arXiv preprint arXiv:2209.00939, 2022.
  29. [Nasr et al., 2019] Milad Nasr, Reza Shokri, and Amir Houmansadr. Comprehensive privacy analysis of deep learning: Passive and active white-box inference attacks against centralized and federated learning. In 2019 IEEE symposium on security and privacy (SP), pages 739-753. IEEE, 2019.
  30. [Sekhari et al., 2021] Ayush Sekhari, Jayadev Acharya, Gautam Kamath, and Ananda Theertha Suresh. Remem- ber what you want to forget: Algorithms for machine un- learning. Advances in Neural Information Processing Sys- tems, 34:18075-18086, 2021.
  31. [Su and Li, 2023] Ningxin Su and Baochun Li. Asyn- chronous federated unlearning. In IEEE INFOCOM 2023- IEEE Conference on Computer Communications, pages 1- 10. IEEE, 2023.
  32. [Sun et al., 2022] Tao Sun, Dongsheng Li, and Bao Wang. Decentralized federated averaging. IEEE Transactions on Pattern Analysis and Machine Intelligence, 45(4):4289- 4301, 2022.
  33. [Wang et al., 2022] Junxiao Wang, Song Guo, Xin Xie, and Heng Qi. Federated unlearning via class-discriminative pruning. In Proceedings of the ACM Web Conference 2022, pages 622-632, 2022.
  34. [Wu et al., 2022] Chen Wu, Sencun Zhu, and Prasenjit Mi- tra. Federated unlearning with knowledge distillation. arXiv preprint arXiv:2201.09441, 2022.
  35. [Yang et al., 2019] Qiang Yang, Yang Liu, Tianjian Chen, and Yongxin Tong. Federated machine learning: Concept and applications. ACM Transactions on Intelligent Systems and Technology (TIST), 10(2):1-19, 2019.
  36. [Zhang et al., 2023] Lefeng Zhang, Tianqing Zhu, Haibin Zhang, Ping Xiong, and Wanlei Zhou. Fedrecovery: Dif- ferentially private machine unlearning for federated learn- ing frameworks. IEEE Transactions on Information Foren- sics and Security, 2023.

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

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