FlashKAN: B-Spline KANs via Truncated Power Form
Mirrored from arXiv — Machine Learning for archival readability. Support the source by reading on the original site.
Computer Science > Machine Learning
Title:FlashKAN: B-Spline KANs via Truncated Power Form
Abstract:Kolmogorov-Arnold Networks (KANs) place learnable B-spline activations on network edges rather than fixed activations on nodes. The standard Cox-de Boor recursion evaluates these activations through k sequential passes for degree-k splines, consuming over 90% of forward-pass time. FlashKAN replaces this recursion with the truncated power form, a classical result from approximation theory that expresses each uniform cubic B-spline as five (x)_+^3 terms at shifted knot positions. This paper makes three contributions: (1) a this http URL-fused implementation that collapses these operations into a single GPU kernel, eliminating all recursion, span lookup, and scatter-gather operations; (2) a bounded-coordinate stabilization that clamps the normalized input to [0, k+1], preventing the catastrophic cancellation that historically motivated the Cox-de Boor recursion; and (3) a production-ready, open-source package (pip install flashkan) that serves as a drop-in replacement for existing KAN layers.
| Comments: | 7 pages, 1 table, under review at ICLR 2027 |
| Subjects: | Machine Learning (cs.LG); Numerical Analysis (math.NA) |
| MSC classes: | 68T07, 65D07, 41A15 |
| ACM classes: | I.2.6; G.1.1 |
| Cite as: | arXiv:2609.01956 [cs.LG] |
| (or arXiv:2609.01956v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2609.01956
arXiv-issued DOI via DataCite (pending registration)
|
Access Paper:
- View PDF
- HTML (experimental)
- TeX Source
Current browse context:
References & Citations
Bookmark
Bibliographic and Citation Tools
Code, Data and Media Associated with this Article
Demos
Recommenders and Search Tools
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.
More from arXiv — Machine Learning
-
M3-Former: Multimodal Transformer with Mixture-of-Experts for Long-Term Vessel Trajectory Prediction
Sep 11
-
Halo: Improving forecast accuracy through heteroscedastic estimation
Sep 11
-
Zero-shot rib design: merging training-free generative prior with topology optimization
Sep 11
-
Byzantine-Robust Federated Fire Detection with a Rotating Coordinator
Sep 11
Discussion (0)
Sign in to join the discussion. Free account, 30 seconds — email code or GitHub.
Sign in →No comments yet. Sign in and be the first to say something.