vix.ing · top · new · best · stats · spec

CipherFormer: Efficient Transformer Private Inference with Low Round Complexity

2024/03/25 by Weize Wang, Wang, Weize, Yi Kuang +1
Computer Science · #Cryptography and Security (cs.CR) #FOS: Computer and information sciences #Quantum Computing Algorithms and Architecture #Quantum Information and Cryptography #Quantum-Dot Cellular Automata

paper · pdf · doi:10.48550/arxiv.2403.16860

openalex publication_date 2024/03/25 · openalex created_date 2024/03/27 · openalex updated_date 2026/07/28

Abstract

There is a growing trend to outsource the inference task of large transformer models to cloud servers. However, this poses a severe threat to users' private data as they are exposed to cloud servers after uploading. Although several works attempted to provide private inference for transformer models, their hundreds of communication rounds limit the application scenarios. Motivated by the desire to minimize round complexity, we propose CipherFormer, a novel transformer private inference scheme using homomorphic encryption and garbled circuits. We present a protocol for quickly computing homomorphic matrix multiplications. We then modify the attention mechanism and design the corresponding garbled circuits. Furthermore, we show how to use a lightweight attention mechanism and mixed-bitwidth to reduce the inference latency while maintaining accuracy. In comparison with an advanced homomorphic encryption scheme on text classification tasks, our model improves accuracy by 3% to 11% while performing private inference with a 7.7x-11.9x speedup.

Related