分享自:

基于同态加密和SGX的隐私保护神经网络推理框架

期刊:2021 IEEE 41st International Conference on Distributed Computing Systems (ICDCS)DOI:10.1109/ICDCS51616.2021.00077

本文介绍了一项关于隐私保护的神经网络推理框架的研究,该框架结合了同态加密(Homomorphic Encryption, HE)和英特尔软件保护扩展(Intel Software Guard Extension, SGX)技术。该研究由Huizi Xiao、Qingyang Zhang、Qingqi Pei和Weisong Shi共同完成,分别来自西安电子科技大学、安徽大学和韦恩州立大学。该论文发表于2021年的IEEE第41届国际分布式计算系统会议(ICDCS)。

研究背景与动机

随着智能设备的普及,边缘计算(Edge Computing)作为一种新兴的计算范式,能够将计算、存储和能源推向网络的边缘,从而提供低延迟、高带宽和高可靠性的服务。神经网络推理在边缘计算中具有广泛的应用,例如图像处理、人脸识别、医疗诊断和行为分析等。然而,边缘服务器不可避免地会收集用户的敏感信息,如何在提供准确推理服务的同时保护用户隐私成为一个重要问题。

同态加密技术允许在加密数据上直接进行数学计算,从而保护数据隐私。然而,HE的计算效率较低,且仅支持有限的加法和乘法操作。SGX则提供了一种可信的执行环境,能够在不可信的环境中保护代码和数据的完整性和机密性。然而,SGX在硬件设计上存在一些限制,例如内存有限、容易受到侧信道攻击等。因此,本文提出了一种结合HE和SGX的混合框架,旨在提高HE推理的准确性和效率,同时避免引入额外的可信第三方。

研究方法与流程

本文提出的混合框架主要包括以下几个步骤:

  1. 密钥分发:通过SGX的远程认证功能,生成同态加密的公钥和私钥,并将密钥分发给用户,避免了引入额外的可信第三方。
  2. 模型权重参数编码:在边缘服务器上,将卷积神经网络(CNN)的权重参数编码为同态加密的明文空间,以便在推理过程中使用。
  3. 同态加密计算:用户使用公钥加密数据并提交给边缘服务器,服务器在SGX外部进行线性计算(如卷积层和全连接层),以减少SGX的频繁内存访问和页面交换。
  4. 明文计算:将非线性计算(如激活层和池化层)放入SGX中进行解密和明文计算,然后将结果重新加密,以提高推理的准确性和效率。
  5. 重线性化:在SGX中进行解密和重新加密,以减少噪声积累,避免引入额外的可信第三方生成重线性化密钥。

实验结果与分析

本文通过实验验证了该框架的可行性和优势。实验结果表明,与纯HE推理方案相比,该框架在保持推理准确性的同时,减少了39.615%的推理时间。具体实验包括: - 密钥生成:在SGX中生成一对公钥和私钥的平均时间为49.593毫秒,比在SGX外生成密钥的时间(20.201毫秒)更长,但避免了引入可信第三方。 - 图像加密:对10张图像进行编码和加密的平均时间为157.013秒,每张图像的处理时间约为15.7秒。 - 权重参数编码:权重参数的编码时间与权重数量呈线性关系,且不受其他因素的影响。 - 推理结果解密:每张图像的推理结果解密和编码的平均时间为6.2391毫秒。

结论与意义

本文提出的混合框架结合了HE和SGX的优势,能够在保护用户隐私的同时提高神经网络推理的准确性和效率。通过实验验证,该框架在边缘计算场景中具有显著的优势,特别是在自动驾驶车辆(CAVs)等需要高隐私保护的场景中。该研究为隐私保护的神经网络推理提供了前瞻性的实践,展示了传统加密方案与可信硬件技术结合的潜力。

研究亮点

  1. 创新性框架:本文首次提出了一种结合HE和SGX的混合框架,用于隐私保护的神经网络推理,避免了引入额外的可信第三方。
  2. 性能提升:通过实验验证,该框架在保持推理准确性的同时,显著减少了推理时间,提高了系统的吞吐量。
  3. 应用前景:该框架在自动驾驶车辆等边缘计算场景中具有广泛的应用前景,能够有效保护用户的隐私数据。

其他有价值的内容

本文还讨论了单指令多数据(SIMD)操作和GPU加速在HE计算中的应用潜力,进一步提升了框架的性能和吞吐量。此外,本文还提供了详细的实验数据和分析,为未来的研究提供了参考。

总的来说,本文的研究为隐私保护的神经网络推理提供了一种高效且安全的解决方案,具有重要的理论和实践意义。

上述解读依据用户上传的学术文献,如有不准确或可能侵权之处请联系本站站长:admin@fmread.com