Public-Decay Homomorphic State Space Models for Private Sequence Inference

TL;DR

This paper introduces public-decay homomorphic state space models (HSSMs) that enable efficient, accurate encrypted sequence inference, outperforming polynomial attention in speed while maintaining high accuracy.

Contribution

The paper presents a novel FHE-compatible state space model with public-decay updates, improving inference speed and accuracy over previous polynomial attention methods.

Findings

HSSMs match plaintext classification accuracy on Rotten Tomatoes and SST-2 datasets.

HSSMs run about 5x faster than HE-friendly polynomial attention.

HSSMs demonstrate lower latency and encrypted state footprint compared to polynomial attention.

Abstract

Fully homomorphic encryption (FHE) changes sequence-model design because rotations, encrypted products, ciphertext materialization, multiplicative depth, and bootstrapping pressure can dominate ordinary neural-network costs. This paper presents public-decay homomorphic state space models (HSSMs), recurrent/state-space blocks whose carried state is updated through ciphertext-plaintext public decay while ciphertext-ciphertext multiplication remains on a local write path. The design keeps a fixed encrypted state across the sequence. The evaluated workflow separates client-side tokenization, frozen fastText lookup, projection, clipping, encryption, decryption, and thresholding from server-side encrypted evaluation over bounded projected features. On full Rotten Tomatoes and SST-2 validation splits, the encrypted HSSM path exactly matches plaintext classifications and reaches 0.7505 and 0.7420 accuracy. Against HE-friendly polynomial attention on the same fastText workloads, HSSM matches or exceeds full-sequence task quality while running about 5x faster. Paired L40S operation-level rows show 1.34-1.62x lower latency than cached final-token polynomial attention, 30-258x lower latency than full-sequence polynomial attention, and lower logical encrypted-state footprint. A T = 16/32 comparator with encrypted public-linear input and Q/K/V projections shows projected HSSM succeeding under depth 8/ring 32768, while projected attention succeeds under depth 10/ring 65536. A matched T = 8 OpenFHE/FIDESlib trace finishes at final level 3 and noise-scale degree 2 on both backends. These results make public-decay carry a practical FHE co-design lever for encrypted sequence inference from bounded projected features.