DP-KAN: Differentially Private Kolmogorov-Arnold Networks

Nikita P. Kalinin,Simone Bombari,Hossein Zakerinia,Christoph H. Lampert
2024-07-17
Abstract:We study the Kolmogorov-Arnold Network (KAN), recently proposed as an alternative to the classical Multilayer Perceptron (MLP), in the application for differentially private model training. Using the DP-SGD algorithm, we demonstrate that KAN can be made private in a straightforward manner and evaluated its performance across several datasets. Our results indicate that the accuracy of KAN is not only comparable with MLP but also experiences similar deterioration due to privacy constraints, making it suitable for differentially private model training.
Machine Learning,Cryptography and Security
What problem does this paper attempt to address?
The paper primarily discusses the application of Kolmogorov-Arnold networks (KAN) in model training under differential privacy protection. KAN is an emerging neural network architecture that serves as an alternative to the traditional multilayer perceptron (MLP), inspired by the Kolmogorov-Arnold representation theorem. In the paper, the authors demonstrate how to train KAN while protecting data privacy by introducing differential privacy mechanisms, particularly the DP-SGD algorithm. The research findings indicate that KAN not only has comparable accuracy to MLP in non-private settings but also exhibits similar performance under differential privacy constraints, meaning the degree of performance degradation is akin to that of MLP. This suggests that KAN could be an effective option for maintaining model performance while ensuring privacy. Specific experiments include: 1. Regression analysis was conducted on multiple tabular datasets, comparing the performance of KAN, linear regression, and their respective differential privacy versions. The results show that KAN experiences a smaller quality decline due to privacy on most datasets. 2. In the MNIST image classification task, a faster implementation of KAN (fasterKAN) was used and compared with MLP and its differential privacy version. The experiments indicate that fasterKAN can achieve high accuracy with fewer parameters and that its performance decline in a differential privacy setting is less pronounced than that of MLP. In summary, the paper proves the capability of KAN in handling differential privacy in network training, offering a new solution for machine learning on sensitive data.