#213 · Primary category: Computer Vision

torch-conv-kan

computer-vision convolutional-neural-networks kolmogorov-arnold-networks

A PyTorch collection of Kolmogorov-Arnold convolutional networks with 1D/2D/3D layers, ResNet/DenseNet models, and CIFAR/Tiny ImageNet training scripts.

Project last updated:08/20/26

GitHub Stars

530

Forks

45

Contributors

3

License

MIT

Why we included this project

Kolmogorov-Arnold networks swap fixed activation functions for learnable ones, and this repository collects convolutional versions of them in one place. You get 1D, 2D, and 3D KAN conv layers built on several basis-function families, from B-splines and Legendre to Chebyshev, Gram, wavelet, and Bernstein, so it is easy to compare how each behaves inside otherwise familiar architectures. The repo also bundles ResNet-, DenseNet-, VGG-, and U-Net-style models with accelerate-based training scripts, which makes reproducing the paper's CIFAR and Tiny ImageNet experiments fairly straightforward. Pretrained ImageNet-1k checkpoints are included, so you can see what these layers actually do at scale instead of only on small datasets. If you are weighing whether KAN convolutions belong in a vision pipeline, this is one of the more complete places to start.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category