Proceedings

Label Leakage in Vertical Federated Learning: A Survey

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Survey Track. Pages 8160-8169.

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

Abstract

Vertical federated learning (VFL) is a distributed machine learning paradigm that collaboratively trains models using passive parties with features and an active party with additional labels. While VFL offers privacy preservation through data localization, the threat of label leakage remains a significant challenge. Label leakage occurs due to label inference attacks, where passive parties attempt to infer labels for their privacy and commercial value. Extensive research has been conducted on this specific VFL attack, but a comprehensive summary is still lacking. To bridge this gap, our paper aims to survey the existing label inference attacks and defenses. We propose two new taxonomies for both label inference attacks and defenses, respectively. Beyond summarizing the current state of research, we highlight techniques that we believe hold potential and could significantly influence future studies. Moreover, experimental benchmark datasets and evaluation metrics are summarized to provide a guideline for subsequent work.

  1. [Abadi et al., 2016] Martin Abadi, Andy Chu, Ian Goodfel- low, H Brendan McMahan, Ilya Mironov, Kunal Talwar, and Li Zhang. Deep learning with differential privacy. In CCS, pages 308-318, 2016.
  2. [Abuadbba et al., 2020] Sharif Abuadbba, Kyuyeon Kim, Minki Kim, Chandra Thapa, Seyit A Camtepe, Yansong Gao, Hyoungshick Kim, and Surya Nepal. Can we use split learning on 1d cnn models for privacy preserving training? In AsiaCCS, pages 305-318, 2020.
  3. [Aggarwal et al., 2021] Abhinav Aggarwal, Shiva Ka- siviswanathan, Zekun Xu, Oluwaseyi Feyisetan, and Nathanael Teissier. Label inference attacks from log-loss scores. In ICML, pages 120-129, 2021.
  4. [Almanifi et al., 2023] Omair Rashed Abdulwareth Alman- ifi, Chee-Onn Chow, Mau-Luen Tham, Joon Huang Chuah, and Jeevan Kanesan. Communication and compu- tation efficiency in federated learning: A survey. Internet of Things, 22:100742, 2023.
  5. [Arazzi et al., 2023] Marco Arazzi, Mauro Conti, Stefanos Koffas, Marina Krcek, Antonino Nocera, Stjepan Picek, and Jing Xu. BlindSage: Label inference attacks against node-level vertical federated graph neural networks. arXiv preprint arXiv:2308.02465, 2023.
  6. [Bai et al., 2023] Yijie Bai, Yanjiao Chen, Hanlei Zhang, Wenyuan Xu, Haiqin Weng, and Dou Goodman. VIL- LAIN: Backdoor attacks against vertical split learning. In USENIX Security, pages 2743-2760, 2023.
  7. [Busa-Fekete et al., 2021] Robert Istvan Busa-Fekete, Umar Syed, and Sergei Vassilvitskii. On the pitfalls of label dif- ferential privacy. In NeurIPS Workshop, 2021.
  8. [Cheng et al., 2021] Kewei Cheng, Tao Fan, Yilun Jin, Yang Liu, Tianjian Chen, Dimitrios Papadopoulos, and Qiang Yang. SecureBoost: A lossless federated learning frame- work. IEEE Intelligent Systems, 36(6):87-98, 2021.
  9. [Cheu et al., 2019] Albert Cheu, Adam Smith, Jonathan Ull- man, David Zeber, and Maxim Zhilyaev. Distributed dif- ferential privacy via shuffling. In EUROCRYPT, pages 375-403, 2019.
  10. [Dwork, 2006] Cynthia Dwork. Differential privacy. In ICALP, 2006.
  11. [Erdogan et al., 2022] Ege Erdogan, Alptekin Küpc ¸ü, and A Ercüment C ¸ic ¸ek. UnSplit: Data-oblivious model inver- sion, model stealing, and label inference attacks against split learning. In WPES, pages 115-124, 2022.
  12. [Erlingsson et al., 2019] Úlfar Erlingsson, Vitaly Feldman, Ilya Mironov, Ananth Raghunathan, Kunal Talwar, and Abhradeep Thakurta. Amplification by shuffling: From lo- cal to central differential privacy via anonymity. In SODA, pages 2468-2479, 2019.
  13. [Fu et al., 2022a] Chong Fu, Xuhong Zhang, Shouling Ji, Jinyin Chen, Jingzheng Wu, Shanqing Guo, Jun Zhou, Alex X Liu, and Ting Wang. Label inference attacks against vertical federated learning. In USENIX Security, pages 1397-1414, 2022.
  14. [Fu et al., 2022b] Fangcheng Fu, Huanran Xue, Yong Cheng, Yangyu Tao, and Bin Cui. BlindFL: Vertical feder- ated machine learning without peeking into your data. In SIGMOD, pages 1316-1330, 2022.
  15. [Ghazi et al., 2021] Badih Ghazi, Noah Golowich, Ravi Ku- mar, Pasin Manurangsi, and Chiyuan Zhang. Deep learn- ing with label differential privacy. In NeurIPS, pages 27131-27145, 2021.
  16. [Gu et al., 2023] Hanlin Gu, Jiahuan Luo, Yan Kang, Lixin Fan, and Qiang Yang. Fedpass: Privacy-preserving verti- cal federated deep learning with adaptive obfuscation. In IJCAI, pages 3759-3767, 2023. IJCAI
  17. [Kairouz et al., 2021] Peter Kairouz, H. Brendan McMa- han, Brendan Avent, Aurélien Bellet, Mehdi Bennis, Ar- jun Nitin Bhagoji, Kallista Bonawit, Zachary Charles, Graham Cormode, Rachel Cummings, Rafael G. L. D'Oliveira, Hubert Eichner, Salim El Rouayheb, David Evans, Josh Gardner, Zachary Garrett, Adrià Gascón, Badih Ghazi, Phillip B. Gibbons, Marco Gruteser, Zaid Harchaoui, Chaoyang He, Lie He, Zhouyuan Huo, Ben Hutchinson, Justin Hsu, Martin Jaggi, Tara Javidi, Gauri Joshi, Mikhail Khodak, Jakub Konecný, Aleksandra Ko- rolova, Farinaz Koushanfar, Sanmi Koyejo, Tancrède Le- point, Yang Liu, Prateek Mittal, Mehryar Mohri, Richard Nock, Ayfer Özgür, Rasmus Pagh, Hang Qi, Daniel Ra- mage, Ramesh Raskar, Mariana Raykova, Dawn Song, Weikang Song, Sebastian U. Stich, Ziteng Sun, Ananda Theertha Suresh, Florian Tramèr, Praneeth Vepakomma, Jianyu Wang, Li Xiong, Zheng Xu, Qiang Yang, Felix X. Yu, Han Yu, and Sen Zhao. Advances and open problems in federated learning. Foundations and Trends® in Ma- chine Learning, 14(1-2):1-210, 2021.
  18. [Kariyappa and Qureshi, 2023] Sanjay Kariyappa and Moin- uddin K Qureshi. ExPLoit: Extracting private labels in split learning. In SaTML, pages 165-175, 2023.
  19. [Knott et al., 2021] Brian Knott, Shobha Venkataraman, Awni Hannun, Shubho Sengupta, Mark Ibrahim, and Lau- rens van der Maaten. CrypTen: Secure multi-party com- putation meets machine learning. In NeurIPS, volume 34, pages 4961-4973, 2021.
  20. [Krizhevsky, 2009] Alex Krizhevsky. Learning multiple lay- ers of features from tiny images. 2009.
  21. [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.
  22. [Li et al., 2022] Oscar Li, Jiankai Sun, Xin Yang, Weihao Gao, Hongyi Zhang, Junyuan Xie, Virginia Smith, and Chong Wang. Label leakage and protection in two-party split learning. In ICLR, 2022.
  23. [Li et al., 2023] Songze Li, Duanyi Yao, and Jin Liu. FedVS: Straggler-resilient and privacy-preserving vertical feder- ated learning for split models. In ICML, pages 20296- 20311, 2023.
  24. [Liu and Lyu, 2022] Junlin Liu and Xinchen Lyu. Clustering label inference attack against practical split learning. arXiv preprint arXiv:2203.05222, 2022.
  25. [Liu et al., 2020] Yang Liu, Zhihao Yi, and Tianjian Chen. Backdoor attacks and defenses in feature-partitioned col- laborative learning. arXiv preprint arXiv:2007.03608, 2020.
  26. [Liu et al., 2024] Yang Liu, Yan Kang, Tianyuan Zou, Yan- hong Pu, Yuanqin He, Xiaozhou Ye, Ye Ouyang, Ya-Qin Zhang, and Qiang Yang. Vertical federated learning: Con- cepts, advances and challenges. IEEE Transactions on Knowledge and Data Engineering, pages 1-20, 2024.
  27. [McMahan et al., 2016] H Brendan McMahan, Eider Moore, Daniel Ramage, and Blaise Agüera y Arcas. Federated learning of deep networks using model averaging. arXiv preprint arXiv:1602.05629, 2016.
  28. [Mo et al., 2021] Fan Mo, Hamed Haddadi, Kleomenis Kat- evas, Eduard Marin, Diego Perino, and Nicolas Kourtellis. PPFL: privacy-preserving federated learning with trusted execution environments. In MobiSys, pages 94-108, 2021.
  29. [Mohassel and Zhang, 2017] Payman Mohassel and Yupeng Zhang. SecureML: A system for scalable privacy- preserving machine learning. In S&P, pages 19-38, 2017.
  30. [Naseri et al., 2023] Mohammad Naseri, Yufei Han, and Emiliano De Cristofaro. BadVFL: Backdoor at- tacks in vertical federated learning. arXiv preprint arXiv:2304.08847, 2023.
  31. [Pasquini et al., 2021] Dario Pasquini, Giuseppe Ateniese, and Massimo Bernaschi. Unleashing the tiger: Inference attacks on split learning. In CCS, pages 2113-2129, 2021.
  32. [Qiu et al., 2023] Haoze Qiu, Fei Zheng, Chaochao Chen, and Xiaolin Zheng. Defending label inference attacks in split learning under regression setting. arXiv preprint arXiv:2308.09448, 2023.
  33. [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.
  34. [Shokri and Shmatikov, 2015] Reza Shokri and Vitaly Shmatikov. Privacy-preserving deep learning. In CCS, pages 1310-1321, 2015.
  35. [Soltani et al., 2023] Behnaz Soltani, Yipeng Zhou, Venus Haghighi, and John Lui. A survey of federated evaluation in federated learning. In IJCAI, pages 6769-6777, 2023. IJCAI
  36. [Sun et al., 2021] Jiankai Sun, Yuanshun Yao, Weihao Gao, Junyuan Xie, and Chong Wang. Defending against re- construction attack in vertical federated learning. arXiv preprint arXiv:2107.09898, 2021.
  37. [Sun et al., 2022] Jiankai Sun, Xin Yang, Yuanshun Yao, and Chong Wang. Label leakage and protection from forward embedding in vertical federated learning. arXiv preprint arXiv:2203.01451, 2022.
  38. [Székely et al., 2007] Gábor J Székely, Maria L Rizzo, and Nail K Bakirov. Measuring and testing dependence by cor- relation of distances. The Annals of Statistics, 35(6):2769- 2794, 2007.
  39. [Takahashi et al., 2023] Hideaki Takahashi, Jingjing Liu, and Yang Liu. Eliminating label leakage in tree-based ver- tical federated learning. arXiv preprint arXiv:2307.10318, 2023.
  40. [Tan et al., 2022] Juntao Tan, Lan Zhang, Yang Liu, Anran Li, and Ye Wu. Residue-based label protection mecha- nisms in vertical logistic regression. In BigCom, pages 356-364, 2022.
  41. [Tran et al., 2018] Brandon Tran, Jerry Li, and Aleksander Madry. Spectral signatures in backdoor attacks. In NeurIPS, pages 8011-8021, 2018.
  42. [Vepakomma et al., 2018] Praneeth Vepakomma, Otkrist Gupta, Tristan Swedish, and Ramesh Raskar. Split learn- ing for health: Distributed deep learning without sharing raw patient data. arXiv preprint arXiv:1812.00564, 2018.
  43. [Vepakomma et al., 2019] Praneeth Vepakomma, Otkrist Gupta, Abhimanyu Dubey, and Ramesh Raskar. Reducing leakage in distributed deep learning for sensitive health data. In ICLR Workshop, 2019.
  44. [Wainakh et al., 2021] Aidmar Wainakh, Till Müßig, Tim Grube, and Max Mühlhäuser. Label leakage from gradi- ents in distributed machine learning. In CCNC, pages 1-4, 2021.
  45. [Wainakh et al., 2022] Aidmar Wainakh, Fabrizio Ventola, Till Müßig, Jens Keim, Carlos Garcia Cordero, Ephraim Zimmer, Tim Grube, Kristian Kersting, and Max Mühlhäuser. User-level label leakage from gradients in federated learning. In PETS, pages 227-244, 2022.
  46. [Wen et al., 2023] Jie Wen, Zhixia Zhang, Yang Lan, Zhihua Cui, Jianghui Cai, and Wensheng Zhang. A survey on fed- erated learning: challenges and applications. International Journal of Machine Learning and Cybernetics, 14(2):513- 535, 2023.
  47. [Wu et al., 2023a] Ruihan Wu, Jin Peng Zhou, et al. Does label differential privacy prevent label inference attacks? In AISTATS, pages 4336-4347, 2023.
  48. [Wu et al., 2023b] Yuncheng Wu, Naili Xing, Gang Chen, Tien Tuan Anh Dinh, Zhaojing Luo, Beng Chin Ooi, Xiaokui Xiao, and Meihui Zhang. Falcon: A privacy- preserving and interpretable vertical federated learn- ing system. Proceedings of the VLDB Endowment, 16(10):2471-2484, 2023.
  49. [Xiao et al., 2017] Han Xiao, Kashif Rasul, and Roland Vollgraf. Fashion-MNIST: a novel image dataset for benchmarking machine learning algorithms. arXiv preprint arXiv:1708.07747, 2017.
  50. [Xie et al., 2023] Shangyu Xie, Xin Yang, Yuanshun Yao, Tianyi Liu, Taiqing Wang, and Jiankai Sun. Label infer- ence attack against split learning under regression setting. arXiv preprint arXiv:2301.07284, 2023.
  51. [Xue et al., 2024] Rui Xue, Kaiping Xue, Bin Zhu, Xinyi Luo, Tianwei Zhang, Qibin Sun, and Jun Lu. Differentially private federated learning with an adaptive noise mecha- nism. IEEE Transactions on Information Forensics and Security, 19:74-87, 2024.
  52. [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, 10(2):1-19, 2019.
  53. [Zeng et al., 2023] Dun Zeng, Siqi Liang, Xiangjing Hu, Hui Wang, and Zenglin Xu. Fedlab: A flexible federated learn- ing framework. Journal of Machine Learning Research, 24(100):1-7, 2023.
  54. [Zhang et al., 2022] Xiaoxue Zhang, Xiuhua Zhou, and Kongyang Chen. Data leakage with label reconstruction in distributed learning environments. In ML4CS, pages 185- 197, 2022.
  55. [Zhao et al., 2020] Bo Zhao, Konda Reddy Mopuri, and Hakan Bilen. iDLG: Improved deep leakage from gradi- ents. arXiv preprint arXiv:2001.02610, 2020.
  56. [Zheng et al., 2022] Fei Zheng, Chaochao Chen, Binhui Yao, and Xiaolin Zheng. Making split learning resilient to label leakage by potential energy loss. arXiv preprint arXiv:2210.09617, 2022.
  57. [Zhou et al., 2022] Jun Zhou, Longfei Zheng, Chaochao Chen, Yan Wang, Xiaolin Zheng, Bingzhe Wu, Cen Chen, Li Wang, and Jianwei Yin. Toward scalable and privacy-preserving deep neural network via algorithmic- cryptographic co-design. ACM Transactions on Intelligent Systems and Technology, 13(4):1-21, 2022.
  58. [Zhu et al., 2019] Ligeng Zhu, Zhijian Liu, and Song Han. Deep leakage from gradients. In NeurIPS, pages 14774- 14784, 2019.
  59. [Zou et al., 2022] Tianyuan Zou, Yang Liu, Yan Kang, Wen- han Liu, Yuanqin He, Zhihao Yi, and Qiang Yang. De- fending batch-level label inference and replacement at- tacks in vertical federated learning. IEEE Transactions on Big Data, pages 1-12, 2022.
  60. [Zou et al., 2023] Tianyuan Zou, Yang Liu, and Ya-Qin Zhang. Mutual information regularization for vertical fed- erated learning. arXiv preprint arXiv:2301.01142, 2023.

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

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