KAN-based Autoencoders in Theory and Practice
- KAN-based autoencoders (KAN-AEs) are architectures that replace fixed node activations with learnable univariate edge functions inspired by the Kolmogorov–Arnold theorem.
- They are applied across domains such as image reconstruction, asset pricing, channel coding, fault detection, and audio, using basis families like polynomial, B-spline, and Fourier.
- Empirical studies reveal competitive gains in reconstruction, classification, and energy efficiency while illustrating trade-offs in scalability, interpretability, and hyperparameter tuning.
Searching arXiv for recent work on KAN-based autoencoders and closely related KAN representation papers. KAN-based autoencoders (KAN-AEs) are autoencoder architectures in which the encoder, the decoder, or both are built from Kolmogorov–Arnold Network (KAN) layers rather than conventional multilayer perceptron blocks with fixed node activations. Their defining structural move is to place learnable univariate functions on edges and to aggregate those edge-wise transforms additively at receiving nodes, an arrangement explicitly connected in the literature to the Kolmogorov–Arnold representation theorem. By 2024–2026, the term had come to cover several related families: deterministic representation-learning autoencoders for images and denoising, conditional latent-factor models in asset pricing, OFDM channel-coding autoencoders with symbolic regression, industrial fault-detection autoencoders using different basis families, and hypernetwork-driven audio implicit representations interpreted from an autoencoder perspective (Yu et al., 2024, Wang et al., 2024, Moradi et al., 2024, Perre et al., 4 Jan 2026, Villagómez et al., 4 Aug 2025, Marszałek et al., 4 Mar 2025).
1. Mathematical basis and defining characteristics
The common mathematical premise is that KANs operationalize the Kolmogorov–Arnold representation theorem through edge-wise univariate functions. One form reported in the literature is
while another widely used statement is
At layer level, the corresponding KAN update is written as an additive aggregation of learned univariate edge maps, for example
or, in a more general multilayer notation,
This differs from an MLP layer of the form , where nonlinearity is node-local and fixed rather than edge-local and learned (Yu et al., 2024, Faroughi et al., 30 Jul 2025).
Within KAN-AEs, this structural change is used to replace the fixed-activation blocks of standard autoencoders. The autoencoder mapping remains conventional at the highest level—an encoder maps to a latent , and a decoder reconstructs —but the internal parameterization becomes a composition of KAN layers. Reported KAN-AE variants instantiate the learnable univariate functions with polynomial, B-spline, Gaussian RBF, Fourier, wavelet, or mixed SiLU-plus-spline bases, depending on the application domain (Yu et al., 2024, Moradi et al., 2024, Villagómez et al., 4 Aug 2025, Perre et al., 4 Jan 2026).
The interpretability claim associated with KAN-AEs follows directly from this edge-function view. In principle, each learned 0 or 1 can be plotted and inspected independently, so the model exposes a per-edge transformation rather than only aggregate weight matrices. Several papers emphasize this property, although not all of them provide systematic interpretability analyses or formal interpretability metrics (Wang et al., 2024, Moradi et al., 2024).
2. Architectural forms
The KAN-AE label covers multiple architectures rather than a single canonical template.
| Setting | Encoder/decoder structure | Basis or edge-function family |
|---|---|---|
| General representation learning | Symmetric KAN encoder and decoder, shallow three-layer topology | Polynomial edge functions with Sigmoid |
| Image reconstruction | KAN layer, ReLU, dense bottleneck; mirrored decoder | B-spline KAN via efficient implementation |
| Asset pricing | KAN beta network plus linear factor network | Learnable spline functions on edges |
| Channel coding | One KAN layer in encoder and one in decoder | B-spline plus SiLU, then SR and pruning |
| Fault detection | Symmetric 2 autoencoders | EfficientKAN, FastKAN, FourierKAN, WavKAN |
| Audio INR from an autoencoder perspective | Hypernetwork encoder predicts INR parameters; KAN INR decodes coordinates | SiLU plus B-spline mixture |
The most explicit deterministic formulation is the Kolmogorov–Arnold Auto-Encoder (KAE), which replaces the MLP layers in both encoder and decoder with KAN layers whose edge functions are learnable polynomials. It uses a shallow topology 3 with 4. Its layer is
5
with polynomial order 6, Sigmoid 7, and a learned constant matrix term 8. The same paper notes that, unlike a concurrent KAN-AE that interleaves KAN with dense/ReLU and uses B-splines, KAE eliminates dense layers and reduces hyperparameter overhead through the polynomial basis (Yu et al., 2024).
A different image-oriented design uses hybrid blocks rather than pure KAN stacks. In that formulation, the encoder is a KAN layer from input size to hidden size, followed by ReLU and a dense layer to the bottleneck; the decoder mirrors this structure in reverse. The paper gives an example encoder 9 via KAN, ReLU, then 0 via dense, and an example decoder 1 via dense, ReLU, then 2 via KAN. The KAN layers are implemented with B-spline basis functions through an efficient KAN library (Moradi et al., 2024).
In latent-factor asset pricing, the architecture departs from image reconstruction but still retains an encoder–decoder interpretation. The encoder, called the beta network, maps lagged firm characteristics 3 to factor exposures 4 using KAN layers built from learnable spline edge functions, wrapped by trainable linear embeddings 5 and 6. The decoder, called the factor network, is linear:
7
with 8 obtained from a ridge regression compression of returns onto characteristics (Wang et al., 2024).
The channel-coding literature uses an OFDM autoencoder in which a single KAN layer is placed in the encoder and another in the decoder. There, edge activations are parameterized as
9
with third-degree B-splines and five control points per activation. After training, these learned functions are converted to symbolic expressions by symbolic regression and then pruned, producing a symbolic-regression autoencoder closely matched to the trained KAN-AE (Perre et al., 4 Jan 2026).
Fault-detection studies compare several basis families under a fixed shallow autoencoder geometry 0: EfficientKAN-AE with B-splines and activity-based regularization, FastKAN-AE with Gaussian RBFs, FourierKAN-AE with truncated Fourier series, and WavKAN-AE with a single wavelet per edge (Villagómez et al., 4 Aug 2025). In audio, a broader “autoencoder perspective” is adopted: a hypernetwork 1 maps an audio segment to a task-specific parameter vector 2, and the decoder is an INR 3 evaluated on time coordinates. In that setting, the “latent” is the parameter vector itself (Marszałek et al., 4 Mar 2025).
3. Objectives, optimization, and evaluation protocols
The dominant objective in KAN-AEs is still reconstruction, but the exact loss is domain-specific. In KAE for representation learning, the model is trained as a deterministic autoencoder with mean squared reconstruction error only,
4
and the full objective is 5. The paper explicitly states that there are no KL divergences or variational terms, and no additional regularization such as smoothness or sparsity on edge functions. Denoising is evaluated by feeding noisy inputs to a model trained on clean data and measuring MSE under Gaussian noise 6 and salt-and-pepper noise with flip probability 7. The same work uses latent-space nearest-neighbor classification accuracy and Recall@10 for similarity search, with Euclidean distance in both input and latent spaces (Yu et al., 2024).
The asset-pricing formulation trains end-to-end on realized-return MSE,
8
without a separate pricing loss or factor-variance penalty. Its training protocol is rolling and temporally structured: train on 1957–1987, validate on 1987–1999, test on 2000–2016, with recursive yearly refits and validation-based tuning of the ridge parameter 9 and other hyperparameters (Wang et al., 2024).
The channel-coding KAN-AE departs from reconstruction loss and instead minimizes a modified cross-entropy over decoder logits, with noise scheduling across 0 epochs. Training uses Adam with learning rate 1, batch size 2, and an AWGN variance schedule corresponding to 3 from 4 dB to 5 dB. Its post-training symbolic-regression stage fits candidate symbolic functions 6 to learned edge activations using a combined score
7
where 8 is a non-linearity score based on the minimum piecewise-linear segmentation needed to satisfy a target approximation error (Perre et al., 4 Jan 2026).
Image-reconstruction KAN-AEs compared with CNN autoencoders use the standard MSE objective, AdamW with learning rate 9, weight decay 0, ten epochs, and additional L1 regularization on KAN weights in the efficient KAN implementation. Their downstream representation quality is assessed with a KNN classifier on the learned latent vectors, reporting accuracy and F1-score (Moradi et al., 2024).
In the audio hypernetwork setting, the objective is a two-term waveform and spectral loss,
1
where 2 is temporal L1 and 3 is a multi-resolution mel-STFT loss. This is optimized as a meta-objective over tasks, with the hypernetwork producing 4 and thus 5 (Marszałek et al., 4 Mar 2025). A broader scientific-machine-learning synthesis further suggests KAN-AE extensions with variational objectives and physics-informed penalties such as 6; this should be read as adapted KAN-AE guidance rather than a single benchmarked autoencoder configuration (Faroughi et al., 30 Jul 2025).
4. Empirical performance across application domains
The strongest general-purpose representation-learning evidence comes from KAE. Across MNIST, FashionMNIST, CIFAR10, and CIFAR100, KAE improved reconstruction, retrieval, classification, and denoising over a Sigmoid AE and over B-spline, Fourier, and wavelet KAN baselines. For reconstruction, KAE with 7 achieved the best errors on all datasets and both latent sizes; examples include MNIST with 8, where AE yielded 9 and KAE 0, and CIFAR100 with 1, where AE yielded 2 and KAE 3. For similarity search, KAE with 4 was best overall; on MNIST with 5, Recall@10 rose from 6 to 7, and on CIFAR100 from 8 to 9. For nearest-neighbor classification in latent space, KAE with 0 gave the highest accuracy, including MNIST 1 versus 2, CIFAR10 3 versus 4, and CIFAR100 5 versus 6. Under Gaussian noise, MNIST denoising MSE at 7 improved from 8 to 9 (Yu et al., 2024).
The image-reconstruction comparison with CNN autoencoders is more mixed. On SVHN and CIFAR-10, AE-KAN consistently achieved lower MSE than AE-CNN across the tested bottlenecks; for example, on SVHN the paper reports 0 versus 1, 2 versus 3, and 4 versus 5, while on CIFAR-10 it reports 6 versus 7, 8 versus 9, and 0 versus 1. On MNIST, AE-KAN outperformed AE-CNN at small and medium bottlenecks but lost at the largest tested bottleneck, where AE-CNN with 2 beat AE-KAN with bottleneck 3 in both MSE and KNN metrics. This study therefore presents KAN-AEs as competitive rather than uniformly dominant (Moradi et al., 2024).
In asset pricing, the KAN-based conditional autoencoder achieved comparable total 4 and better out-of-sample predictive 5 than the MLP-based conditional autoencoder across all tested numbers of latent factors. Predictive 6 values were 7, 8, and 9 for KAN-CA at 00, compared with 01, 02, and 03 for the MLP baseline. Long–short Sharpe ratios were 04, 05, and 06 for KAN-CA, versus 07, 08, and 09 for the baseline, with the clearest gain at 10 (Wang et al., 2024).
In communication systems, the KAN-AE paired with symbolic regression maintained BLER comparable to the MLP-AE under both AWGN and flat Rayleigh fading while reducing measured GPU energy. Reported hardware-based energy was 11 J for the MLP-AE and 12 J for the KAN-based SR-AE, giving an energy ratio of approximately 13. The hardware-independent non-linearity score 14 was 15 for the MLP-AE and 16 for the KAN-based SR-AE (Perre et al., 4 Jan 2026).
In fault detection on the Tennessee Eastman Process, WavKAN-AE achieved the highest overall FDR, reaching 17 using 4,000 training samples and remaining the top performer as other variants scaled. EfficientKAN-AE reached 18 FDR with only 500 samples and dominated the extreme low-data regime, while FastKAN-AE became competitive only at larger scales such as 19 samples. FourierKAN-AE consistently underperformed, and the Orthogonal Autoencoder baseline improved only gradually with more data (Villagómez et al., 4 Aug 2025).
For audio, the directly benchmarked KAN result is a KAN implicit representation rather than a standard bottleneck autoencoder, but the paper explicitly frames the hypernetwork-plus-INR pipeline as an autoencoder perspective. In standalone INR experiments, KAN achieved the best Log-Spectral Distance of 20 and the best PESQ of 21 for 22 s audio, and the best LSD of 23 for 24 s audio. In the hypernetwork setting, FewSound outperformed HyperSound, with a 25 reduction in MSE and a 26 increase in SI-SNR; among target networks, NeRF was best overall and KAN was a close second (Marszałek et al., 4 Mar 2025).
5. Interpretability, efficiency, and recurrent trade-offs
A recurring theme is that KAN-AEs trade fixed activations for inspectable edge functions. The clearest operational statement appears in asset pricing, where the learned spline functions 27 in the beta network can be visualized to understand how each characteristic contributes to each factor exposure, enabling partial-dependence visualization and sensitivity analysis at the granularity of characteristic-to-factor exposure channels (Wang et al., 2024). Related papers on image reconstruction and fault detection make similar interpretability claims, but they also note that explicit visualization or quantitative interpretability analysis is absent (Moradi et al., 2024, Villagómez et al., 4 Aug 2025).
Efficiency claims are highly application-dependent and should not be generalized indiscriminately. In KAE for representation learning, the polynomial design reduced hyperparameter overhead relative to spline-based KANs and achieved higher accuracy with fewer parameters than some competing KAN variants, with capacity reported at approximately 28–29K parameters versus approximately 30K for KAN and 31K for FourierKAN, and training times of 32–33 seconds (Yu et al., 2024). In channel coding, symbolic regression and pruning further reduced complexity and yielded lower measured energy than the MLP-AE baseline (Perre et al., 4 Jan 2026). In fault detection, WavKAN-AE used 34 parameters, fewer than OAE at 35, EfficientKAN-AE at 36, FastKAN-AE at 37, and FourierKAN-AE at 38 (Villagómez et al., 4 Aug 2025). By contrast, the image-reconstruction study found AE-KAN to have dramatically more parameters than AE-CNN—on CIFAR-10, roughly 39–40 million versus 41–42—even though reported runtimes were similar (Moradi et al., 2024). A common misconception is therefore that KAN-AEs are intrinsically parameter-efficient; the literature supports a narrower statement that efficiency depends on basis family, task geometry, and baseline choice.
Basis choice is another consistent trade-off. In KAE, polynomial order 43 tended to preserve local distances better and gave the best Recall@10, whereas 44 gave the best classification accuracy and slightly better reconstructions (Yu et al., 2024). In industrial fault detection, localized and multiresolution families performed best: EfficientKAN-AE was strongest in data-scarce regimes, WavKAN-AE was best overall, FastKAN-AE improved with scale, and FourierKAN-AE lagged, which the paper attributes to the mismatch between global periodic bases and localized nonstationary faults (Villagómez et al., 4 Aug 2025). In audio, spline-based KAN with a SiLU mixture improved perceptual fidelity relative to fixed-activation INR baselines, especially in LSD and PESQ (Marszałek et al., 4 Mar 2025).
This suggests that “KAN-AE” is best understood as a design space organized around edge-function parameterization rather than as a single model class with uniform behavior.
6. Limitations, open questions, and future directions
The existing KAN-AE literature remains heterogeneous and in several domains explicitly preliminary. The KAE representation-learning model is shallow and deterministic, with no KL regularization or probabilistic modeling; its authors propose combining it with VAE objectives, scaling to deeper architectures with KAN-based CNN layers and residual connections, exploring alternative activations, and applying it to more challenging generation tasks (Yu et al., 2024). The asset-pricing study notes that KAN best practices for depth, spline parameterization, regularization, and initialization are not yet standardized, and that identifiability remains limited to rotations and scalings, as in latent factor models generally (Wang et al., 2024).
The image-reconstruction study identifies a separate set of limitations: AE-KAN parameter counts are orders of magnitude higher than those of CNN baselines, scalability to larger images and datasets is not evaluated, training stability and hyperparameter sensitivity are not deeply analyzed, and interpretability is asserted rather than quantified (Moradi et al., 2024). The channel-coding study is limited to 45 codes, AWGN and flat Rayleigh channels, perfect CSI for Rayleigh equalization, and a specific GPU measurement setup; it explicitly leaves sensitivity analysis of the non-linearity metric and its weighting, larger block sizes, and broader robustness evaluation for future work (Perre et al., 4 Jan 2026). The fault-detection study is restricted to single time-step inputs, emphasizes detection rather than diagnosis, and does not systematically evaluate interpretability despite stressing the transparency potential of structured edge functions (Villagómez et al., 4 Aug 2025).
A broader review of scientific machine learning with KANs identifies the most persistent open problems as computational efficiency, theoretical guarantees, hyperparameter tuning, and algorithm complexity, together with robustness, scalability, and physical consistency (Faroughi et al., 30 Jul 2025). For KAN-AEs specifically, the literature therefore points toward four unsettled research directions: principled basis selection, scalable implementations for high-dimensional data, probabilistic and physics-informed extensions, and interpretability methods that move beyond visual plausibility to formal analysis.