Publications
2024
- ICLR 2024VertiBench: Advancing Feature Distribution Diversity in Vertical Federated Learning BenchmarksZhaomin Wu, Junyi Hou, and Bingsheng HeIn The Twelfth International Conference on Learning Representations , 2024
Vertical Federated Learning (VFL) is a crucial paradigm for training machine learning models on feature-partitioned, distributed data. However, due to privacy restrictions, few public real-world VFL datasets exist for algorithm evaluation, and these represent a limited array of feature distributions. Existing benchmarks often resort to synthetic datasets, derived from arbitrary feature splits from a global set, which only capture a subset of feature distributions, leading to inadequate algorithm performance assessment. This paper addresses these shortcomings by introducing two key factors affecting VFL performance - feature importance and feature correlation - and proposing associated evaluation metrics and dataset splitting methods. Additionally, we introduce a real VFL dataset to address the deficit in image-image VFL scenarios. Our comprehensive evaluation of cutting-edge VFL algorithms provides valuable insights for future research in the field.
@inproceedings{wu2024vertibench, title = {VertiBench: Advancing Feature Distribution Diversity in Vertical Federated Learning Benchmarks}, author = {Wu, Zhaomin and Hou, Junyi and He, Bingsheng}, booktitle = {The Twelfth International Conference on Learning Representations}, year = {2024}, url = {https://openreview.net/forum?id=glwwbaeKm2}, google_scholar_id = {_FxGoFyzp5QC} }
2023
- SIGMOD 2023DeltaBoost: Gradient Boosting Decision Trees with Efficient Machine UnlearningZhaomin Wu, Junhui Zhu, Qinbin Li, and Bingsheng HeProc. ACM Manag. Data, 2023
DeltaBoost has won Honorable Mention for Best Artifact Award in SIGMOD 2023
As machine learning (ML) has been widely developed in real-world applications, the privacy of ML models draws an increasing concern. In this paper, we study how to forget specific data records from ML models to preserve the privacy of these data. Although some studies propose efficient unlearning algorithms on random forests and extremely randomized trees, Gradient Boosting Decision Trees (GBDT), which are widely used in practice, have not been explored. The efficient unlearning of GBDT faces two major challenges: 1) the training of each tree is deterministic and non-robust; 2) the training of a tree depends on all the previous trees. To solve the first challenge, we propose a robust GBDT-like ML model DeltaBoost that enables efficient and accurate deletion according to our theoretical analysis. For the second challenge, we design a training algorithm for DeltaBoost that minimizes the dependency among trees. Our experiments on five datasets demonstrate that DeltaBoost can remove data records from the trained model efficiently and effectively. Our unlearning approach achieves up to two orders of magnitude speedup compared to retraining GBDT. Besides, DeltaBoost produces competitive performance to existing decision-tree-based ML models.
@article{10.1145/3589313, author = {Wu, Zhaomin and Zhu, Junhui and Li, Qinbin and He, Bingsheng}, title = {DeltaBoost: Gradient Boosting Decision Trees with Efficient Machine Unlearning}, year = {2023}, issue_date = {June 2023}, publisher = {Association for Computing Machinery}, address = {New York, NY, USA}, volume = {1}, number = {2}, url = {https://doi.org/10.1145/3589313}, doi = {10.1145/3589313}, journal = {Proc. ACM Manag. Data}, articleno = {168}, numpages = {26}, keywords = {data deletion, gradient boosting decision trees, machine unlearning}, google_scholar_id = {ufrVoPGSRksC} }
- MLSys 2023FedTree: A Federated Learning System For TreesQinbin Li, Zhaomin Wu, Yanzheng Cai, yuxuan han, Ching Man Yung, and 2 more authorsIn Proceedings of Machine Learning and Systems , 2023
While the quality of machine learning services largely relies on the volume of training data, data regulations such as the General Data Protection Regulation (GDPR) impose stringent requirements on data transfer. Federated learning has emerged as a popular approach for enabling collaborative machine learning without sharing raw data. To facilitate the rapid development of federated learning, efficient and user-friendly federated learning systems are essential. Despite many existing federated learning systems designed for deep learning, tree-based federated learning systems have not been well exploited. This paper presents a tree-based federated learning system under a histogram-sharing scheme, named FedTree, that supports both horizontal and vertical federated training of GBDTs with configurable privacy protection techniques. Our extensive experiments show that FedTree achieves competitive accuracy to centralized training while incurring much less computational cost than the other generic federated learning systems.
@inproceedings{MLSYS2023_3430e705, author = {Li, Qinbin and Wu, Zhaomin and Cai, Yanzheng and han, yuxuan and Yung, Ching Man and Fu, Tianyuan and He, Bingsheng}, booktitle = {Proceedings of Machine Learning and Systems}, editor = {Song, D. and Carbin, M. and Chen, T.}, pages = {89--103}, publisher = {Curan}, title = {FedTree: A Federated Learning System For Trees}, url = {https://proceedings.mlsys.org/paper_files/paper/2023/file/3430e7055936cb8e26451ed49fce84a6-Paper-mlsys2023.pdf}, volume = {5}, year = {2023}, google_scholar_id = {WF5omc3nYNoC} }
- TKDE 2023A Survey on Federated Learning Systems: Vision, Hype and Reality for Data Privacy and ProtectionQinbin Li, Zeyi Wen, Zhaomin Wu, Sixu Hu, Naibo Wang, and 3 more authorsIEEE Transactions on Knowledge & Data Engineering, 2023
As data privacy increasingly becomes a critical societal concern, federated learning has been a hot research topic in enabling the collaborative training of machine learning models among different organizations under the privacy restrictions. As researchers try to support more machine learning models with different privacy-preserving approaches, there is a requirement in developing systems and infrastructures to ease the development of various federated learning algorithms. Similar to deep learning systems such as PyTorch and TensorFlow that boost the development of deep learning, federated learning systems (FLSs) are equivalently important, and face challenges from various aspects such as effectiveness, efficiency, and privacy. In this survey, we conduct a comprehensive review on federated learning systems. To understand the key design system components and guide future research, we introduce the definition of federated learning systems and analyze the system components. Moreover, we provide a thorough categorization for federated learning systems according to six different aspects, including data distribution, machine learning model, privacy mechanism, communication architecture, scale of federation and motivation of federation. The categorization can help the design of federated learning systems as shown in our case studies. By systematically summarizing the existing federated learning systems, we present the design factors, case studies, and future research opportunities.
@article{9599369, author = {Li, Qinbin and Wen, Zeyi and Wu, Zhaomin and Hu, Sixu and Wang, Naibo and Li, Yuan and Liu, Xu and He, Bingsheng}, journal = {IEEE Transactions on Knowledge & Data Engineering}, title = {A Survey on Federated Learning Systems: Vision, Hype and Reality for Data Privacy and Protection}, year = {2023}, volume = {35}, number = {04}, issn = {1558-2191}, pages = {3347-3366}, keywords = {collaborative work;data models;machine learning;data privacy;computational modeling;deep learning;servers}, doi = {10.1109/TKDE.2021.3124599}, publisher = {IEEE Computer Society}, address = {Los Alamitos, CA, USA}, google_scholar_id = {u-x6o8ySG0sC} }
2022
- NeurIPS 2022A Coupled Design of Exploiting Record Similarity for Practical Vertical Federated LearningZhaomin Wu, Qinbin Li, and Bingsheng HeIn Advances in Neural Information Processing Systems , 2022
Federated learning is a learning paradigm to enable collaborative learning across different parties without revealing raw data. Notably, vertical federated learning (VFL), where parties share the same set of samples but only hold partial features, has a wide range of real-world applications. However, most existing studies in VFL disregard the" record linkage”process. They design algorithms either assuming the data from different parties can be exactly linked or simply linking each record with its most similar neighboring record. These approaches may fail to capture the key features from other less similar records. Moreover, such improper linkage cannot be corrected by training since existing approaches provide no feedback on linkage during training. In this paper, we design a novel coupled training paradigm, FedSim, that integrates one-to-many linkage into the training process. Besides enabling VFL in many real-world applications with fuzzy identifiers, FedSim also achieves better performance in traditional VFL tasks. Moreover, we theoretically analyze the additional privacy risk incurred by sharing similarities. Our experiments on eight datasets with various similarity metrics show that FedSim outperforms other state-of-the-art baselines. The codes of FedSim are available at https://github. com/Xtra-Computing/FedSim.
@inproceedings{NEURIPS2022_84b74416, author = {Wu, Zhaomin and Li, Qinbin and He, Bingsheng}, booktitle = {Advances in Neural Information Processing Systems}, editor = {Koyejo, S. and Mohamed, S. and Agarwal, A. and Belgrave, D. and Cho, K. and Oh, A.}, pages = {21087--21100}, publisher = {Curran Associates, Inc.}, title = {A Coupled Design of Exploiting Record Similarity for Practical Vertical Federated Learning}, url = {https://proceedings.neurips.cc/paper_files/paper/2022/file/84b744165a0597360caad96b06e69313-Paper-Conference.pdf}, volume = {35}, year = {2022}, google_scholar_id = {YsMSGLbcyi4C} }
- TBD 2022Practical Vertical Federated Learning with Unsupervised Representation LearningZhaomin Wu, Qinbin Li, and Bingsheng HeIEEE Transactions on Big Data, 2022
As societal concerns on data privacy recently increase, we have witnessed data silos among multiple parties in various applications. Federated learning emerges as a new learning paradigm that enables multiple parties to collaboratively train a machine learning model without sharing their raw data. Vertical federated learning, where each party owns different features of the same set of samples and only a single party has the label, is an important and challenging topic in federated learning. Communication costs among different parties have been a major hurdle for practical vertical learning systems. In this paper, we propose a novel communication-efficient vertical federated learning algorithm named FedOnce, which requires only one-shot communication among parties. To improve model accuracy and provide privacy guarantee, FedOnce features unsupervised learning representations in the federated setting and privacy-preserving techniques based on moments accountant. The comprehensive experiments on 10 datasets demonstrate that FedOnce achieves close performance compared to state-of-the-art vertical federated learning algorithms with much lower communication costs. Meanwhile, our privacy-preserving technique significantly outperforms the state-of-the-art approaches under the same privacy budget.
@article{9789268, author = {Wu, Zhaomin and Li, Qinbin and He, Bingsheng}, journal = {IEEE Transactions on Big Data}, title = {Practical Vertical Federated Learning with Unsupervised Representation Learning}, year = {2022}, volume = {}, number = {01}, issn = {2332-7790}, pages = {1-1}, keywords = {collaborative work;privacy;differential privacy;training;data privacy;costs;unsupervised learning}, doi = {10.1109/TBDATA.2022.3180117}, publisher = {IEEE Computer Society}, address = {Los Alamitos, CA, USA}, google_scholar_id = {IjCSPb-OGe4C} }
- TIST 2022The OARF Benchmark Suite: Characterization and Implications for Federated Learning SystemsSixu Hu, Yuan Li, Xu Liu, Qinbin Li, Zhaomin Wu, and 1 more authorACM Trans. Intell. Syst. Technol., 2022
This article presents and characterizes an Open Application Repository for Federated Learning (OARF), a benchmark suite for federated machine learning systems. Previously available benchmarks for federated learning (FL) have focused mainly on synthetic datasets and use a limited number of applications. OARF mimics more realistic application scenarios with publicly available datasets as different data silos in image, text, and structured data. Our characterization shows that the benchmark suite is diverse in data size, distribution, feature distribution, and learning task complexity. The extensive evaluations with reference implementations show the future research opportunities for important aspects of FL systems. We have developed reference implementations, and evaluated the important aspects of FL, including model accuracy, communication cost, throughput, and convergence time. Through these evaluations, we discovered some interesting findings such as FL can effectively increase end-to-end throughput. The code of OARF is publicly available on GitHub.1
@article{10.1145/3510540, author = {Hu, Sixu and Li, Yuan and Liu, Xu and Li, Qinbin and Wu, Zhaomin and He, Bingsheng}, title = {The OARF Benchmark Suite: Characterization and Implications for Federated Learning Systems}, year = {2022}, issue_date = {August 2022}, publisher = {Association for Computing Machinery}, address = {New York, NY, USA}, volume = {13}, number = {4}, issn = {2157-6904}, url = {https://doi.org/10.1145/3510540}, doi = {10.1145/3510540}, journal = {ACM Trans. Intell. Syst. Technol.}, articleno = {63}, numpages = {32}, keywords = {framework, dataset, benchmark, machine learning, Federated learning}, google_scholar_id = {9yKSN-GCB0IC} }
2020
- AAAI 2020Privacy-Preserving Gradient Boosting Decision TreesQinbin Li, Zhaomin Wu, Zeyi Wen, and Bingsheng HeIn The Thirty-Fourth AAAI Conference on Artificial Intelligence , 2020
The Gradient Boosting Decision Tree (GBDT) is a popular machine learning model for various tasks in recent years. In this paper, we study how to improve model accuracy of GBDT while preserving the strong guarantee of differential privacy. Sensitivity and privacy budget are two key design aspects for the effectiveness of differential private models. Existing solutions for GBDT with differential privacy suffer from the significant accuracy loss due to too loose sensitivity bounds and ineffective privacy budget allocations (especially across different trees in the GBDT model). Loose sensitivity bounds lead to more noise to obtain a fixed privacy level. Ineffective privacy budget allocations worsen the accuracy loss especially when the number of trees is large. Therefore, we propose a new GBDT training algorithm that achieves tighter sensitivity bounds and more effective noise allocations. Specifically, by investigating the property of gradient and the contribution of each tree in GBDTs, we propose to adaptively control the gradients of training data for each iteration and leaf node clipping in order to tighten the sensitivity bounds. Furthermore, we design a novel boosting framework to allocate the privacy budget between trees so that the accuracy loss can be further reduced. Our experiments show that our approach can achieve much better model accuracy than other baselines.
@inproceedings{DBLP:conf/aaai/LiWWH20, author = {Li, Qinbin and Wu, Zhaomin and Wen, Zeyi and He, Bingsheng}, title = {Privacy-Preserving Gradient Boosting Decision Trees}, booktitle = {The Thirty-Fourth {AAAI} Conference on Artificial Intelligence}, pages = {784--791}, publisher = {{AAAI} Press}, year = {2020}, url = {https://doi.org/10.1609/aaai.v34i01.5422}, doi = {10.1609/AAAI.V34I01.5422}, timestamp = {Sat, 30 Sep 2023 09:33:11 +0200}, biburl = {https://dblp.org/rec/conf/aaai/LiWWH20.bib}, bibsource = {dblp computer science bibliography, https://dblp.org}, google_scholar_id = {u5HHmVD_uO8C} }