Abstract:Privacy protection has attracted increasing attention, and privacy concerns often prevent flexible data utilization. In most industries, data are distributed across multiple organizations due to privacy concerns. Federated learning (FL), which enables cross-organizational machine learning by communicating statistical information, is a state-of-the-art technology that is used to solve this problem. However, for gradient boosting decision tree (GBDT) in FL, balancing communication efficiency and security while maintaining sufficient accuracy remains an unresolved problem. In this paper, we propose an FL scheme for GBDT, i.e., efficient FL for GBDT (eFL-Boost), which minimizes accuracy loss, communication costs, and information leakage. The proposed scheme focuses on appropriate allocation of local computation (performed individually by each organization) and global computation (performed cooperatively by all organizations) when updating a model. It is known that tree structures incur high communication costs for global computation, whereas leaf weights do not require such costs and are expected to contribute relatively more to accuracy. Thus, in the proposed eFL-Boost, a tree structure is determined locally at one of the organizations, and leaf weights are calculated globally by aggregating the local gradients of all organizations. Specifically, eFL-Boost requires only three communications per update, and only statistical information that has low privacy risk is leaked to other organizations. Through performance evaluation on public data sets (ROC AUC, Log loss, and F1-score are used as metrics), the proposed eFL-Boost outperforms existing schemes that incur low communication costs and was comparable to a scheme that offers no privacy protection.

SilentWood: Private Inference Over Gradient-Boosting Decision Forests

SoK: Modular and Efficient Private Decision Tree Evaluation.

A Hybrid-Domain Framework for Secure Gradient Tree Boosting.

Securely and Efficiently Outsourcing Decision Tree Inference

Differentially Private Greedy Decision Forest

FDPBoost: Federated differential privacy gradient boosting decision trees

Federated Boosted Decision Trees with Differential Privacy

OnePath: Efficient and Privacy-Preserving Decision Tree Inference in the Cloud

Efficient and Privacy-Preserving Tree-Based Inference via Additive Homomorphic Encryption

eFL-Boost: Efficient Federated Learning for Gradient Boosting Decision Trees

Efficient Encrypted Inference on Ensembles of Decision Trees

Poster: gbdt-rs: Fast and Trustworthy Gradient Boosting Decision Tree

Privet: A Privacy-Preserving Vertical Federated Learning Service for Gradient Boosted Decision Tables

Level Up: Private Non-Interactive Decision Tree Evaluation using Levelled Homomorphic Encryption

PriVDT: An Efficient Two-Party Cryptographic Framework for Vertical Decision Trees

Secure Collaborative Training and Inference for XGBoost

A(DP)$^2$SGD: Asynchronous Decentralized Parallel Stochastic Gradient Descent with Differential Privacy

SecureBoost+: Large Scale and High-Performance Vertical Federated Gradient Boosting Decision Tree

Vectorized Secure Evaluation of Decision Forests

FederBoost: Private Federated Learning for GBDT

GTree: GPU-Friendly Privacy-preserving Decision Tree Training and Inference