mlsanityData to 14 Oct 2026

LUT-KAN: Segment-wise LUT Quantization for Fast KAN Inference

arXiv 2601.03332 ↗First posted 6 Jan 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Kolmogorov--Arnold Networks (KAN) replace scalar weights by learnable univariate functions, often implemented with B-splines. This design can be accurate and interpretable, but it makes inference expensive on CPU because each layer requires many spline evaluations. Standard quantization toolchains are also hard to apply because the main computation is not a matrix multiply but repeated spline basis evaluation. This paper introduces LUT-KAN, a segment-wise lookup-table (LUT) compilation and quantization method for PyKAN-style KAN layers. LUT-KAN converts each edge function into a per-segment LUT with affine int8/uint8 quantization and linear interpolation. The method provides an explicit and reproducible inference contract, including boundary conventions and out-of-bounds (OOB) policies. We propose an ``honest baseline'' methodology for speed evaluation: B-spline evaluation and LUT evaluation are compared under the same backend optimization (NumPy vs NumPy and Numba vs Numba), which separates representation gains from vectorization and JIT effects. Experiments include controlled sweeps over LUT resolution L in 16, 32, 64, 128 and two quantization schemes (symmetric int8 and asymmetric uint8). We report accuracy, speed, and memory metrics with mean and standard deviation across multiple seeds. A two-by-two OOB robustness matrix evaluates behavior under different boundary modes and OOB policies. In a case study, we compile a trained KAN model for DoS attack detection (CICIDS2017 pipeline) into LUT artifacts. The compiled model preserves classification quality (F1 drop below 0.0002) while reducing steady-state CPU inference latency by 12x under NumPy and 10x under Numba backends (honest baseline). The memory overhead is approximately 10x at L=64. All code and artifacts are publicly available with fixed release tags for reproducibility.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 6 of its 28 references that are in the map. First posted on arXiv 6 Jan 2026.

Builds on

6 of 28 references.

References · 6

  1. 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
  2. 2024DeepOKAN: Deep Operator Network Based on Kolmogorov Arnold Networks for Mechanics Problemscited by 84
  3. 2024Kolmogorov Arnold Informed neural network: A physics-informed deep learning framework for solving forward and inverse problems based on Kolmogorov Arnold Networkscited by 67
  4. 2019Visual Wake Words Datasetcited by 46
  5. 2025Kolmogorov-Arnold Recurrent Network for Short Term Load Forecasting Across Diverse Consumerscited by 7
  6. 2025asKAN: Active Subspace embedded Kolmogorov-Arnold Networkcited by 5

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025QuantKAN: A Unified Quantization Framework for Kolmogorov Arnold Networks
  2. 2026KANtize: Exploring Low-bit Quantization of Kolmogorov-Arnold Networks for Efficient Inference
  3. 2025SHARe-KAN: Post-Training Vector Quantization for Cache-Resident KAN Inference
  4. 2025KANEL'E: Kolmogorov-Arnold Networks for Efficient LUT-based Evaluation
  5. 2024Hardware Acceleration of Kolmogorov-Arnold Network (KAN) for Lightweight Edge Inference
  6. 2025GS-KAN: Parameter-Efficient Kolmogorov-Arnold Networks via Sprecher-Type Shared Basis Functions
  7. 2025A Practitioner's Guide to Kolmogorov-Arnold Networks
  8. 2025Vec-LUT: Vector Table Lookup for Parallel Ultra-Low-Bit LLM Inference on Edge Devices