FI-KAN: Fractal Interpolation Kolmogorov-Arnold Networks
Organizations: Axiom Research Group · Dept. of Applied Mathematics and Computational Science, NM-AIST, Arusha, Tanzania · AIMS Research and Innovation Centre, Kigali, Rwanda
Abstract
Kolmogorov-Arnold Networks (KAN) employ B-spline bases on a fixed grid, providing no intrinsic multi-scale decomposition for non-smooth function approximation. We introduce Fractal Interpolation KAN (FI-KAN), which incorporates learnable fractal interpolation function (FIF) bases from iterated function system (IFS) theory into KAN. Two variants are presented: Pure FI-KAN (Barnsley, 1986) replaces B-splines entirely with FIF bases; Hybrid FI-KAN (Navascues, 2005) retains the B-spline path and adds a learnable fractal correction. The IFS contraction parameters give each edge a differentiable fractal dimension that adapts to target regularity during training. On a Holder regularity benchmark (), Hybrid FI-KAN outperforms KAN at every regularity level (1.3x to 33x). On fractal targets, FI-KAN achieves up to 6.3x MSE reduction over KAN, maintaining 4.7x advantage at 5 dB SNR. On non-smooth PDE solutions (scikit-fem), Hybrid FI-KAN achieves up to 79x improvement on rough-coefficient diffusion and 3.5x on L-shaped domain corner singularities. Pure FI-KAN's complementary behavior, dominating on rough targets while underperforming on smooth ones, provides controlled evidence that basis geometry must match target regularity. A fractal dimension regularizer provides interpretable complexity control whose learned values recover the true fractal dimension of each target. These results establish regularity-matched basis design as a principled strategy for neural function approximation.
Figures & tables
| Pure FI-KAN experiment | Hybrid FI-KAN experiment | ||||||
|---|---|---|---|---|---|---|---|
| Target | MLP | KAN | Pure | MLP | KAN | Hybrid | |
| polynomial | 1.00 | 1.42e-2 | 2.97e-3 | 3.20e-1 | 1.44e-2 | 2.97e-3 | 1.3e-5 |
| exp_sin | 1.00 | 2.18e-1 | 3.32e-4 | 4.41e-1 | 1.68e-1 | 3.32e-4 | 7.0e-6 |
| chirp | 1.00 | 4.51e-1 | 2.95e-1 | 3.34e-1 | 4.51e-1 | 2.95e-1 | 1.57e-1 |
| weierstrass | 1.64 | 2.20e-1 | 1.23e-1 | 1.01e-1 | 2.20e-1 | 1.23e-1 | 4.38e-2 |
| weierstrass | 1.68 | 4.48e-1 | 1.98e-1 | 2.15e-1 | 4.48e-1 | 1.98e-1 | 9.31e-2 |
| MLP | KAN | Hybrid FI-KAN | Improvement over KAN | |
|---|---|---|---|---|
| 0.2 | 7.42e-1 | 3.99e-1 | 3.02e-1 | |
| 0.4 | 4.59e-1 | 1.69e-1 | 6.77e-2 | |
| 0.6 | 2.94e-1 | 7.36e-2 | 1.20e-2 | |
| 0.8 | 1.94e-1 | 3.14e-2 | 2.34e-3 | |
| 1.0 | 1.34e-1 | 1.32e-2 | 6.30e-4 | |
| 1.5 | 6.99e-2 | 1.58e-3 | 4.8e-5 |
| Target | KAN (864p) | Hybrid FI-KAN (840p) | Ratio | Winner |
|---|---|---|---|---|
| polynomial | 1.58e-4 | 1.2e-5 | FI-KAN | |
| exp_sin | 4.0e-6 | 6.0e-6 | KAN | |
| weierstrass | 4.58e-2 | 4.24e-2 | FI-KAN | |
| sawtooth | 4.46e-3 | 1.81e-3 | FI-KAN |
| Test MSE at grid size | Scaling exponent | ||||||||
| Target | 3 | 5 | 8 | 12 | 16 | 20 | MLP | KAN | FI-KAN |
| Hybrid FI-KAN, smooth target: exp_sin | |||||||||
| MLP | 2.29e-1 | 2.02e-1 | 1.68e-1 | 1.08e-1 | 6.82e-2 | 4.16e-2 | 0.58 | 3.21 | 0.18 |
| KAN | 5.91e-2 | 3.10e-3 | 3.32e-4 | 9.1e-5 | 3.9e-5 | 3.3e-5 | |||
| FI-KAN | 9.0e-5 | 1.5e-5 | 6.0e-6 | 6.0e-6 | 1.2e-4 | 1.5e-5 | |||
| Hybrid FI-KAN, fractal target: sawtooth ( ) | |||||||||
| Weierstrass ( ) | Sawtooth ( ) | |||
|---|---|---|---|---|
| SNR (dB) | KAN | Hybrid FI-KAN | KAN | Hybrid FI-KAN |
| 100.0 | 1.567e-1 | 4.52e-2 | 1.139e-2 | 1.87e-3 |
| 40.0 | 1.568e-1 | 4.62e-2 | 1.140e-2 | 1.85e-3 |
| 30.0 | 1.569e-1 | 4.62e-2 | 1.141e-2 | 1.84e-3 |
| 20.0 | 1.573e-1 | 4.74e-2 | 1.144e-2 | 1.92e-3 |
| 10.0 | 1.579e-1 | 5.17e-2 | 1.155e-2 | 2.13e-3 |
| Model | Final MSE | Std | Relative to KAN |
|---|---|---|---|
| MLP | 1.294 | 0.008 | worse |
| KAN | 1.248 | 0.022 | (baseline) |
| Pure FI-KAN | 1.158 | 0.051 | better |
| Hybrid FI-KAN | 1.376 | 0.035 | worse |
| Polynomial ( ) | Weierstrass ( ) | |||
|---|---|---|---|---|
| Test MSE | Learned | Test MSE | Learned | |
| 0 | 3.74e-1 | 1.302 | 9.68e-2 | 1.150 |
| 3.69e-1 | 1.299 | 9.63e-2 | 1.146 | |
| 3.41e-1 | 1.294 | 9.15e-2 | 1.137 | |
| 1.23e-1 | 1.231 | 7.28e-2 | 1.053 | |
| 9.27e-3 | 1.001 | 6.09e-2 | 1.008 | |
| Depth | Test MSE | Std | Wall time (s) |
|---|---|---|---|
| 1 | 3.39e-2 | 8.35e-4 | 5.8 |
| 2 | 2.72e-2 | 8.33e-4 | 7.1 |
| 4 | 3.03e-2 | 7.95e-4 | 9.6 |
| 6 | 4.11e-2 | 1.45e-3 | 12.0 |
| 8 | 5.43e-2 | 3.63e-3 | 14.4 |
| 10 | 6.26e-2 | 3.87e-3 | 16.9 |
| Target | MLP | KAN | Pure FI-KAN | Hybrid FI-KAN |
|---|---|---|---|---|
| Ackley 2D | 4.27e-1 | 2.56e-2 | 6.03e-2 | 1.75e-3 |
| Weierstrass 2D | 4.32e-1 | 1.87e-1 | 1.25e-1 | 8.95e-2 |
| Problem | Roughness | MLP | KAN | Hybrid FI-KAN | Improv. | Source |
| Rough-coefficient diffusion , | ||||||
| rough coeff | 2.20e-3 | 1.26e-3 | 1.74e-5 | FEM | ||
| rough coeff | 3.05e-3 | 1.19e-3 | 1.50e-5 | FEM | ||
| rough coeff | 2.11e-3 | 1.15e-3 | 1.50e-5 | FEM | ||
| smooth coeff | 8.39e-4 | 1.11e-3 | 1.71e-5 | FEM | ||
| L-shaped domain , corner singularity | ||||||
| Target | True | Pure | Hybrid |
|---|---|---|---|
| smooth ( ) | 2.0 | ||
| ( ) | 1.0 | ||
| Hölder 0.6 | 0.6 | ||
| Hölder 0.3 | 0.3 | ||
| Weierstrass | 0.64 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Mean MSE | Params | |||||
|---|---|---|---|---|---|---|
| Target | MLP | KAN | Hybrid | MLP | KAN | Hybrid |
| Bessel | 8.62e-2 | 1.54e-4 | 2.18e-4 | 424 | 208 | 424 |
| 7.79e-1 | 4.32e-2 | 6.16e-2 | 609 | 312 | 608 | |
| 4.6e-5 | 8.3e-5 | 8.2e-5 | 609 | 312 | 608 | |
| 10D sum | 4.57e-3 | 1.21e-2 | 4.03e-4 | 1081 | 572 | 1080 |
| 4D composite | 2.50 | 1.38 | 4.27 | 639 | 338 | 652 |
| Target | MLP | KAN | Pure FI-KAN |
|---|---|---|---|
| Bessel | 8.62e-2 | 1.54e-4 | 7.30e-3 |
| 7.93e-1 | 4.32e-2 | 1.98 | |
| 6.1e-5 | 8.3e-5 | 8.22e-4 | |
| 10D sum | 8.17e-3 | 1.21e-2 | 3.05e-1 |
| 4D composite | 2.75 | 1.38 | 9.59 |
| Equation | MLP | KAN | Hybrid FI-KAN | Best |
|---|---|---|---|---|
| I.6.2 Gaussian | 6.46e-1 | 1.32e-2 | 9.77e-3 | FI-KAN |
| I.12.11 Lorentz | 9.40e-4 | 2.15e-2 | 4.53e-2 | MLP |
| I.16.6 Relativity | 8.27e-2 | 7.35e-3 | 1.97e-3 | FI-KAN |
| I.29.16 Distance | 2.00e-2 | 2.15e-3 | 2.92e-3 | KAN |
| I.30.3 Diffraction | 2.10e-1 | 1.07e-1 | 5.45e-2 | FI-KAN |
| I.50.26 Cosine | 7.33e-2 | 1.12e-2 | 1.33e-2 | KAN |