arrow
Return

Accelerating Private Large Transformers Inference Through Fine-Grained Collaborative Computation

delete2025-01-01
delete0
PRE
AI
Y
Yuntian Chen
Z
Zhanyong Tang
T
Tianpei Lu
B
Bingsheng Zhang
Z
Zhiying Shi
Z
Zheng Wang
DOI:10.1109/TIFS.2025.3584639delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Homomorphic encryption (HE) and secret sharing (SS) enable computations on encrypted data, providing significant privacy benefits for large transformer-based models (TBM) in sensitive sectors like medicine and finance. However, private TBM inference incurs significant costs due to the coarse-grained application of HE and SS. We present FASTLMPI, a new approach to accelerate private TBM inference through fine-grained computation optimization. Specifically, through the fine-grained co-design of homomorphic encryption and secret sharing, FASTLMPI achieves efficient protocols for matrix multiplication, SoftMax, LayerNorm, and GeLU. In addition, FASTLMPI introduces a precise segmented approximation technique for differentiable non-linear functions, improving its fitting accuracy while maintaining a low polynomial degree. Compared to solution BOLT (S&P’24), FASTLMPI shows a remarkable 25.1% to 55.3% decrease in runtime and an impressive 39.0% reduction in communication costs.
Keywords:
Secure multiparty computation
homomorphic encryption
privacy preserving
large transformer models

Journal

IEEE Transactions on Information Forensics and Security cover
IEEE Transactions on Information Forensics and Security
IF:
8
Papers:
5.3K
Citations:
2.3W

Organization

N
Northwest University
Scholars:
3.3K
Papers: 971
Citations: 158
U
university of leeds
Scholars:
3.6W
Papers: 3.3W
Citations: 45
Z
zhejiang university
Scholars:
17.7W
Papers: 12.1W
Citations: 152
researcher View more organizations
Cited Papers

Cited Papers

Elliptic curve cryptosystem and its applications
err
IF0
err2024-09-02
err0
PREAI
errG.V.S. Raju; R. Akbani
errShare
errSave
Multi-party computation with conversion of secret sharing
err2011-05-10
err0
PREAI
errHossein Ghodosi; Josef Pieprzyk; Ron Steinfeld
errShare
errSave
CrypTFlow2: Practical 2-Party Secure Inference
err2020-11-02
err0
errOAAI
errDeevashwer Rathee; Mayank Rathee; Nishant Kumar; Nishanth Chandran; Divya Gupta; Aseem Rastogi; Rahul Sharma
errShare
errSave
THE-X: Privacy-Preserving Transformer Inference with Homomorphic Encryption
err2022-01-01
err0
errOAAI
errTianyu Chen; Hangbo Bao; Shaohan Huang; Li Dong; Binxing Jiao; Daxin Jiang; Haoyi Zhou; Jianxin Li; Furu Wei
errShare
errSave
Orca: FSS-based Secure Training and Inference with GPUs
err2024-05-19
err0
PREAI
errNeha Jawalkar; Kanav Gupta; Arkaprava Basu; Nishanth Chandran; Divya Gupta; Rahul Sharma
errShare
errSave
SortingHat
err2022-11-07
err0
PREAI
errKelong Cong; Debajyoti Das; Jeongeun Park; Hilder V.L. Pereira
errShare
errSave
errShare
errSave
errShare
errSave
researcher View more