中文

Falcon:加速同态加密卷积以实现高效私有移动网络推理

密码学与安全 2023-08-28 v1 人工智能

摘要

高效网络(如MobileNetV2、EfficientNet等)以轻量计算实现了最先进的(SOTA)精度。然而,现有基于同态加密(HE)的两方计算(2PC)框架未针对这些网络优化,且存在较高的推理开销。我们观察到低效主要源于打包算法,其忽略了同态加密深度可分离卷积的计算特征与通信瓶颈。因此,本文提出Falcon,一种用于基于HE的2PC框架的有效密集打包算法。Falcon具有零感知贪心打包算法与通信感知算子分块策略,以提升深度可分离卷积的打包密度。与SOTA基于HE的2PC框架(如CrypTFlow2、Iron和Cheetah)相比,Falcon在算子级别分别实现了超过15.6倍、5.1倍和1.8倍的延迟降低。同时,在网络级别,Falcon在同等通信量下于CIFAR-100和TinyImagenet数据集上分别比Cheetah精度提升1.4%和4.2%。

关键词

引用

@article{arxiv.2308.13189,
  title  = {Falcon: Accelerating Homomorphically Encrypted Convolutions for Efficient Private Mobile Network Inference},
  author = {Tianshi Xu and Meng Li and Runsheng Wang and Ru Huang},
  journal= {arXiv preprint arXiv:2308.13189},
  year   = {2023}
}

备注

8 pages. ICCAD 2023