Incorporating Arbitrary Matrix Group Equivariance into KANs
Not on the map. Why?
Abstract (arXiv)
Kolmogorov-Arnold Networks (KANs) have seen great success in scientific domains thanks to spline activation functions, becoming an alternative to Multi-Layer Perceptrons (MLPs). However, spline functions may not respect symmetry in tasks, which is crucial prior knowledge in machine learning. In this paper, we propose Equivariant Kolmogorov-Arnold Networks (EKAN), a method for incorporating arbitrary matrix group equivariance into KANs, aiming to broaden their applicability to more fields. We first construct gated spline basis functions, which form the EKAN layer together with equivariant linear weights, and then define a lift layer to align the input space of EKAN with the feature space of the dataset, thereby building the entire EKAN architecture. Compared with baseline models, EKAN achieves higher accuracy with smaller datasets or fewer parameters on symmetry-related tasks, such as particle scattering and the three-body problem, often reducing test MSE by several orders of magnitude. Even in non-symbolic formula scenarios, such as top quark tagging with three jet constituents, EKAN achieves comparable results with state-of-the-art equivariant architectures using fewer than 40% of the parameters, while KANs do not outperform MLPs as expected. Code and data are available at https://github.com/hulx2002/EKAN .
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 44 of its 61 references that are in the map. First posted on arXiv 1 Oct 2024.
Builds on
44 of 61 references.
References · 44
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2020GLU Variants Improve Transformercited by 1,804
- 2017Deep Setscited by 1,421
- 2016Group Equivariant Convolutional Networkscited by 1,165
- 2021E(n) Equivariant Graph Neural Networkscited by 965
- 2019Hamiltonian Neural Networkscited by 712
- 2016Language Modeling with Gated Convolutional Networkscited by 702
- 2024KAN: Kolmogorov-Arnold Networkscited by 649
- 2019General E(2)-Equivariant Steerable CNNscited by 446
- 20183D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Datacited by 354
- 2018Invariant and Equivariant Graph Networkscited by 346
- 2016Steerable CNNscited by 305
32 more
- 2020Self-supervised Equivariant Attention Mechanism for Weakly Supervised Semantic Segmentationcited by 300
- 2017Learning Steerable Filters for Rotation Equivariant CNNscited by 194
- 2021A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groupscited by 187
- 2019Universal Invariant and Equivariant Graph Neural Networkscited by 183
- 2024U-KAN Makes Strong Backbone for Medical Image Segmentation and Generationcited by 179
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 174
- 2020Incorporating Symmetry into Deep Dynamics Models for Improved Generalizationcited by 126
- 2020Lorentz Group Equivariant Neural Network for Particle Physicscited by 121
- 2024KAN 2.0: Kolmogorov-Arnold Networks Meet Sciencecited by 118
- 2024Kolmogorov-Arnold Networks (KANs) for Time Series Analysiscited by 98
- 2018Roto-Translation Covariant Convolutional Networks for Medical Image Analysiscited by 92
- 2024Convolutional Kolmogorov-Arnold Networkscited by 77
- 2024KAN or MLP: A Fairer Comparisoncited by 73
- 2024TKAN: Temporal Kolmogorov-Arnold Networkscited by 71
- 2024Kolmogorov-Arnold Networks for Time Series: Bridging Predictive Power and Interpretabilitycited by 70
- 2021Equivariant Contrastive Learningcited by 66
- 2024GKAN: Graph Kolmogorov-Arnold Networkscited by 53
- 2024A Temporal Kolmogorov-Arnold Transformer for Time Series Forecastingcited by 52
- 2024KAGNNs: Kolmogorov-Arnold Networks meet Graph Learningcited by 49
- 2024Kolmogorov-Arnold Network for Satellite Image Classification in Remote Sensingcited by 47
- 2023Self-supervised learning of Split Invariant Equivariant representationscited by 44
- 2024Demonstrating the Efficacy of Kolmogorov-Arnold Networks in Vision Taskscited by 40
- 2020PDO-eConvs: Partial Differential Operator Based Equivariant Convolutionscited by 34
- 2024Unveiling the Power of Wavelets: A Wavelet-based Kolmogorov-Arnold Network for Hyperspectral Image Classificationcited by 34
- 2024Suitability of KANs for Computer Vision: A preliminary investigationcited by 34
- 2020Dense Steerable Filter CNNs for Exploiting Rotational Symmetry in Histology Imagescited by 33
- 2024GraphKAN: Enhancing Feature Extraction with Graph Kolmogorov Arnold Networkscited by 32
- 2024Kolmogorov-Arnold Graph Neural Networkscited by 32
- 2022Self-Supervised Equivariant Learning for Oriented Keypoint Detectioncited by 26
- 2023Structuring Representation Geometry with Rotationally Equivariant Contrastive Learningcited by 25
- 2021PDO-eS2CNNs: Partial Differential Operator Based Equivariant Spherical CNNscited by 19
- 2022PDO-s3DCNNs: Partial Differential Operator Based Steerable 3D CNNscited by 4
Built on by
Citing papers per year: 2025: 3; 2026: 1.
4 papers.
- 2025MatrixKAN: Parallelized Kolmogorov-Arnold Networkcited by 7
- 2025Variational Kolmogorov-Arnold Networkcited by 1
- 2025FS-KAN: Permutation Equivariant Kolmogorov-Arnold Networks via Function Sharingcited by 0
- 2026Clifford Kolmogorov-Arnold Networkscited by 0
- 2026Clifford Kolmogorov-Arnold Networks5 Feb 2026
- 2025FS-KAN: Permutation Equivariant Kolmogorov-Arnold Networks via Function Sharing29 Sep 2025
- 2025Variational Kolmogorov-Arnold Network3 Jul 2025
- 2025MatrixKAN: Parallelized Kolmogorov-Arnold Network11 Feb 2025