paper-with-me

홈 › Papers

Secure Transformer Inference Protocol

2023-11-14 · Mu Yuan, Lan Zhang, Xiang-Yang Li

Security of model parameters and user data is critical for Transformer-based services, such as ChatGPT. While recent strides in secure two-party protocols have successfully addressed security concerns in serving Transformer models, their adoption is practically infeasible due to the prohibitive cryptographic overheads involved. Drawing insights from our hands-on experience in developing two real-world Transformer-based services, we identify the inherent efficiency bottleneck in the two-party assumption. To overcome this limitation, we propose a novel three-party threat model. Within this framework, we design a semi-symmetric permutation-based protection scheme and present STIP, the first secure Transformer inference protocol without any inference accuracy loss. Experiments on representative Transformer models in real systems show that STIP has practical security and outperforms state-of-the-art secure two-party protocols in efficiency by millions of times.

📄 PDF Abstract BibTeX arXiv:2312.00025

Code (1)

yuanmu97/secure-transformer-inference 공식 구현 pytorch

Methods 이 논문이 사용한 방법론

Attention 설명 없음
Absolute Position Encodings Absolute Position Encodings are a type of position embeddings for [Transformer-based models] where positional encodings are…
Dense Connections Dense Connections, or Fully Connected Connections, are a type of layer in a deep neural network that use a linear operation where every input is connected to every output…
Dropout Dropout is a regularization technique for neural networks that drops a unit (along with connections) at training time with a specified probability $p$ (a common value is…
BPE Byte Pair Encoding, or BPE, is a subword segmentation algorithm that encodes rare and unknown words as sequences of subword units. The intuition is that various word…
Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…
Layer Normalization Unlike batch normalization, Layer Normalization directly estimates the normalization statistics from the summed inputs…
Position-Wise Feed-Forward Layer 설명 없음

Similar Papers 제목 키워드 기반

East: Efficient and Accurate Secure Transformer Framework for Inference

2023-08-19 · Yuanchao Ding, Hua Guo, Yewei Guan, Weixin Liu 외

Transformer has been successfully used in practical applications, such as ChatGPT, due to its powerful advantages. However, users' input is leaked to the model provider during the service. With people's attention to priv…

Privacy Preserving

SecureRouter: Encrypted Routing for Efficient Secure Inference

2026-04-16 · Yukuan Zhang, Mengxin Zheng, Qian Lou arxiv

Cryptographically secure neural network inference typically relies on secure computing techniques such as Secure Multi-Party Computation (MPC), enabling cloud servers to process client inputs without decrypting them. Alt…

CrypTFlow2: Practical 2-Party Secure Inference

2020-10-13 · Deevashwer Rathee, Mayank Rathee, Nishant Kumar, Nishanth Chandran 외

We present CrypTFlow2, a cryptographic framework for secure inference over realistic Deep Neural Networks (DNNs) using secure 2-party computation. CrypTFlow2 protocols are both correct -- i.e., their outputs are bitwise …

CrypTFlow: Secure TensorFlow Inference

2019-09-16 · Nishant Kumar, Mayank Rathee, Nishanth Chandran, Divya Gupta 외

We present CrypTFlow, a first of its kind system that converts TensorFlow inference code into Secure Multi-party Computation (MPC) protocols at the push of a button. To do this, we build three components. Our first compo…

Nimbus: Secure and Efficient Two-Party Inference for Transformers

2024-11-24 · Zhengyi Li, Kang Yang, Jin Tan, Wen-jie Lu 외

Transformer models have gained significant attention due to their power in machine learning tasks. Their extensive deployment has raised concerns about the potential leakage of sensitive information during inference. How…