AIセキュリティポータル K Program
PackVFL: Efficient HE Packing for Vertical Federated Learning
Share
Abstract
As an essential tool of secure distributed machine learning, vertical federated learning (VFL) based on homomorphic encryption (HE) suffers from severe efficiency problems due to data inflation and time-consuming operations. To this core, we propose PackVFL, an efficient VFL framework based on packed HE (PackedHE), to accelerate the existing HE-based VFL algorithms. PackVFL packs multiple cleartexts into one ciphertext and supports single-instruction-multiple-data (SIMD)-style parallelism. We focus on designing a high-performant matrix multiplication (MatMult) method since it takes up most of the ciphertext computation time in HE-based VFL. Besides, devising the MatMult method is also challenging for PackedHE because a slight difference in the packing way could predominantly affect its computation and communication costs. Without domain-specific design, directly applying SOTA MatMult methods is hard to achieve optimal. Therefore, we make a three-fold design: 1) we systematically explore the current design space of MatMult and quantify the complexity of existing approaches to provide guidance; 2) we propose a hybrid MatMult method according to the unique characteristics of VFL; 3) we adaptively apply our hybrid method in representative VFL algorithms, leveraging distinctive algorithmic properties to further improve efficiency. As the batch size, feature dimension and model size of VFL scale up to large sizes, PackVFL consistently delivers enhanced performance. Empirically, PackVFL propels existing VFL algorithms to new heights, achieving up to a 51.52X end-to-end speedup. This represents a substantial 34.51X greater speedup compared to the direct application of SOTA MatMult methods.
State-of-the-art in artificial neural network applications: A survey
Oludare Isaac Abiodun, Aman Jantan, Abiodun Esther Omolara, Kemi Victoria Dada, Nachaat AbdElatif Mohamed, Humaira Arshad
Published: 2018
A survey on homomorphic encryption schemes: Theory and implementation
Abbas Acar, Hidayet Aksu, A. Selcuk Uluagac, Mauro Conti
Published: 2018
Homomorphic encryption standard
Martin Albrecht, Melissa Chase, Hao Chen, Jintai Ding, Shafi Goldwasser, Sergey Gorbunov, Shai Halevi, Jeffrey Hoffstein, Kim Laine, Kristin Lauter
Published: 2021
Searching for exotic particles in high-energy physics with deep learning
Pierre Baldi, Peter Sadowski, Daniel Whiteson
Published: 2014
Secret-sharing schemes: A survey
Amos Beimel
Published: 2011
Comparing local and central differential privacy using membership inference attacks
Daniel Bernau, Jonas Robl, Philip W Grassal, Steffen Schneider, Florian Kerschbaum
Published: 2021
Packed ciphertexts in LWE-based homomorphic encryption
Zvika Brakerski, Craig Gentry, Shai Halevi
Published: 2013
(leveled) fully homomorphic encryption without bootstrapping
Z. Brakerski, C. Gentry, V. Vaikuntanathan
Published: 2014
Language models are few-shot learners
T. B. Brown, B. Mann, N. Ryder, M. Subbiah, J. Kaplan, P. Dhariwal, A. Neelakantan, P. Shyam, G. Sastry, A. Askell, S. Agarwal, A. Herbert-Voss, G. Krueger, T. Henighan, R. Child, A. Ramesh, D. M. Ziegler, J. Wu, C. Winter, C. Hesse, M. Chen, E. Sigler, M. Litwin, S. Gray, B. Chess, J. Clark, C. Berner, S. McCandlish, A. Radford, I. Sutskever, D. Amodei
Published: 2020
The bit security of Paillier’s encryption scheme and its applications
Dario Catalano, Rosario Gennaro, Nick Howgrave-Graham
Published: 2001
Efficient homomorphic conversion between (ring) LWE ciphertexts
Hao Chen, Wei Dai, Miran Kim, Yongsoo Song
Published: 2021
Share