cs.LGSep 3, 2026

SW-KAN: Kolmogorov-Arnold Networks with Stieltjes-Wigert q-Orthogonal Polynomials

Authors: Amirhosein Azarpour, Seyyed Moein Kazemi

Organizations: Faculty of Mathematical Sciences, Department of Computer Science, Shahid Beheshti University, Tehran · Faculty of Mathematical Sciences, Shahid Beheshti University, Tehran

Abstract

Kolmogorov-Arnold Networks (KANs) represent a paradigmatic shift in deep learning by replacing fixed node activations with learnable univariate functions on edges, offering enhanced interpretability and parameter efficiency. While recent polynomial-based KAN variants have addressed the computational overhead of original B-spline implementations, they introduce a fundamental yet underexplored challenge: the domain mismatch between unbounded real-valued inputs and the bounded or semi-infinite support of orthogonal polynomial bases. To address this limitation, we propose the Stieltjes-Wigert Kolmogorov-Arnold Network (SW-KAN), a novel architecture that employs Stieltjes-Wigert q-orthogonal polynomials defined on the semi-infinite domain (0, infinity). We introduce a smooth exponential-of-tanh mapping that stably bridges the domain gap while preserving well-conditioned gradients, and leverage a numerically stable three-term recurrence that evaluates polynomial expansions in O(N) operations without special-function calls. Through comprehensive experiments spanning image classification and continuous function approximation, we demonstrate that SW-KAN achieves superior accuracy-efficiency trade-offs across diverse tasks. The log-normal weight structure and learnable q-parameter of Stieltjes-Wigert polynomials provide a distinct inductive bias that enables robust performance under resource-constrained conditions, including reduced feature dimensionality and limited training data. The proposed architecture not only outperforms established polynomial KAN baselines on standard benchmarks but also exhibits strong representational capacity for approximating complex multivariate functions with remarkably few parameters, making it a compelling alternative for efficient function approximation and classification in resource-constrained settings.

Figures & tables

Explore similar work

CardsList
  1. QuantKAN: A Unified Quantization Framework for Kolmogorov Arnold Networks

    Nov 24, 2025Kazi Ahmed Asif Fuad, Lizhong ChenQuantization-Aware TrainingKolmogorov-Arnold Networks

  2. DualFlexKAN: Dual-stage Kolmogorov-Arnold Networks with Independent Function Control

    Mar 9, 2026Andrés Ortiz, Nicolás J. Gallego-Molina, Carmen Jiménez-Mesa +2Kolmogorov-Arnold NetworksNeural Network Approximation Theory

  3. FEKAN: Feature-Enriched Kolmogorov-Arnold Networks

    Feb 18, 2026Sidharth S. Menon, Ameya D. JagtapKolmogorov-Arnold NetworksNeural Network Approximation Theory