本文提出了一種高效的隱私保護柯爾莫哥羅夫-阿諾夫網路(KAN)推論方案。KAN網路相比傳統神經網路具有更好的可解釋性,但其複雜的結構包含非線性元素如SiLU激活函數和B樣條函數,使得現有的隱私保護推論技術無法直接應用。
為此,本文提出了以下關鍵技術:
動態調整SiLU激活函數的近似範圍,並使用加權最小二乘法進行多項式近似,在保證高精度的同時,降低了計算複雜度。
開發了一種高效的B樣條函數計算方法,利用重複打包、延遲組合和比較函數等技術,大幅提升了計算效率。
將上述兩種技術集成到KAN推論框架中,在保持與明文KAN相當的準確度的同時,在CIFAR-10數據集上實現了超過7倍的推論速度提升。
實驗結果表明,本文提出的方法不僅在圖像分類任務上表現出色,在符號公式求值任務中也優於傳統多層感知機。此外,本文還分析了不同參數設置下的推論延遲,為實際應用提供了有價值的指導。
Sang ngôn ngữ khác
từ nội dung nguồn
arxiv.org
Thông tin chi tiết chính được chắt lọc từ
by Zhizheng Lai... lúc arxiv.org 09-13-2024
https://arxiv.org/pdf/2409.07751.pdfYêu cầu sâu hơn