Alert button

Gradient-less Federated Gradient Boosting Trees with Learnable Learning Rates

Apr 15, 2023
Chenyang Ma, Xinchi Qiu, Daniel J. Beutel, Nicholas D. Lane

Figure 1 for Gradient-less Federated Gradient Boosting Trees with Learnable Learning Rates
Figure 2 for Gradient-less Federated Gradient Boosting Trees with Learnable Learning Rates
Figure 3 for Gradient-less Federated Gradient Boosting Trees with Learnable Learning Rates
Figure 4 for Gradient-less Federated Gradient Boosting Trees with Learnable Learning Rates

Share this with someone who'll enjoy it:

The privacy-sensitive nature of decentralized datasets and the robustness of eXtreme Gradient Boosting (XGBoost) on tabular data raise the needs to train XGBoost in the context of federated learning (FL). Existing works on federated XGBoost in the horizontal setting rely on the sharing of gradients, which induce per-node level communication frequency and serious privacy concerns. To alleviate these problems, we develop an innovative framework for horizontal federated XGBoost which does not depend on the sharing of gradients and simultaneously boosts privacy and communication efficiency by making the learning rates of the aggregated tree ensembles learnable. We conduct extensive evaluations on various classification and regression datasets, showing our approach achieves performance comparable to the state-of-the-art method and effectively improves communication efficiency by lowering both communication rounds and communication overhead by factors ranging from 25x to 700x.

* accepted at the 3rd ACM Workshop on Machine Learning and Systems (EuroMLSys), May 8th 2023, Rome, Italy  
View paper onarxiv icon

Share this with someone who'll enjoy it: