CipherFormer: Efficient Transformer Private Inference with Low Round Complexity

TL;DR

CipherFormer is proposed, a novel transformer private inference scheme using homomorphic encryption and garbled circuits, and a protocol for quickly computing homomorphic matrix multiplications for quickly computing homomorphic matrix multiplications.

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.

Figures (3)

• Figure 1: Private model inference between clients and servers.
• Figure 2: A transformer-based text classification model divided into HE and GC parts by operations.
• Figure 3: The accuracy in different activation function bitwidth strategy over two datasets AG_NEWS and IMDB.