---
title: 'LeanKAN: Parameter-Efficient Neural Surrogates'
url: https://www.emergentmind.com/topics/leankan
type: topic
---

# LeanKAN: Parameter-Efficient Neural Surrogates

LeanKAN is a parameter-efficient Kolmogorov–Arnold Network (KAN) layer that extends the expressiveness of neural surrogates by incorporating both additive and multiplicative interactions in a minimal-parameter architecture. Designed as a direct, modular replacement for traditional AddKAN and MultKAN layers, LeanKAN achieves increased memory efficiency, faster convergence, improved generalization, and superior interpretability across diverse scientific and engineering domains. With formal error guarantees, LeanKAN enables scalable modeling of complex nonlinear and dynamical systems in settings ranging from real-time diagnostics to hybrid physics-informed neural ODEs and adaptive control.

## 1. Mathematical Formulation and Architectural Principles

LeanKAN is rooted in the Kolmogorov–Arnold representation theorem, which states that any continuous multivariate function on a compact domain can be expressed as a finite composition of sums of univariate functions. The canonical form of a D-layer KAN is:
\[
f : \mathbb{R}^n \rightarrow \mathbb{R}^m, \quad \alpha_{d+1, b} = \sum_{a=1}^{\omega_d} \phi_{d, b, a}(\alpha_{d, a}), \quad b = 1, \ldots, \omega_{d+1}
\]
where $\phi_{d, b, a}$ are learnable one-dimensional activations. Each $\phi$ is parameterized, for example, as a sum of radial basis functions (RBFs) with a base nonlinearity:
\[
\phi_{l,i,j}(x) = \sum_{m=1}^N w^{\psi}_{l,i,j,m} \exp\Bigl(-\frac{|x-c_m|^2}{2h^2}\Bigr) + w^b_{l,i,j} b(x)
\]
The LeanKAN layer introduces a joint additive-multiplicative channel decomposition: for any layer $l$ with input $\mathbf{x}_l \in \mathbb{R}^{n_l}$ and output $\mathbf{y}_l \in \mathbb{R}^{n_{l+1}}$, a specified "multiplicative sub-dimension" $n_l^{\rm mu} \leq n_l$ is chosen. Each output applies
\[
y_{l,i}^{\rm mult} = \prod_{j=1}^{n_l^{\rm mu}} \phi_{l,i,j}(x_{l,j}), \quad y_{l,i}^{\rm add} = \sum_{j=n_l^{\rm mu}+1}^{n_l} \phi_{l,i,j}(x_{l,j}), \quad z_{l,i} = y_{l,i}^{\rm mult} + y_{l,i}^{\rm add}
\]
This structure—with $n_l n_{l+1}$ univariate functions per layer and no dummy activations—generalizes AddKAN (setting $n_l^{\rm mu}=0$) and eliminates the parameter inflation and restricted expressivity of previous MultKAN variants. The only new hyperparameter is $n_l^{\rm mu}$, typically chosen as $n_l/2$ for expressivity/memory balance [2502.17844].

## 2. Parameter Efficiency and Memory Improvements

LeanKAN architectures achieve strict parameter minimality for their given expressive capacity. For each layer:
\[
P_{\rm lean} = n_l\, n_{l+1}\, (N+1)
\]
where $N$ is the RBF grid size. Unlike MultKAN, which for order-$k$ multiplication and $n^m_{l+1}$ multiplicative outputs incurs
\[
P_{\rm mult} = n_l\, (n^a_{l+1} + k\, n^m_{l+1})\,(N+1)
\]
LeanKAN achieves joint add/mul representation with strictly $n_l n_{l+1}(N+1)$ parameters—identical to AddKAN, but with superior representational power. Empirical studies confirm 2–3$\times$ compression for equivalent test error in scientific regression, ODE, and PDE surrogate tasks [2502.17844, 2504.12580].

LeanKAN's lack of dummy activations and direct allocation of learnable $\phi_{l,i,j}$ to each channel ensures full memory efficiency, minimal forward-evaluation cost, and optimal utilization of learnable degrees of freedom.

## 3. Convergence, Expressivity, and Empirical Performance

In benchmark studies, LeanKAN outperforms AddKAN and MultKAN in convergence speed, generalization, and overfitting resistance. Empirical results include:

- **Neural ODEs (Lotka–Volterra):** LeanKAN achieves test MSE $2.8 \times 10^{-7}$ for a 120-parameter model, compared to $2.7 \times 10^{-6}$ for a 156-parameter MultKAN [2502.17844].
- **Complex-valued Schrödinger PDEs:** For equal parameter counts, LeanKAN models yield 2–10$\times$ lower test MSE than AddKAN, scaling linearly with parameter count, and maintain stability in noisy or irregular sampling regimes.
- **Toy regression:** LeanKAN’s multiplicative outputs achieve MSE as low as $10^{-3}$, while MultKAN plateaus at $2 \times 10^{-2}$.
- **Chemistry modeling (ChemKAN):** On hydrogen–air chemistry, a 344-parameter LeanKAN core achieves MSE $\approx 10^{-4}$ across hundreds of initial conditions and preserves prediction robustness in the presence of 15% synthetic noise—where DeepONet experiences $5\times$ degradation [2504.12580].
- **Real-time diagnostics (battery core temperature):** A 105-parameter LeanKAN, trained on high-fidelity data, enables sub-ms inference for model-free core-temperature estimation [2602.21155].

Convergence is facilitated by RBF activation smoothness and channel normalization, with no need for weight decay or explicit regularization. Early-epoch training dynamics show LeanKAN models escaping high-loss regimes significantly faster than MultKAN.

## 4. Applications and Domain-specific Schemes

LeanKAN has demonstrated impact in several scientific domains:

- **Battery Thermal Diagnostics:** The LeanKAN core provides real-time, model-free estimation of battery core temperature, interfaced with an online Koopman-based anomaly detector for rapid and reliable thermal fault identification, with analytical guarantees on false-alarm rate and detection latency. Simulation scenarios on commercial LiFePO$_4$ cells report up to 60% reductions in anomaly detection latency and sub-ms inference feasibility for embedded controllers [2602.21155].
- **Combustion Chemistry Surrogates ("ChemKAN"):** LeanKAN-parameterized KAN-ODEs compress complex reaction networks into O(100) parameters while maintaining 2$\times$ speedup over detailed solvers and order-of-magnitude MSE improvements over DeepONet, even under substantial noise and data sparsity [2504.12580].
- **Adaptive Control:** In Lyapunov-based adaptive control of nonlinear systems, LeanKAN designs enable explicit visualizable functional decompositions, real-time Jacobian-based weight updates, and formal stability with parameter convergence. Function approximation error is reduced by 18–20% relative to DNN/LSTM surrogates at equivalent tracking error levels [2512.21437].

Additional domains include learned PDE operators, parameter-efficient surrogates for stiff dynamical systems, and interpretable data-driven physical model extraction.

## 5. Theoretical Guarantees and Diagnostics

LeanKAN enables rigorous error and diagnostic guarantees. Its approximation properties are governed by:

\[
\|f - (\phi^G_D \circ \dots \circ \phi^G_1)\|_\infty \leq D G^{-(\kappa-1)}
\]
with $D=$ number of layers, $G=$ spline grid size, and $\kappa=$ spline order (for B-spline-parameterized $\phi$). This allows a priori selection of grid resolution to meet arbitrary sup-norm error cutoffs, independent of input dimensionality [2602.21155].

In hierarchical diagnostic settings, such as the LeanKAN+Koopman battery anomaly detector, key theoretical results include:

- **No-anomaly residual upper bound:** Residual $\hat{r}_k$ is bounded by model error and spline resolution, enabling false-alarm rate control via $G$.
- **Reliable anomaly detection:** Above a critical threshold (tuned by $G$ and Koopman sensitivity), all anomalies of physical relevance are provably detected with specified sensitivity and delay [2602.21155].

In Lyapunov-adaptive LeanKAN controllers, the reconstruction error's independence from input dimension avoids the curse of dimensionality, and Jacobian-based gradient updates guarantee global asymptotic tracking [2512.21437].

## 6. Practical Recommendations, Hyperparameters, and Limitations

LeanKAN can be deployed as a direct drop-in replacement for AddKAN/MultKAN layers without codebase structural change. Recommendations include:

- **Hyperparameter defaults:** $n_l^{\rm mu} \approx n_l/2$ is generally effective for balancing add and mul channels. The RBF grid size $N$ should be chosen based on smoothness/complexity tradeoff (4–10 recommended); base activation normalization (e.g., Swish, layermax) stabilizes training [2502.17844].
- **Single-layer model caveats:** Multiplicative interactions apply only to the first $n_l^{\rm mu}$ inputs; stacking with AddKAN layers or randomizing input order mitigates this.
- **Multiparameter scaling:** For large-scale problems (PDE surrogates, high-dimensional ODEs), parameter counts remain sublinear in output dimension due to channel-wise parameter sharing [2504.12580].

Identified limitations include absence of explicit symbolic guarantees (approximate interpretability—analytic forms require pruning), input-splitting effects on representational coverage in shallow stacks, and limited empirical benefit from hybrid MultKAN–LeanKAN cascades.

## 7. Outlook and Research Trajectories

The LeanKAN paradigm signals a synthesis of high expressivity, parameter-efficiency, and interpretability for scientific-ML applications. Open directions include:

- **Scalability:** Further improvements in parameter compression and domain-specific architectural bias (e.g., physics-preserving inductive splits) to extend application to multi-dimensional operator learning and long-range time-accurate surrogates [2504.12580].
- **Hardware acceleration:** Compiler optimizations and specialized hardware kernels for LeanKAN evaluation, promising further improvements beyond current $\sim2\times$ solver accelerations.
- **Diagnostics with guarantees:** Extended formal convergence properties for hybrid LeanKAN-operator pipelines in real-time monitoring, with provable safety margins in critical control and diagnostics domains [2602.21155, 2512.21437].
- **Analytic extraction:** Automated methods for symbolic simplification and downstream physical insight extraction from LeanKAN-trained functional surrogates.

LeanKAN’s continued incorporation into neural ODE, adaptive control, and operator learning frameworks is expected to drive future advances in trustworthy, high-speed, low-memory scientific computation.

Source: https://www.emergentmind.com/topics/leankan