核心概念
本文提出了一種高效的隱私保護柯爾莫哥羅夫-阿諾夫網路推論方案,通過動態調整激活函數近似範圍和高效的B樣條函數計算方法,實現了與明文柯爾莫哥羅夫-阿諾夫網路相當的準確度,同時大幅提升了推論效率。
摘要
本文提出了一種高效的隱私保護柯爾莫哥羅夫-阿諾夫網路(KAN)推論方案。KAN網路相比傳統神經網路具有更好的可解釋性,但其複雜的結構包含非線性元素如SiLU激活函數和B樣條函數,使得現有的隱私保護推論技術無法直接應用。
為此,本文提出了以下關鍵技術:
-
動態調整SiLU激活函數的近似範圍,並使用加權最小二乘法進行多項式近似,在保證高精度的同時,降低了計算複雜度。
-
開發了一種高效的B樣條函數計算方法,利用重複打包、延遲組合和比較函數等技術,大幅提升了計算效率。
-
將上述兩種技術集成到KAN推論框架中,在保持與明文KAN相當的準確度的同時,在CIFAR-10數據集上實現了超過7倍的推論速度提升。
實驗結果表明,本文提出的方法不僅在圖像分類任務上表現出色,在符號公式求值任務中也優於傳統多層感知機。此外,本文還分析了不同參數設置下的推論延遲,為實際應用提供了有價值的指導。
统计
在MNIST數據集上,我們的方法達到了97.17%的準確率,優於Remez方法的96.76%和最小二乘法的97.01%。
在Fashion-MNIST數據集上,我們的方法達到了89.35%的準確率,優於Remez方法的89.16%和最小二乘法的89.27%。
在CIFAR-10數據集上,我們的方法達到了57.38%的準確率,優於Remez方法的57.17%和最小二乘法的57.25%。
在Toy公式數據集上,我們的方法的RMSE為0.00157,優於多層感知機的0.14072。
在lpmv0公式數據集上,我們的方法的RMSE為0.00461,優於多層感知機的0.01404。
在lpmv1公式數據集上,我們的方法的RMSE為0.03489,優於多層感知機的0.07378。
引用
"本文提出了一種高效的隱私保護柯爾莫哥羅夫-阿諾夫網路(KAN)推論方案,通過動態調整激活函數近似範圍和高效的B樣條函數計算方法,實現了與明文KAN相當的準確度,同時大幅提升了推論效率。"
"在CIFAR-10數據集上,我們的推論延遲實現了超過7倍的速度提升,相比於簡單的實現方法。"