分享自:

基于全同态加密的优化隐私保护CNN推理

期刊:IEEE Transactions on Information Forensics and SecurityDOI:10.1109/TIFS.2023.3263631

这篇文档属于类型a,即报告了一项原创研究的学术论文。以下是对该研究的详细介绍:

作者及机构
本研究的作者为Dongwoo Kim(隶属于韩国东国大学人工智能融合学院)和Cyril Guyot(隶属于美国西部数字研究所)。该研究发表在2023年的《IEEE Transactions on Information Forensics and Security》期刊上,属于信息安全和隐私保护领域。

研究背景
随着机器学习模型在金融、医疗等依赖敏感数据的领域中的广泛应用,数据隐私问题日益受到关注。隐私保护机器学习(PPML)成为研究热点,尤其是基于全同态加密(Fully Homomorphic Encryption, FHE)的安全推理因其严格的隐私保护特性而备受青睐。然而,由于FHE的高计算成本,之前的研究仅限于浅层和窄神经网络以及简单任务。本研究提出了一种更高效的方法,用于在FHE框架下评估卷积运算,显著降低了计算成本,使得深层和宽卷积神经网络(CNN)的安全推理成为可能。

研究目标
本研究旨在通过优化FHE下的卷积运算,提升CNN模型的隐私保护推理效率,特别是针对深层和宽网络结构。具体目标包括:
1. 提出一种与卷积核大小无关的恒定时间卷积评估方法;
2. 结合FHE的自举(bootstrapping)技术,减少多层CNN的推理时间;
3. 探索适用于FHE推理的CNN架构,以在CIFAR10/100和ImageNet数据集上实现更高的推理速度和准确性。

研究流程
1. 卷积运算的优化
- 提出了一种基于FHE的恒定时间卷积评估方法,通过将输入和卷积核编码为多项式系数,将单次卷积运算简化为一次乘法运算,无需旋转操作。
- 将该方法推广到批量卷积运算,通过扩展[5]的算法,将尽可能多的卷积输出打包到每个密文中。
- 结合FHE自举技术,优化卷积层的评估流程,使得卷积运算在系数域进行,而激活函数在槽域进行。

  1. 实验设计与实施

    • 微基准测试:对比了向量编码方法(作为基线)和本研究提出的方法在卷积和卷积层评估中的性能。实验表明,本研究方法在卷积核宽度为3、5、7时的评估速度分别提高了12倍、25倍和46倍。
    • CNN分类器评估:在CIFAR10/100和ImageNet数据集上评估了20层CNN分类器的性能。结果显示,本研究方法在CIFAR10/100和ImageNet上的推理时间分别减少了18.9%和48.1%。
    • CNN架构探索:研究了不同深度、宽度和卷积核大小的CNN变体,发现较低深度、较宽层和较大卷积核的CNN在FHE推理中表现出更好的性能。
  2. 数据分析与方法创新

    • 采用Lattigo库实现了向量编码和本研究方法的对比实验,确保了实验的公平性。
    • 提出了基于多项式表示和稀疏打包的卷积优化方法,显著减少了计算和通信成本。
    • 结合FHE自举技术,优化了卷积层和激活函数的评估流程,减少了推理时间。

主要结果
1. 卷积运算优化效果
- 对于不同卷积核大小(3、5、7),本研究方法的评估时间分别比基线方法减少了12倍、25倍和46倍,且与卷积核大小无关。
- 在卷积层评估中,结合FHE自举技术,推理时间减少了13%至83%,具体取决于卷积核大小和批量大小。

  1. CNN分类器性能提升

    • 在CIFAR10/100数据集上,20层CNN分类器的推理时间减少了18.9%,达到了94.0%的分类准确率。
    • 在ImageNet数据集上,后8层CNN的推理时间减少了48.1%,达到了73.5%的top-5准确率。
    • 通过探索更宽层和更大卷积核的CNN变体,本研究在CIFAR10/100数据集上实现了比之前工作快5倍以上的推理速度,同时保持或提高了准确性。
  2. CNN架构探索结果

    • 较低深度、较宽层和较大卷积核的CNN在FHE推理中表现出更好的性能。例如,深度为8、宽度因子为3、卷积核大小为5的CNN在CIFAR10上的准确率达到了93.5%,推理时间仅为255秒。

结论与意义
本研究提出了一种基于FHE的高效卷积评估方法,显著降低了CNN模型的安全推理成本。主要贡献包括:
1. 提出了与卷积核大小无关的恒定时间卷积评估方法,为深层和宽网络的隐私保护推理提供了技术支持。
2. 结合FHE自举技术,优化了卷积层和激活函数的评估流程,减少了推理时间。
3. 探索了适用于FHE推理的CNN架构,展示了在较低深度和较大卷积核条件下实现高准确性和低延迟的可行性。

研究亮点
1. 提出了新颖的基于多项式表示和稀疏打包的卷积优化方法,显著减少了FHE框架下的计算成本。
2. 结合FHE自举技术,实现了深层CNN的高精度推理,为隐私保护机器学习提供了新的可能性。
3. 通过探索更宽层和更大卷积核的CNN变体,展示了在FHE推理中实现高效性和准确性的新方向。

其他有价值内容
本研究的方法不仅适用于CKKS FHE方案,还可扩展到其他FHE方案(如BGV和B/FV),特别是在混合PPML推理中作为子程序使用。此外,该方法还可用于减少卷积运算中的通信成本,为高效的隐私保护机器学习推理提供了更广泛的应用前景。

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