一句话:提出 Orion 全自动 FHE 框架,将 PyTorch 深度网络编译为高效同态加密程序,实现隐私神经推理。
全同态加密(FHE)可对加密数据直接计算,为云端深度学习隐私保护提供可能,但将神经网络映射到 FHE 原语需解决向量打包、噪声累积、自举管理与有限指令集等挑战,构建大型 FHE 网络十分困难。本文提出 Orion,一个全自动 FHE 私有推理框架:它接受 PyTorch 深度网络并翻译为高效 FHE 程序,采用面向任意卷积的单次多路打包策略,并自动完成自举放置与尺度管理。在 FHE 深度学习基准上,Orion 在 ResNet-20 上较最优实现取得 2.38× 加速;它还能处理更深更大网络,在 ImageNet 上评估 ResNet-50,并首次用 1.39 亿参数的 YOLO-v1 完成高分辨率 FHE 目标检测。框架已开源。
Fully Homomorphic Encryption (FHE) has the potential to substantially improve privacy and security by enabling computation directly on encrypted data. This is especially true with deep learning, as today, many popular user services are powered by neural networks in the cloud. Beyond its well-known high computational costs, one of the major challenges facing wide-scale deployment of FHE-secured neural inference is effectively mapping these networks to FHE primitives. FHE poses many programming challenges including packing large vectors, managing accumulated noise, and translating arbitrary and general-purpose programs to the limited instruction set provided by FHE. These challenges make building large FHE neural networks intractable using the tools available today. In this paper we address these challenges with Orion, a fully-automated framework for private neural inference using FHE. Orion accepts deep neural networks written in PyTorch and translates them into efficient FHE programs. We achieve this by proposing a novel single-shot multiplexed packing strategy for arbitrary convolutions and through a new, efficient technique to automate bootstrap placement and scale management. We evaluate Orion on common benchmarks used by the FHE deep learning community and outperform state-of-the-art by 2.38 × on ResNet-20, the largest network they report. Orion's techniques enable processing much deeper and larger networks. We demonstrate this by evaluating ResNet-50 on ImageNet and present the first high-resolution FHE object detection experiments using a YOLO-v1 model with 139 million parameters. Orion is open-source for all to use at: \hrefhttps://github.com/baahl-nyu/orion https://github.com/baahl-nyu/orion.