KeyMemRT 编译器和运行时：解锁内存可扩展的全同态加密
KeyMemRT Compiler and Runtime: Unlocking Memory-Scalable FHE
AI总结：
KeyMemRT通过自动密钥管理和数据流分析，降低FHE内存消耗并提升运行效率。
摘要：
全同态加密（FHE）能够实现隐私保护计算，但其存在高延迟和内存消耗的问题。计算过程通过称为旋转密钥的特殊密钥进行保护，这些密钥通常占据了大部分内存。在复杂的FHE应用中，这些旋转密钥可能导致较大的内存瓶颈，限制程序吞吐量。现有的编译器几乎没有努力解决这个问题，而是依赖于具有大量内存可用性的系统。这种资源需求是FHE普及的障碍，因为手动优化FHE程序具有挑战性，因为它们的规模、复杂性和所需的专业知识。在本工作中，我们提出了KeyMemRT；一个基于MLIR的编译器和运行时框架，能够单独管理旋转密钥的生命周期以降低内存利用率，并允许支持任意数量的旋转索引而不会导致内存膨胀。KeyMemRT依赖于数据流分析来确定密钥生命周期，并是首个提供自动密钥管理、处理细粒度密钥管理并管理启动密钥的FHE编译器。我们为Orion和HEIR实现了前端，并展示了比现有最先进FHE编译器的改进。KeyMemRT在ANT-ACE上实现了1.74倍的内存减少和1.20倍的速度提升，在内存优化编译器Fhelipe上实现了1.16倍的内存减少和1.73倍的速度提升。我们提供KeyMemRT作为后优化编译器，可以被任何FHE编译器所针对。
英文摘要：
Fully Homomorphic Encryption (FHE) enables privacy preserving computation but it suffers from high latency and memory consumption. The computations are secured with special keys called rotation keys which often take up the majority of memory. In complex FHE applications, these rotation keys can cause a large memory bottleneck limiting program throughput. Existing compilers make little effort to solve this problem, instead relying on systems with massive memory availability. This resource requirement is a barrier to FHE uptake because optimizing FHE programs by hand is challenging due to their scale, complexity and expertise required. In this work, we present KeyMemRT; an MLIR based compiler and runtime framework that individually manages rotation key lifetimes to lower memory utilization and to allow arbitrary number of rotation indices to be supported without memory bloating. KeyMemRT relies on dataflow analysis to determine key lifetimes and is the first FHE compiler to provide automatic key management, handle fine-grained key-mangement and manage boostrap keys. We implement frontends for Orion and HEIR and show improvements over state-of-the-art FHE compilers. KeyMemRT achieves memory reduction of 1.74x and a speedup of 1.20x over ANT-ACE, and memory reduction of 1.16x and a speedup of 1.73x over memory-optimized compiler Fhelipe. We provide KeyMemRT as a post-optimizing compiler that can be targeted by any FHE compiler.
机构、总结、中文摘要均由AI分析而来，可能存在错误，请以论文正文为准。
