arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.07751cs.LGcs.CR

基于同态加密的高效隐私保护KAN推理

Efficient Privacy-Preserving KAN Inference Using Homomorphic Encryption

Sun Yat-sen University(中山大学)

Zhizheng Lai, Yufei Zhou, Peijia Zheng, Lin Chen

AI总结:

针对KAN推理的隐私泄露问题,提出基于同态加密的高效隐私保护推理方案,通过SiLU函数任务专用多项式近似、HE域B样条高效计算实现高精度推理,性能优于朴素方法与明文MLP。

摘要:

近期提出的Kolmogorov-Arnold Networks(KAN,柯尔莫哥洛夫-阿诺德网络)具备更强的可解释性与更高的模型表达能力。但KAN也存在推理过程中的隐私泄露挑战。同态加密(HE)为深度学习模型提供隐私保护推理能力,可让资源有限的用户在保障数据安全的前提下使用深度学习服务。然而KAN结构复杂,包含SiLU激活函数、B样条函数等非线性组件,导致现有隐私保护推理技术无法适用。为解决该问题,我们提出了一种专为KAN设计的精准高效的隐私保护推理方案。我们的方法针对SiLU激活函数引入了任务专用的多项式近似,通过动态调整近似范围确保在真实数据集上的高精度。此外,我们利用重复打包、延迟合并、比较函数等技术,提出了一种在HE域内计算B样条函数的高效方法。我们在符号公式计算和图像分类两类任务上评估了该隐私保护KAN推理方案的效果。实验结果表明,我们的模型在多个数据集上的精度与明文KAN相当,且优于明文MLP;此外在CIFAR-10数据集上,我们的推理延迟相比朴素方法实现了7倍以上的加速。

英文摘要:

The recently proposed Kolmogorov-Arnold Networks (KANs) offer enhanced interpretability and greater model expressiveness. However, KANs also present challenges related to privacy leakage during inference. Homomorphic encryption (HE) facilitates privacy-preserving inference for deep learning models, enabling resource-limited users to benefit from deep learning services while ensuring data security. Yet, the complex structure of KANs, incorporating nonlinear elements like the SiLU activation function and B-spline functions, renders existing privacy-preserving inference techniques inadequate. To address this issue, we propose an accurate and efficient privacy-preserving inference scheme tailored for KANs. Our approach introduces a task-specific polynomial approximation for the SiLU activation function, dynamically adjusting the approximation range to ensure high accuracy on real-world datasets. Additionally, we develop an efficient method for computing B-spline functions within the HE domain, leveraging techniques such as repeat packing, lazy combination, and comparison functions. We evaluate the effectiveness of our privacy-preserving KAN inference scheme on both symbolic formula evaluation and image classification. The experimental results show that our model achieves accuracy comparable to plaintext KANs across various datasets and outperforms plaintext MLPs. Additionally, on the CIFAR-10 dataset, our inference latency achieves over 7 times speedup compared to the naive method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑