CVAILGJun 21, 2024

Demonstrating the Efficacy of Kolmogorov-Arnold Networks in Vision Tasks

arXiv:2406.14916v169 citations
Originality Incremental advance
AI Analysis

This work addresses the gap in applying KANs to image classification for researchers, though it is incremental as it builds on existing KAN methods.

The study tackled the problem of validating Kolmogorov-Arnold Networks (KAN) for vision tasks, showing that KAN outperformed MLP-Mixer on CIFAR10 and CIFAR100 but performed slightly worse than ResNet-18.

In the realm of deep learning, the Kolmogorov-Arnold Network (KAN) has emerged as a potential alternative to multilayer projections (MLPs). However, its applicability to vision tasks has not been extensively validated. In our study, we demonstrated the effectiveness of KAN for vision tasks through multiple trials on the MNIST, CIFAR10, and CIFAR100 datasets, using a training batch size of 32. Our results showed that while KAN outperformed the original MLP-Mixer on CIFAR10 and CIFAR100, it performed slightly worse than the state-of-the-art ResNet-18. These findings suggest that KAN holds significant promise for vision tasks, and further modifications could enhance its performance in future evaluations.Our contributions are threefold: first, we showcase the efficiency of KAN-based algorithms for visual tasks; second, we provide extensive empirical assessments across various vision benchmarks, comparing KAN's performance with MLP-Mixer, CNNs, and Vision Transformers (ViT); and third, we pioneer the use of natural KAN layers in visual tasks, addressing a gap in previous research. This paper lays the foundation for future studies on KANs, highlighting their potential as a reliable alternative for image classification tasks.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes