arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.03425cs.CRcs.AIcs.DC
收藏

AEGIS:通过多GPU系统的混合并行性实现长序列同态加密Transformer推断的扩展

AEGIS: Scaling Long-Sequence Homomorphic Encrypted Transformer Inference via Hybrid Parallelism on Multi-GPU Systems

North Carolina State University(北卡罗来纳州立大学);University of Central Florida(中佛罗里达大学)

Zhaoting Gong, Ran Ran, Fan Yao, Wujie Wen

AI总结:

AEGIS通过结合Transformer数据流和CKKS多项式耦合,实现多GPU平台上的长序列加密Transformer推断,减少通信开销并提升效率。

摘要:

全同态加密(FHE)使隐私保护的Transformer推断成为可能,但长序列加密的Transformer因编码权重较大且加密激活值随序列长度增长而迅速增加,导致单GPU内存不足。多GPU执行不可避免,但扩展仍具挑战性,因为通信由应用层聚合和加密层RNS耦合共同诱导。现有方法要么频繁在设备间同步,要么在设备间复制加密张量,导致通信过多和延迟。我们提出AEGIS,一种应用-加密引导的推断系统,用于多GPU平台上的可扩展长序列加密Transformer推断。AEGIS从Transformer数据流和CKKS多项式耦合共同诱导的密文依赖中推导出设备放置,将模数一致和token一致的数据共置,使通信仅在应用依赖要求时引入,同时重新排列多项式运算以重叠剩余的收集与计算。在2048个token输入上,AEGIS在前馈网络中将GPU间通信减少高达57.9%,在自注意力中减少81.3%。在四块GPU上,其达到96.62%的扩展效率、3.86倍的端到端加速和69.1%的每设备内存减少。这些结果确立了协调的应用-加密并行性作为可扩展同态Transformer推断的实用基础。

英文摘要:

Fully Homomorphic Encryption (FHE) enables privacy-preserving Transformer inference, but long-sequence encrypted Transformers quickly exceed single-GPU memory capacity because encoded weights are already large and encrypted activations grow rapidly with sequence length. Multi-GPU execution therefore becomes unavoidable, yet scaling remains challenging because communication is jointly induced by application-level aggregation and encryption-level RNS coupling. Existing approaches either synchronize between devices frequently or replicate encrypted tensors across devices, leading to excessive communication and latency. We present AEGIS, an Application-Encryption Guided Inference System for scalable long-sequence encrypted Transformer inference on multi-GPU platforms. AEGIS derives device placement from ciphertext dependencies jointly induced by Transformer dataflow and CKKS polynomial coupling, co-locating modulus-coherent and token-coherent data so that communication is introduced only when application dependencies require it, while reordering polynomial operators to overlap the remaining collectives with computation. On 2048-token inputs, AEGIS reduces inter-GPU communication by up to 57.9% in feed-forward networks and 81.3% in self-attention versus prior state-of-the-art designs. On four GPUs, it achieves up to 96.62% scaling efficiency, 3.86x end-to-end speedup, and 69.1% per-device memory reduction. These results establish coordinated application-encryption parallelism as a practical foundation for scalable homomorphic Transformer inference.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑