Skip to content

Layer-Parallel Inference Reduces Encrypted Nonlinear Depth in Transformers

Jul 2026 · arXiv.org · Vol abs/2607.04819 · 0 citations · 33 references
Computer Science

TL;DR

Ablations show that softmax approximation dominates the error budget and CKKS arithmetic noise is negligible in the authors' setting, suggesting that SNLP is complementary to block-level FHE-friendly operator design rather than a replacement for it.

Abstract

Fully homomorphic encryption (FHE) enables computation on encrypted data, but practical encrypted Transformer inference is bottlenecked by the sequential composition of many nonlinear blocks. We study whether Structured Newton Layer Parallelism (SNLP) can make this inter-layer composition more FHE-friendly: each Transformer block still requires polynomial approximations for operations such as softmax and RMSNorm, but SNLP reduces the layerwise sequential nonlinear depth from L stages to a small number of solver iterations plus linear structured corrections. Using a simulation framework based on Chebyshev polynomial approximations, we measure error accumulation under sequential versus SNLP inference across 8 models and 4 architecture families. On a 0.5B IDN-trained model, SNLP reduces symbolic bootstraps from 53 to 20 (2.65x) with only +1.2% perplexity degradation, while lowering error amplification (1.36x vs. 1.42x). Across all tested models, SNLP has lower amplification than sequential inference. Ablations show that softmax approximation dominates the error budget and CKKS arithmetic noise is negligible in our setting, suggesting that SNLP is complementary to block-level FHE-friendly operator design rather than a replacement for it.

View source

Similar papers

Preprint Sep 2026

Memory-Efficient Designs for Word-Wise Universal Fully Homomorphic Encryption

Fully Homomorphic Encryption (FHE) enables computation on encrypted data, preserving privacy throughout analysis. While its privacy is very strong, FHE is much slower to execute than the original computation. In particular, due to the recent success in accelerating its compute, the performance bottleneck shifts to the...

A. W. B. Yudha, Erwin Eko Wahyudi, R. Rajagede et al. · 0 citations
Jul 2026

PRISM: Sensitivity-Aware PolynoMial PRuning for EffIcient Neural Network Encryption

Structured pruning is essential for making neural network inference feasible under homomorphic encryption (HE), yet its impact on model reliability has remained unexplored. This paper presents a systematic reliability characterization of pruned CKKS-encrypted neural networks and introduces Polynomial-Sensitivity-Aware...

Sahaj Majavdia, Mahdi Taheri · 0 citations
Open access Aug 2026

EHEIR: Efficient Homomorphic Encrypted Inference via Architectural Redesign

This work presents a framework that reformulates HE-aware model design as a constrained neural architecture search problem, where the objective is to identify architectures that are both cryptographically feasible and computationally efficient while preserving task performance.

Reeshav Chowdhury, Anoop Mishra, Deepak Khazanchi et al. · 0 citations
Jul 2026

ATLAS: Automated Approximation of Transformers for Efficient Homomorphic Inference in One Hour

ATLAS is a training-free framework that automates this search by treating each layer's approximation setting as a multi-objective optimization over latency and accuracy, and works across encoder-only, decoder-only, and vision Transformers, complementing parallel work on packing and matrix multiplication.

Jianhang Xie, Sicheng Tan, V. Boddeti et al. · 1 citation
Preprint Aug 2026

FESC: Remodeling Long-Context Private Inference with Encrypted State-Space Models

Factorized Encrypted Scan-Contract (FESC), a hybrid FHE-MPC system for private long-context selective SSM inference, is presented, and is the first private long-document inference system to complete native end-to-end execution at $L \geq 1{,}024$ on a single GPU.

Yu-Fan Zhu, Chao Jin, Khin Mi Mi Aung et al. · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.