PUMA: Secure Inference of LLaMA-7B in Five Minutes
Not on the map. Why?
Abstract (arXiv)
With ChatGPT as a representative, tons of companies have began to provide services based on large Transformers models. However, using such a service inevitably leak users' prompts to the model provider. Previous studies have studied secure inference for Transformer models using secure multiparty computation (MPC), where model parameters and clients' prompts are kept secret. Despite this, these frameworks are still limited in terms of model performance, efficiency, and deployment. To address these limitations, we propose framework PUMA to enable fast and secure Transformer model inference. Our framework designs high quality approximations for expensive functions such as GeLU and softmax, and significantly reduce the cost of secure inference while preserving the model performance. Additionally, we design secure Embedding and LayerNorm procedures that faithfully implement the desired functionality without undermining the Transformer architecture. PUMA is about 2× faster than the state-of-the-art framework MPCFORMER(ICLR 2023) and has similar accuracy as plaintext models without fine-tuning (which the previous works failed to achieve). PUMA can even evaluate LLaMA-7B in around 5 minutes to generate 1 token. To our best knowledge, this is the first time that a model with such a parameter size is able to be evaluated under MPC. PUMA has been open-sourced in the Github repository of SecretFlow-SPU.
Key facts. Cited by 48 papers in the map (data through 14 Oct 2026). First posted on arXiv 24 Jul 2023.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2023: 6; 2024: 14; 2025: 16; 2026: 12.
48 papers.
Sort:
By area:
- 2023Privacy in Large Language Models: Attacks, Defenses and Future Directionscited by 51
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Frameworkcited by 24
- 2024Deconstructing The Ethics of Large Language Models from Long-standing Issues to New-emerging Dilemmas: A Surveycited by 23
- 2024Encryption-Friendly LLM Architecturecited by 17
- 2024Nimbus: Secure and Efficient Two-Party Inference for Transformerscited by 16
- 2024No Free Lunch Theorem for Privacy-Preserving LLM Inferencecited by 13
- 2025CipherPrune: Efficient and Scalable Private Transformer Inferencecited by 10
- 2023East: Efficient and Accurate Secure Transformer Framework for Inferencecited by 9
- 2024SecFormer: Fast and Accurate Privacy-Preserving Inference for Transformer Models via SMPCcited by 9
- 2024Ditto: Quantization-aware Secure Inference of Transformers upon MPCcited by 8
- 2024Power-Softmax: Towards Secure LLM Inference over Encrypted Datacited by 7
- 2023Lambda-Split: A Privacy-Preserving Split Computing Framework for Cloud-Powered Generative AIcited by 6
36 more
- 2024MPC-Minimized Secure LLM Inferencecited by 6
- 2024CENTAUR: Bridging the Impossible Trinity of Privacy, Efficiency, and Performance in Privacy-Preserving Transformer Inferencecited by 6
- 2025PPMI: Privacy-Preserving LLM Interaction with Socratic Chain-of-Thought Reasoning and Homomorphically Encrypted Vector Databasescited by 6
- 2023Input Reconstruction Attack against Vertical Federated Large Language Modelscited by 5
- 2025Cape: Context-Aware Prompt Perturbation Mechanism with Differential Privacycited by 5
- 2025Comet: Accelerating Private Inference for Large Language Model by Predicting Activation Sparsitycited by 4
- 2024CipherDM: Secure Three-Party Inference for Diffusion Model Samplingcited by 3
- 2024TruncFormer: Private LLM Inference Using Only Truncationscited by 2
- 2025Private Transformer Inference in MLaaS: A Surveycited by 2
- 2025An Efficient Private GPT Never Autoregressively Decodescited by 2
- 2025An Attack to Break Permutation-Based Private Third-Party Inference Schemes for LLMscited by 2
- 2025ENSI: Efficient Non-Interactive Secure Inference for Large Language Modelscited by 2
- 2026SecMoE: Communication-Efficient Secure MoE Inference via Select-Then-Computecited by 2
- 2026On the (In-)Security of the Shuffling Defense in the Transformer Secure Inferencecited by 2
- 2026ATLAS: Automated Approximation of Transformers for Efficient Homomorphic Inference in One Hourcited by 2
- 2023SelectFormer: Private and Practical Data Selection for Transformerscited by 1
- 2024A Framework for Cost-Effective and Self-Adaptive LLM Shaking and Recovery Mechanismcited by 1
- 2025SecP-Tuning: Efficient Privacy-Preserving Prompt Tuning for Large Language Models via MPCcited by 1
- 2025Towards Provable (In)Secure Model Weight Release Schemescited by 1
- 2025Cascade: Token-Sharded Private LLM Inferencecited by 1
- 2025CrypTorch: PyTorch-based Auto-tuning Compiler for Machine Learning with Multi-party Computationcited by 1
- 2026OSNIP: Breaking the Privacy-Utility-Efficiency Trilemma in LLM Inference via Obfuscated Semantic Null Spacecited by 1
- 2026An Efficient and Privacy-Preserving Architecture for Cross-Institutional Collaborative RAGcited by 1
- 2026Privacy Failure in Split-LLM Training, The Returned Gradient Nullifies the Decoyscited by 1
- 2024Spin: An Efficient Secure Computation Framework with GPU Accelerationcited by 0
- 2024TensorTEE: Unifying Heterogeneous TEE Granularity for Efficient Secure Collaborative Tensor Computingcited by 0
- 2025Privacy-Preserving Inference for Quantized BERT Modelscited by 0
- 2025PRIVMARK: Private Large Language Models Watermarking with MPCcited by 0
- 2025ObCLIP: Oblivious CLoud-Device Hybrid Image Generation with Privacy Preservationcited by 0
- 2025PrivLLMSwarm: Privacy-Preserving LLM-Driven UAV Swarms for Secure IoT Surveillancecited by 0
- 2026Privacy-Preserving Mechanisms Enable Cheap Verifiable Inference of LLMscited by 0
- 2026HEAT: Faster Fully Homomorphic Inference via Approximations-Weights Co-Adaptationcited by 0
- 2026CipherGenome: Homomorphic Inference for Genomic Mixture-of-Expertscited by 0
- 2026Static Bootstrap Placement for Encrypted Language Model Decodingcited by 0
- 2026HE-OFT: Privacy-Preserving One-Shot Federated Fine-Tuning under Homomorphic Encryptioncited by 0
- 2026A Graph-Based Analysis of Comparability, Scale, and Attribution in Private Transformer Inferencecited by 0
- 2026A Graph-Based Analysis of Comparability, Scale, and Attribution in Private Transformer Inference14 Oct 2026
- 2026HE-OFT: Privacy-Preserving One-Shot Federated Fine-Tuning under Homomorphic Encryption6 Oct 2026
- 2026Static Bootstrap Placement for Encrypted Language Model Decoding4 Oct 2026
- 2026CipherGenome: Homomorphic Inference for Genomic Mixture-of-Experts27 Sep 2026
- 2026Privacy Failure in Split-LLM Training, The Returned Gradient Nullifies the Decoys3 Sep 2026
- 2026HEAT: Faster Fully Homomorphic Inference via Approximations-Weights Co-Adaptation1 Sep 2026
- 2026ATLAS: Automated Approximation of Transformers for Efficient Homomorphic Inference in One Hour26 Jul 2026
- 2026An Efficient and Privacy-Preserving Architecture for Cross-Institutional Collaborative RAG25 May 2026
- 2026On the (In-)Security of the Shuffling Defense in the Transformer Secure Inference6 May 2026
- 2026Privacy-Preserving Mechanisms Enable Cheap Verifiable Inference of LLMs19 Feb 2026
- 2026OSNIP: Breaking the Privacy-Utility-Efficiency Trilemma in LLM Inference via Obfuscated Semantic Null Space30 Jan 2026
- 2026SecMoE: Communication-Efficient Secure MoE Inference via Select-Then-Compute11 Jan 2026
36 more
- 2025PrivLLMSwarm: Privacy-Preserving LLM-Driven UAV Swarms for Secure IoT Surveillance7 Dec 2025
- 2025CrypTorch: PyTorch-based Auto-tuning Compiler for Machine Learning with Multi-party Computation24 Nov 2025
- 2025ObCLIP: Oblivious CLoud-Device Hybrid Image Generation with Privacy Preservation5 Oct 2025
- 2025PRIVMARK: Private Large Language Models Watermarking with MPC29 Sep 2025
- 2025ENSI: Efficient Non-Interactive Secure Inference for Large Language Models11 Sep 2025
- 2025Privacy-Preserving Inference for Quantized BERT Models3 Aug 2025
- 2025Cascade: Token-Sharded Private LLM Inference7 Jul 2025
- 2025Towards Provable (In)Secure Model Weight Release Schemes23 Jun 2025
- 2025PPMI: Privacy-Preserving LLM Interaction with Socratic Chain-of-Thought Reasoning and Homomorphically Encrypted Vector Databases19 Jun 2025
- 2025SecP-Tuning: Efficient Privacy-Preserving Prompt Tuning for Large Language Models via MPC18 Jun 2025
- 2025An Attack to Break Permutation-Based Private Third-Party Inference Schemes for LLMs23 May 2025
- 2025An Efficient Private GPT Never Autoregressively Decodes21 May 2025
- 2025Private Transformer Inference in MLaaS: A Survey15 May 2025
- 2025Comet: Accelerating Private Inference for Large Language Model by Predicting Activation Sparsity12 May 2025
- 2025Cape: Context-Aware Prompt Perturbation Mechanism with Differential Privacy9 May 2025
- 2025CipherPrune: Efficient and Scalable Private Transformer Inference24 Feb 2025
- 2024CENTAUR: Bridging the Impossible Trinity of Privacy, Efficiency, and Performance in Privacy-Preserving Transformer Inference14 Dec 2024
- 2024TruncFormer: Private LLM Inference Using Only Truncations2 Dec 2024
- 2024Nimbus: Secure and Efficient Two-Party Inference for Transformers24 Nov 2024
- 2024Power-Softmax: Towards Secure LLM Inference over Encrypted Data12 Oct 2024
- 2024Encryption-Friendly LLM Architecture3 Oct 2024
- 2024CipherDM: Secure Three-Party Inference for Diffusion Model Sampling9 Sep 2024
- 2024MPC-Minimized Secure LLM Inference7 Aug 2024
- 2024TensorTEE: Unifying Heterogeneous TEE Granularity for Efficient Secure Collaborative Tensor Computing12 Jul 2024
- 2024Deconstructing The Ethics of Large Language Models from Long-standing Issues to New-emerging Dilemmas: A Survey8 Jun 2024
- 2024No Free Lunch Theorem for Privacy-Preserving LLM Inference31 May 2024
- 2024Ditto: Quantization-aware Secure Inference of Transformers upon MPC9 May 2024
- 2024A Framework for Cost-Effective and Self-Adaptive LLM Shaking and Recovery Mechanism12 Mar 2024
- 2024Spin: An Efficient Secure Computation Framework with GPU Acceleration4 Feb 2024
- 2024SecFormer: Fast and Accurate Privacy-Preserving Inference for Transformer Models via SMPC1 Jan 2024
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Framework29 Nov 2023
- 2023Input Reconstruction Attack against Vertical Federated Large Language Models7 Nov 2023
- 2023Lambda-Split: A Privacy-Preserving Split Computing Framework for Cloud-Powered Generative AI23 Oct 2023
- 2023Privacy in Large Language Models: Attacks, Defenses and Future Directions16 Oct 2023
- 2023SelectFormer: Private and Practical Data Selection for Transformers3 Oct 2023
- 2023East: Efficient and Accurate Secure Transformer Framework for Inference19 Aug 2023
- 2023Privacy in Large Language Models: Attacks, Defenses and Future Directionscited by 51
- 2023Grounding Foundation Models through Federated Transfer Learning: A General Frameworkcited by 24
- 2024Deconstructing The Ethics of Large Language Models from Long-standing Issues to New-emerging Dilemmas: A Surveycited by 23
- 2024Encryption-Friendly LLM Architecturecited by 17
- 2024Nimbus: Secure and Efficient Two-Party Inference for Transformerscited by 16
- 2024No Free Lunch Theorem for Privacy-Preserving LLM Inferencecited by 13
- 2025CipherPrune: Efficient and Scalable Private Transformer Inferencecited by 10
- 2023East: Efficient and Accurate Secure Transformer Framework for Inferencecited by 9
- 2024SecFormer: Fast and Accurate Privacy-Preserving Inference for Transformer Models via SMPCcited by 9
- 2024Ditto: Quantization-aware Secure Inference of Transformers upon MPCcited by 8
- 2024Power-Softmax: Towards Secure LLM Inference over Encrypted Datacited by 7
- 2024MPC-Minimized Secure LLM Inferencecited by 6
12 most cited of 39 All 39 (JSON)
- 2026A Graph-Based Analysis of Comparability, Scale, and Attribution in Private Transformer Inference14 Oct 2026
- 2026HE-OFT: Privacy-Preserving One-Shot Federated Fine-Tuning under Homomorphic Encryption6 Oct 2026
- 2026Privacy Failure in Split-LLM Training, The Returned Gradient Nullifies the Decoys3 Sep 2026
- 2026HEAT: Faster Fully Homomorphic Inference via Approximations-Weights Co-Adaptation1 Sep 2026
- 2026ATLAS: Automated Approximation of Transformers for Efficient Homomorphic Inference in One Hour26 Jul 2026
- 2026On the (In-)Security of the Shuffling Defense in the Transformer Secure Inference6 May 2026
- 2026Privacy-Preserving Mechanisms Enable Cheap Verifiable Inference of LLMs19 Feb 2026
- 2026OSNIP: Breaking the Privacy-Utility-Efficiency Trilemma in LLM Inference via Obfuscated Semantic Null Space30 Jan 2026
- 2025PrivLLMSwarm: Privacy-Preserving LLM-Driven UAV Swarms for Secure IoT Surveillance7 Dec 2025
- 2025CrypTorch: PyTorch-based Auto-tuning Compiler for Machine Learning with Multi-party Computation24 Nov 2025
- 2025ENSI: Efficient Non-Interactive Secure Inference for Large Language Models11 Sep 2025
- 2025Privacy-Preserving Inference for Quantized BERT Models3 Aug 2025
12 newest of 39 All 39 (JSON)