Publication Type
Conference Proceeding Article
Version
acceptedVersion
Publication Date
12-2024
Abstract
Outsourcing Decision tree (DT) training and inference to cloud platforms raises privacy concerns. Recent Secure Multi-Party Computation (MPC)-based methods are hindered by heavy overhead. Few recent studies explored GPUs to improve MPC-protected deep learning, yet integrating GPUs into MPC-protected DT with massive data-dependent operations remains challenging, raising question: can MPC-protected DT training and inference fully leverage GPUs for optimal performance?We present GTree, the first scheme that exploits GPU to accelerate MPC-protected secure DT training and inference. GTree is built across 3 parties who jointly perform DT training and inference with GPUs. GTree is secure against semi-honest adversaries, ensuring that no sensitive information is disclosed. GTree offers enhanced security than prior solutions, which only reveal tree depth and data size while prior solutions also leak tree structure. With our oblivious array access, access patterns on GPU are also protected. To harness the full potential of GPUs, we design a novel tree encoding method and craft our MPC protocols into GPU-friendly versions. GTree achieves ~11× and ~21× improvements in training SPECT and Adult datasets, compared to prior most efficient CPU-based work. For inference, GTree outperforms the prior most efficient work by 126× when inferring 104 instances with a 7-level tree.
Keywords
decision trees, GPU, privacy-preserving machine learning, secure computation
Discipline
Information Security
Publication
2024 IEEE 23rd International Conference on Trust, Security and Privacy in Computing and Communications (TrustCom): Sanya, China, December 17-21: Proceedings
Issue
2024
First Page
775
Last Page
785
ISBN
9798331506209
Identifier
10.1109/TrustCom63139.2024.00118
City or Country
Piscataway, NJ
Citation
WANG, Qifan; CUI, Shujie; ZHOU, Lei; DONG, Ye; BAI, Jianli; KOH, Yun Sing; and RUSSELLO, Giovanni.
GTree: GPU-friendly privacy-preserving decision tree training and inference. (2024). 2024 IEEE 23rd International Conference on Trust, Security and Privacy in Computing and Communications (TrustCom): Sanya, China, December 17-21: Proceedings. 775-785.
Available at: https://ink.library.smu.edu.sg/sis_research/10238
Creative Commons License

This work is licensed under a Creative Commons Attribution-NonCommercial-No Derivative Works 4.0 International License.
Additional URL
https://doi.org/10.1109/TrustCom63139.2024.00118