QuantKAN: A Unified Quantization Framework for Kolmogorov Arnold Networks

22d ago · Global · primary source: export.arxiv.org

A new framework called QuantKAN provides the first unified approach for compressing Kolmogorov–Arnold Networks (KANs) to run efficiently on hardware, according to a paper posted on arXiv [1]. The work addresses a key barrier to deploying the spline-based models in low-precision environments. KANs differ from conventional neural networks by replacing linear weights with spline-based functions, a design that boosts expressivity but creates uneven parameter distributions that complicate quantization [1]. The QuantKAN framework tackles this with branch-aware quantizers that separately handle base and spline parameters, and it extends modern quantization-aware training (QAT) and post-training quantization (PTQ) techniques to spline-based layers across four KAN variants: EfficientKAN, FastKAN, PyKAN, and KAGN [1]. The paper, authored by researchers including Lizhong Chen, reports the first unified QAT and PTQ benchmarks for KANs on datasets ranging from MNIST to ImageNet [1]. Among QAT methods, Differentiable Soft Quantization (DSQ) proved most robust under aggressive low-bit constraints, while GPTQ emerged as the strongest PTQ method at moderate precision [1]. Sensitivity analyses uncovered architecture-specific failure modes: spline and basis parameters dominate in FastKAN, whereas base or scaling parameters are the critical points in EfficientKAN, GRAM, and PyKAN [1]. Hardware projections using Vivado HLS on a Xilinx UltraScale+ device suggest that a W4A4 configuration can deliver up to 3.32 times the throughput and 7.7 times lower estimated dynamic energy per inference compared to a baseline [1]. However, the analysis also identifies a residual “basis-evaluation tax” that persists even after quantization, prompting the authors to call for basis-aware microarchitecture design [1]. The QuantKAN codebase is publicly available on GitHub [1]. The paper has been revised twice since its initial submission in November 2025, with the latest version posted in June 2026 [1]. arXiv, the open-access repository hosting the work, was founded in 1991 and now receives roughly 24,000 submissions per month, serving as a primary distribution channel for preprints in computer science, physics, and mathematics [6].

tool-releaseresearch-papermodel-releaseproduct-launchinfrastructure

Background sources we checked (7)
  • arxiv.org ↗ Kolmogorov--Arnold Networks (KANs) replace linear weights with spline-based functions, offering strong expressivity but posing challenges for low-precision deployment due to heterogeneous parameter distributions. We introduce QuantKAN, the first unified framework for quantization…
  • info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…
  • blog.arxiv.org ↗ arXivLabs: a space for community innovation – arXiv blog arXiv has launched a new, formalized framework enabling innovative collaborations with individuals and organizations. “Members of our community want to contribute tools that enhance the arXiv experience, and we val…
  • info.arxiv.org ↗ arXivLabs: Showcase - arXiv info | arXiv e-print repository ... # arXivLabs: Showcase ... arXiv is surrounded by a community of researchers and developers working at the cutting edge of information science and technology. ... While the arXiv team is focused on our core mission—pr…
  • en.wikipedia.org ↗ arXiv (pronounced as "archive"—the X represents the Greek letter chi ⟨χ⟩) is an open-access repository of electronic preprints and postprints (known as e-prints) approved for posting after moderation, but not peer reviewed. It consists of scientific papers in the fields of mathem…
  • en.wikipedia.org ↗ 14 (fourteen) is the natural number following 13 and preceding 15.…
  • en.wikipedia.org ↗ A large language model (LLM) is a type of machine learning model designed for natural language processing tasks such as language generation. LLMs are language models with many parameters, and are trained with self-supervised learning on a vast amount of text.…

Sources

Spot something wrong? Report an issue