STKAN: Kolmogorov-Arnold Networks for Spatio-Temporal Forecasting

arXiv cs.LG Papers

Summary

This paper introduces STKAN, a spatio-temporal forecasting architecture that integrates Taylor-polynomial Kolmogorov-Arnold Network modules for spatial and temporal token mixing. Experiments on five traffic benchmarks show competitive performance, suggesting nonlinear function approximators can complement architectural design.

arXiv:2607.13108v1 Announce Type: new Abstract: Real-world traffic data exhibit heterogeneous spatial correlations and nonlinear temporal dynamics, posing substantial challenges for accurate spatio-temporal forecasting. Existing approaches have developed increasingly sophisticated graph, attention, and decomposition architectures, while the influence of the underlying nonlinear function approximator has received comparatively less attention. In this work, we propose STKAN, a spatio-temporal forecasting architecture that introduces Taylor-polynomial Kolmogorov--Arnold Network modules into spatial and temporal token mixing. STKAN first constructs high-level spatial representations through a learnable soft node-group assignment mechanism, applies group-wise spatial mixing, and subsequently models temporal dependencies over the compressed sequence. Spatial and temporal self-attention layers are further employed to capture long-range interactions. Experiments on five traffic forecasting benchmarks show that STKAN achieves competitive performance and performs better than the evaluated MLP-based variant in the tested settings. These results suggest that the design of nonlinear function approximators can serve as a useful complement to architectural design in spatio-temporal forecasting.
Original Article
View Cached Full Text

Cached at: 07/16/26, 04:21 AM

# STKAN: Kolmogorov-Arnold Networks for Spatio-Temporal Forecasting
Source: [https://arxiv.org/html/2607.13108](https://arxiv.org/html/2607.13108)
,Yuehong HuThe Hong Kong University of Science and Technology \(Guangzhou\)China,Siru ZhongThe Hong Kong University of Science and Technology \(Guangzhou\)China,Si QiaoThe Hong Kong University of Science and Technology \(Guangzhou\)China,Yuxuan LiangThe Hong Kong University of Science and Technology \(Guangzhou\)ChinaandGuangyin JinChang’an UniversityChinaChina University of GeosciencesChina

###### Abstract\.

Real\-world traffic data exhibit heterogeneous spatial correlations and nonlinear temporal dynamics, posing substantial challenges for accurate spatio\-temporal forecasting\. Existing approaches have developed increasingly sophisticated graph, attention, and decomposition architectures, while the influence of the underlying nonlinear function approximator has received comparatively less attention\. In this work, we propose STKAN, a spatio\-temporal forecasting architecture that introduces Taylor\-polynomial Kolmogorov–Arnold Network modules into spatial and temporal token mixing\. STKAN first constructs high\-level spatial representations through a learnable soft node\-group assignment mechanism, applies group\-wise spatial mixing, and subsequently models temporal dependencies over the compressed sequence\. Spatial and temporal self\-attention layers are further employed to capture long\-range interactions\. Experiments on five traffic forecasting benchmarks show that STKAN achieves competitive performance and performs better than the evaluated MLP\-based variant in the tested settings\. These results suggest that the design of nonlinear function approximators can serve as a useful complement to architectural design in spatio\-temporal forecasting\.

††copyright:none## 1\.Introduction

Spatio\-temporal forecasting serves as a foundational pillar within the AI4Transport ecosystem\. It drives intelligent transportation technologies\(Chenet al\.,[2018](https://arxiv.org/html/2607.13108#bib.bib59); Wanget al\.,[2022](https://arxiv.org/html/2607.13108#bib.bib60); Linet al\.,[2022](https://arxiv.org/html/2607.13108#bib.bib61)\)\. By predicting future road traffic conditions from historical data, AI agents enable services such as autonomous traffic management, route optimization, and road safety support\(Wanget al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib62); Denget al\.,[2024](https://arxiv.org/html/2607.13108#bib.bib63); Hanet al\.,[2022](https://arxiv.org/html/2607.13108#bib.bib64)\)\. Traffic signals exhibit spatial heterogeneity and complex temporal dynamics\. Spatially, interactions between regions are intricate and non\-Euclidean\. Temporally, signals exhibit strong periodicity entangled with abrupt changes and non\-stationary patterns\. These properties motivate forecasting models that can capture both spatial dependencies and temporal variations\.

![Refer to caption](https://arxiv.org/html/2607.13108v1/11.png)Figure 1\.Illustration of different spatio\-temporal models\.To address these challenges, researchers have developed sophisticated architectures to explicitly model these dependencies\. As illustrated in Figure[1](https://arxiv.org/html/2607.13108#S1.F1), Graph\-based methods perform forecasting by assuming predefined or learned graph structures\. These structures capture spatial correlations among nodes\(Jinet al\.,[2023](https://arxiv.org/html/2607.13108#bib.bib67)\)\. Representative works such as STGCN\(Yuet al\.,[2018](https://arxiv.org/html/2607.13108#bib.bib32)\)and Graph WaveNet\(Wuet al\.,[2019](https://arxiv.org/html/2607.13108#bib.bib36)\)leverage graph convolutions and adaptive adjacency matrices\. Meanwhile, Transformer\-based methods utilize attention mechanisms\. They model long\-range dependencies across spatio\-temporal tokens\(Liuet al\.,[2023](https://arxiv.org/html/2607.13108#bib.bib56)\)\. Additionally, spectral decomposition techniques like StemGNN\(Caoet al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib53)\)and STWave\(Fanget al\.,[2023](https://arxiv.org/html/2607.13108#bib.bib54)\)focus on frequency\-domain transforms\. Frameworks like STID\(Shaoet al\.,[2022](https://arxiv.org/html/2607.13108#bib.bib41)\)utilize identity\-based embeddings\. These decoupled methods aim to obtain latent spatio\-temporal representations that support more accurate predictions\.

Despite diverse structural designs, many spatio\-temporal architectures employ MLPs with fixed activation functions as their default nonlinear mapping modules\. Although MLPs are expressive general\-purpose approximators, their nonlinearities are usually selected in advance and shared across units\. This observation motivates a complementary research question: beyond architectural design, to what extent does the choice of nonlinear function approximator affect spatio\-temporal forecasting performance?

In this work, we investigate whether the choice of nonlinear function approximator constitutes an underexplored factor in spatio\-temporal forecasting performance\. We propose STKAN, a Kolmogorov–Arnold Network \(KAN\)\-based architecture that decomposes spatial and temporal dependencies while introducing TaylorKAN layers into the spatial and temporal token\-mixing mappings\. STKAN provides a controlled architectural setting for examining whether Taylor\-polynomial KAN token mixers can complement conventional spatio\-temporal modeling components\. The channel mixer and Transformer feed\-forward network remain MLP\-based, so the model does not replace all MLP components\.

Our main contributions are summarized as follows:

- •We investigate the role of nonlinear function\-approximator design in spatio\-temporal forecasting and introduce STKAN, which incorporates Taylor\-polynomial KAN mappings into spatial and temporal token\-mixing modules\.
- •We develop a learnable soft node\-group assignment mechanism together with dedicated spatial and temporal mixing blocks\. The grouping mechanism constructs compact spatial representations, while the two token mixers model interactions along the spatial\-group and temporal dimensions, respectively\.
- •We evaluate STKAN on five traffic forecasting benchmarks\. The results demonstrate competitive forecasting accuracy, while the existing ablation results suggest that the TaylorKAN token mixers, adaptive spatial grouping, and attention components each contribute to the overall model in the evaluated configurations\.

## 2\.Related Work

### 2\.1\.Spatio\-Temporal Forecasting

Spatio\-temporal forecasting extends traditional time\-series forecasting by incorporating both temporal dynamics and spatial dependencies, such as in traffic management, where multiple traffic sensors’ data is used to predict future conditions\. Early deep learning approaches combined Convolutional Neural Networks \(CNNs\) and Recurrent Neural Networks \(RNNs\) to capture spatial and temporal dependencies\(Shiet al\.,[2015](https://arxiv.org/html/2607.13108#bib.bib26); Yaoet al\.,[2018](https://arxiv.org/html/2607.13108#bib.bib27); Laiet al\.,[2018](https://arxiv.org/html/2607.13108#bib.bib28)\)\. However, grid\-based CNNs may not effectively handle non\-Euclidean spatial relationships, leading to the development of Graph Convolutional Networks \(GCNs\)\(Defferrardet al\.,[2016](https://arxiv.org/html/2607.13108#bib.bib29); Kipf and Welling,[2016](https://arxiv.org/html/2607.13108#bib.bib30)\)and spatio\-temporal Graph Neural Networks \(STGNNs\)\(Liet al\.,[2017](https://arxiv.org/html/2607.13108#bib.bib31); Yuet al\.,[2018](https://arxiv.org/html/2607.13108#bib.bib32); Liet al\.,[2023](https://arxiv.org/html/2607.13108#bib.bib34)\)\. These models, such as DCRNN\(Liet al\.,[2017](https://arxiv.org/html/2607.13108#bib.bib31)\), ST\-MetaNet\(Panet al\.,[2019](https://arxiv.org/html/2607.13108#bib.bib33)\), and DGCRN\(Liet al\.,[2023](https://arxiv.org/html/2607.13108#bib.bib34)\), integrate GCNs with RNNs\(Choet al\.,[2014](https://arxiv.org/html/2607.13108#bib.bib35)\), while others like Graph WaveNet\(Wuet al\.,[2019](https://arxiv.org/html/2607.13108#bib.bib36)\)and STGCN\(Yuet al\.,[2018](https://arxiv.org/html/2607.13108#bib.bib32)\)combine GCNs with gated Temporal Convolutional Networks \(TCNs\)\. Attention mechanisms have also been widely adopted in STGNNs\(Zhenget al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib37)\)\. Some studies criticize the reliance on pre\-defined graphs, suggesting alternatives like AGCRN\(Baiet al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib38)\)and MTGNN\(Wuet al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib39)\), which learn latent graph structures\. Recent non\-GCN solutions, such as STNorm\(Denget al\.,[2021](https://arxiv.org/html/2607.13108#bib.bib40)\)and STID\(Shaoet al\.,[2022](https://arxiv.org/html/2607.13108#bib.bib41)\), further highlight the need for a better understanding of spatial dependencies in forecasting tasks\.

### 2\.2\.Kolmogorov\-Arnold Network

KANs are motivated by the Kolmogorov–Arnold representation theorem\(Liuet al\.,[2024](https://arxiv.org/html/2607.13108#bib.bib42)\), which represents multivariate functions through compositions of univariate functions\. Building upon this paradigm, several variants have been proposed, including wavelet\-based WavKAN\(Bozorgasl and Chen,[2024](https://arxiv.org/html/2607.13108#bib.bib44)\), Taylor polynomial\-based TaylorKAN\(Yuet al\.,[2025](https://arxiv.org/html/2607.13108#bib.bib45)\), fKAN with trainable Jacobi basis functions\(Aghaei,[2025](https://arxiv.org/html/2607.13108#bib.bib46)\), and FastKAN\(Li,[2024](https://arxiv.org/html/2607.13108#bib.bib47)\), which employs radial basis function approximations of B\-spline bases\.

Although KANs are motivated by the Kolmogorov–Arnold representation theorem, the theorem itself does not imply that a finite KAN model will universally outperform an MLP in empirical forecasting tasks\. In this work, KANs are therefore treated as an alternative nonlinear parameterization with a different inductive bias, rather than as a theoretically guaranteed replacement for MLPs\.

KANs have recently attracted attention in time series forecasting\. Prior work has explored KAN\-based models for time series analysis\(Vaca\-Rubioet al\.,[2024](https://arxiv.org/html/2607.13108#bib.bib65)\), mixture\-of\-experts KAN designs\(Hanet al\.,[2024](https://arxiv.org/html/2607.13108#bib.bib43)\), symbolic\-regression\-oriented T\-KAN and MT\-KAN models\(Xuet al\.,[2024](https://arxiv.org/html/2607.13108#bib.bib51)\), collaborative time\-frequency learning with iTFKAN\(Lianget al\.,[2025](https://arxiv.org/html/2607.13108#bib.bib48)\), and frequency\-decomposition learning with TimeKAN\(Huanget al\.,[2025](https://arxiv.org/html/2607.13108#bib.bib49)\)\. BiLSTM\-KAN integrates KAN layers with a bidirectional LSTM backbone for traffic flow forecasting\(Chenet al\.,[2024](https://arxiv.org/html/2607.13108#bib.bib66)\)\. STKAN differs from these studies by replacing the token\-mixing mappings in its spatial and temporal mixer blocks with Taylor\-polynomial KAN layers, while retaining conventional MLPs in the channel\-mixing and Transformer feed\-forward components\.

## 3\.Preliminary

Spatio\-temporal forecasting is a specialized multivariate time\-series forecasting problem\. Given a historical input sequence𝐗t−Lin\+1:t=\[𝐗t−Lin\+1,…,𝐗t\]∈ℝLin×N×Cin\\mathbf\{X\}\_\{t\-L\_\{\\mathrm\{in\}\}\+1:t\}=\[\\mathbf\{X\}\_\{t\-L\_\{\\mathrm\{in\}\}\+1\},\\ldots,\\mathbf\{X\}\_\{t\}\]\\in\\mathbb\{R\}^\{L\_\{\\mathrm\{in\}\}\\times N\\times C\_\{\\mathrm\{in\}\}\}, the goal is to predict𝐘^t\+1:t\+H=\[𝐘^t\+1,…,𝐘^t\+H\]∈ℝH×N×Cout\\widehat\{\\mathbf\{Y\}\}\_\{t\+1:t\+H\}=\[\\widehat\{\\mathbf\{Y\}\}\_\{t\+1\},\\ldots,\\widehat\{\\mathbf\{Y\}\}\_\{t\+H\}\]\\in\\mathbb\{R\}^\{H\\times N\\times C\_\{\\mathrm\{out\}\}\}\. Here,LinL\_\{\\mathrm\{in\}\}denotes the historical input length,HHdenotes the prediction horizon,NNis the number of spatial nodes, andCinC\_\{\\mathrm\{in\}\}andCoutC\_\{\\mathrm\{out\}\}are the input and output channel dimensions\.

## 4\.Methodology

![Refer to caption](https://arxiv.org/html/2607.13108v1/123.png)Figure 2\.Overview of our proposed STKAN framework\.In this paper, we propose STKAN to model spatial interactions and temporal dynamics by introducing TaylorKAN token mixers into a spatio\-temporal forecasting architecture\. The overall architecture of STKAN is shown in Figure[2](https://arxiv.org/html/2607.13108#S4.F2)\.

### 4\.1\.Embedding Layers

To capture spatio\-temporal dependencies in traffic sequences, we embed the raw input𝐗∈ℝLin×N×Cin\\mathbf\{X\}\\in\\mathbb\{R\}^\{L\_\{\\mathrm\{in\}\}\\times N\\times C\_\{\\mathrm\{in\}\}\}with a fully connected layer as𝐄f=FC​\(𝐗\)∈ℝLin×N×df\\mathbf\{E\}\_\{f\}=\\mathrm\{FC\}\(\\mathbf\{X\}\)\\in\\mathbb\{R\}^\{L\_\{\\mathrm\{in\}\}\\times N\\times d\_\{f\}\}\. We use a learnable spatial embedding𝐄s∈ℝN×ds\\mathbf\{E\}\_\{s\}\\in\\mathbb\{R\}^\{N\\times d\_\{s\}\}\. The spatial embedding is shared across time steps and is broadcast along the temporal dimension before concatenation\. To incorporate temporal periodicity, we use two embedding dictionaries: a time\-of\-day embedding withTtod=288T\_\{\\mathrm\{tod\}\}=288slots for five\-minute data and a day\-of\-week embedding withTdow=7T\_\{\\mathrm\{dow\}\}=7categories\. The resulting embeddings are denoted as𝐄tod\\mathbf\{E\}\_\{\\mathrm\{tod\}\}and𝐄dow\\mathbf\{E\}\_\{\\mathrm\{dow\}\}after lookup and broadcasting to the node dimension\.

### 4\.2\.Temporal Convolution Blocks

All embeddings are concatenated along the feature dimension to form the initial representation:

\(1\)𝐗\(0\)=𝐄f​‖𝐄s‖​𝐄tod∥𝐄dow∈ℝLin×N×dh,\\mathbf\{X\}^\{\(0\)\}=\\mathbf\{E\}\_\{f\}\\\|\\mathbf\{E\}\_\{s\}\\\|\\mathbf\{E\}\_\{\\mathrm\{tod\}\}\\\|\\mathbf\{E\}\_\{\\mathrm\{dow\}\}\\in\\mathbb\{R\}^\{L\_\{\\mathrm\{in\}\}\\times N\\times d\_\{h\}\},wheredh=df\+ds\+dtod\+ddowd\_\{h\}=d\_\{f\}\+d\_\{s\}\+d\_\{\\mathrm\{tod\}\}\+d\_\{\\mathrm\{dow\}\}\. To capture local temporal context while reducing the temporal resolution, we define a patch\-extraction operator that applies a1×w1\\times wconvolution along the time axis with stridess:

\(2\)𝐗p=PatchConv​\(𝐗\(0\);w,s\)∈ℝLp×N×dh,\\mathbf\{X\}\_\{p\}=\\mathrm\{PatchConv\}\\left\(\\mathbf\{X\}^\{\(0\)\};w,s\\right\)\\in\\mathbb\{R\}^\{L\_\{p\}\\times N\\times d\_\{h\}\},whereLp=⌊Lin−ws⌋\+1L\_\{p\}=\\left\\lfloor\\frac\{L\_\{\\mathrm\{in\}\}\-w\}\{s\}\\right\\rfloor\+1\. Thus,𝐗p\\mathbf\{X\}\_\{p\}serves as a patch\-level spatio\-temporal feature map for subsequent mixer and attention blocks\.

### 4\.3\.Spatio\-Temporal KAN Blocks

The STKAN blocks combine learnable spatial grouping, spatial TaylorKAN token mixing, and temporal TaylorKAN token mixing\. STKAN first uses a soft assignment matrix to aggregate raw nodes into macro\-level spatial tokens\. The aggregated representations are then processed by the Spatial KAN \(SKAN\) block\. The spatial output is mapped back to node\-level features and passed to the Temporal KAN \(TKAN\) block, which models dependencies over the compressed temporal dimension\.

Learnable Spatial Grouping\.To transform the original spatial nodes into macro\-level spatial tokens, we introduce an unnormalized learnable parameter matrix𝐀∈ℝN×G\\mathbf\{A\}\\in\\mathbb\{R\}^\{N\\times G\}\. The assignment matrix is obtained by applying softmax along the group dimension:

\(3\)𝐆n,g=exp⁡\(𝐀n,g\)∑g′=1Gexp⁡\(𝐀n,g′\)\.\\mathbf\{G\}\_\{n,g\}=\\frac\{\\exp\(\\mathbf\{A\}\_\{n,g\}\)\}\{\\sum\_\{g^\{\\prime\}=1\}^\{G\}\\exp\(\\mathbf\{A\}\_\{n,g^\{\\prime\}\}\)\}\.Thus,∑g=1G𝐆n,g=1\\sum\_\{g=1\}^\{G\}\\mathbf\{G\}\_\{n,g\}=1for each nodenn\. Given𝐗p∈ℝLp×N×dh\\mathbf\{X\}\_\{p\}\\in\\mathbb\{R\}^\{L\_\{p\}\\times N\\times d\_\{h\}\}, the group\-level representation𝐗~∈ℝLp×G×dh\\widetilde\{\\mathbf\{X\}\}\\in\\mathbb\{R\}^\{L\_\{p\}\\times G\\times d\_\{h\}\}is computed as

\(4\)𝐗~τ,g,c=∑n=1N𝐆n,g​𝐗p,τ,n,c\.\\widetilde\{\\mathbf\{X\}\}\_\{\\tau,g,c\}=\\sum\_\{n=1\}^\{N\}\\mathbf\{G\}\_\{n,g\}\\mathbf\{X\}\_\{p,\\tau,n,c\}\.
Spatial KAN Block\.The SKAN block follows a mixer\-style design\. Although MLPs are expressive general\-purpose approximators, their activation functions are usually fixed before training\. KAN\-based mappings provide a different parameterization in which learnable univariate functions are associated with network connections\. The transmission from thejj\-th neuron in layerℓ\+1\\ell\+1to all neurons in layerℓ\\ellis formulated as:

\(5\)zℓ\+1,j=∑i=1nℓϕℓ,j,i​\(zℓ,i\),z\_\{\\ell\+1,j\}=\\sum\_\{i=1\}^\{n\_\{\\ell\}\}\\phi\_\{\\ell,j,i\}\(z\_\{\\ell,i\}\),wherezℓ,iz\_\{\\ell,i\}is theii\-th neuron in layerℓ\\ell,nℓn\_\{\\ell\}is the number of neurons in that layer, andϕℓ,j,i​\(⋅\)\\phi\_\{\\ell,j,i\}\(\\cdot\)is a learnable univariate mapping\.

In our model,ϕ\\phiis instantiated by a TaylorKAN layer\. A Taylor expansion motivates the polynomial basis:

\(6\)f​\(x\)≈∑r=0Kf\(r\)​\(0\)r\!​xr\.f\(x\)\\approx\\sum\_\{r=0\}^\{K\}\\frac\{f^\{\(r\)\}\(0\)\}\{r\!\}x^\{r\}\.The implemented layer uses learnable coefficients:

\(7\)ϕq​\(𝐱\)=∑p=1C∑r=0Kθq,p,r​xpr\+bq\.\\phi\_\{q\}\(\\mathbf\{x\}\)=\\sum\_\{p=1\}^\{C\}\\sum\_\{r=0\}^\{K\}\\theta\_\{q,p,r\}x\_\{p\}^\{r\}\+b\_\{q\}\.The coefficientsθq,p,r\\theta\_\{q,p,r\}are optimized directly and are not constrained to equal the analytical derivatives of an underlying function\. Accordingly, the adopted layer is more precisely interpreted as a Taylor\-inspired polynomial\-basis KAN layer\. The polynomial basis allows the mapping to combine nonlinear terms of different orders\. Its effectiveness for spatial and temporal mixing is evaluated empirically in the subsequent experiments\.

To avoid ambiguity when mixing tensor dimensions, we define a spatial permutation operator𝒫s:ℝLp×G×dh→ℝLp×dh×G\\mathcal\{P\}\_\{s\}:\\mathbb\{R\}^\{L\_\{p\}\\times G\\times d\_\{h\}\}\\rightarrow\\mathbb\{R\}^\{L\_\{p\}\\times d\_\{h\}\\times G\}\. The spatial token\-mixing stage is then

\(8\)𝐔s=𝒫s​\(𝐗~\)\+TaylorKANs​\(LN​\(𝒫s​\(𝐗~\)\)\),\\mathbf\{U\}\_\{s\}=\\mathcal\{P\}\_\{s\}\(\\widetilde\{\\mathbf\{X\}\}\)\+\\mathrm\{TaylorKAN\}\_\{s\}\\left\(\\mathrm\{LN\}\\left\(\\mathcal\{P\}\_\{s\}\(\\widetilde\{\\mathbf\{X\}\}\)\\right\)\\right\),where𝐔s∈ℝLp×dh×G\\mathbf\{U\}\_\{s\}\\in\\mathbb\{R\}^\{L\_\{p\}\\times d\_\{h\}\\times G\}\. We restore the original group\-feature order by

\(9\)𝐔¯s=𝒫s−1​\(𝐔s\)∈ℝLp×G×dh\.\\overline\{\\mathbf\{U\}\}\_\{s\}=\\mathcal\{P\}\_\{s\}^\{\-1\}\(\\mathbf\{U\}\_\{s\}\)\\in\\mathbb\{R\}^\{L\_\{p\}\\times G\\times d\_\{h\}\}\.The channel mixer remains an MLP applied to the feature dimension:

\(10\)𝐕s=𝐔¯s\+𝐖2​σ​\(𝐖1​LN​\(𝐔¯s\)\+𝐛1\)\+𝐛2,\\mathbf\{V\}\_\{s\}=\\overline\{\\mathbf\{U\}\}\_\{s\}\+\\mathbf\{W\}\_\{2\}\\sigma\\left\(\\mathbf\{W\}\_\{1\}\\mathrm\{LN\}\(\\overline\{\\mathbf\{U\}\}\_\{s\}\)\+\\mathbf\{b\}\_\{1\}\\right\)\+\\mathbf\{b\}\_\{2\},where𝐕s∈ℝLp×G×dh\\mathbf\{V\}\_\{s\}\\in\\mathbb\{R\}^\{L\_\{p\}\\times G\\times d\_\{h\}\}andσ\\sigmadenotes the GELU activation\. The node\-level spatial output is

\(11\)𝐒τ,n,c=𝐗p,τ,n,c\+∑g=1G𝐆n,g​𝐕s,τ,g,c\.\\mathbf\{S\}\_\{\\tau,n,c\}=\\mathbf\{X\}\_\{p,\\tau,n,c\}\+\\sum\_\{g=1\}^\{G\}\\mathbf\{G\}\_\{n,g\}\\mathbf\{V\}\_\{s,\\tau,g,c\}\.
Temporal KAN Blocks\.The TKAN block applies TaylorKAN token mixing along the compressed temporal dimension\. We define a temporal permutation operator𝒫t:ℝLp×N×dh→ℝN×dh×Lp\\mathcal\{P\}\_\{t\}:\\mathbb\{R\}^\{L\_\{p\}\\times N\\times d\_\{h\}\}\\rightarrow\\mathbb\{R\}^\{N\\times d\_\{h\}\\times L\_\{p\}\}\. The temporal token mixer is

\(12\)𝐔t=𝒫t​\(𝐒\)\+TaylorKANt​\(LN​\(𝒫t​\(𝐒\)\)\),\\mathbf\{U\}\_\{t\}=\\mathcal\{P\}\_\{t\}\(\\mathbf\{S\}\)\+\\mathrm\{TaylorKAN\}\_\{t\}\\left\(\\mathrm\{LN\}\\left\(\\mathcal\{P\}\_\{t\}\(\\mathbf\{S\}\)\\right\)\\right\),where𝐔t∈ℝN×dh×Lp\\mathbf\{U\}\_\{t\}\\in\\mathbb\{R\}^\{N\\times d\_\{h\}\\times L\_\{p\}\}\. The channel mixer is written as

\(13\)𝐕t=𝒫t−1​\(𝐔t\)\+MLPt​\(LN​\(𝒫t−1​\(𝐔t\)\)\),\\mathbf\{V\}\_\{t\}=\\mathcal\{P\}\_\{t\}^\{\-1\}\(\\mathbf\{U\}\_\{t\}\)\+\\mathrm\{MLP\}\_\{t\}\\left\(\\mathrm\{LN\}\\left\(\\mathcal\{P\}\_\{t\}^\{\-1\}\(\\mathbf\{U\}\_\{t\}\)\\right\)\\right\),and the final temporal representation is

\(14\)𝐓=𝐕t∈ℝLp×N×dh\.\\mathbf\{T\}=\\mathbf\{V\}\_\{t\}\\in\\mathbb\{R\}^\{L\_\{p\}\\times N\\times d\_\{h\}\}\.

### 4\.4\.Spatio\-Temporal Fusion Blocks

To complement the TaylorKAN token mixers, we introduce Transformer layers along both spatial and temporal axes\. Given a hidden representation𝐓∈ℝLp×N×dh\\mathbf\{T\}\\in\\mathbb\{R\}^\{L\_\{p\}\\times N\\times d\_\{h\}\}, the attention block projects it into query, key, and value matrices:

\(15\)𝐐=𝐓𝐖Q,𝐊=𝐓𝐖K,𝐕=𝐓𝐖V,\\mathbf\{Q\}=\\mathbf\{T\}\\mathbf\{W\}\_\{Q\},\\quad\\mathbf\{K\}=\\mathbf\{T\}\\mathbf\{W\}\_\{K\},\\quad\\mathbf\{V\}=\\mathbf\{T\}\\mathbf\{W\}\_\{V\},where𝐖Q,𝐖K,𝐖V∈ℝdh×dh\\mathbf\{W\}\_\{Q\},\\mathbf\{W\}\_\{K\},\\mathbf\{W\}\_\{V\}\\in\\mathbb\{R\}^\{d\_\{h\}\\times d\_\{h\}\}\. The scaled dot\-product attention is

\(16\)Attention​\(𝐐,𝐊,𝐕\)=Softmax​\(𝐐𝐊⊤dk\)​𝐕\.\\mathrm\{Attention\}\(\\mathbf\{Q\},\\mathbf\{K\},\\mathbf\{V\}\)=\\mathrm\{Softmax\}\\left\(\\frac\{\\mathbf\{Q\}\\mathbf\{K\}^\{\\top\}\}\{\\sqrt\{d\_\{k\}\}\}\\right\)\\mathbf\{V\}\.Spatial attention is computed independently at each time step along the node dimension, so the softmax operates over theNNspatial nodes\. Temporal attention is computed independently at each node along the compressed temporal dimension, so the softmax operates over theLpL\_\{p\}temporal tokens\. Figure[2](https://arxiv.org/html/2607.13108#S4.F2)indicates that the attention block contains LayerNorm, FFN, and residual connections in addition to the attention operation\.

### 4\.5\.Prediction Head

After feature extraction and fusion via STKAN blocks, the prediction head aggregates temporal information across compressed time steps for each node and applies a linear projection to generate multi\-step forecasts\. Let𝐙\\mathbf\{Z\}denote the output representation after the fusion block\. The prediction head is expressed as:

\(17\)𝐘^t\+1:t\+H=fout​\(𝐙\),\\widehat\{\\mathbf\{Y\}\}\_\{t\+1:t\+H\}=f\_\{\\mathrm\{out\}\}\(\\mathbf\{Z\}\),where𝐘^t\+1:t\+H∈ℝH×N×Cout\\widehat\{\\mathbf\{Y\}\}\_\{t\+1:t\+H\}\\in\\mathbb\{R\}^\{H\\times N\\times C\_\{\\mathrm\{out\}\}\}\. The output functionfoutf\_\{\\mathrm\{out\}\}denotes the reshape and linear projection operations used to map the fused representation to the future horizon\.

## 5\.Experiment

### 5\.1\.Experimental Setup

Datasets\.We evaluate our model on five traffic forecasting datasets, including PEMS04, PEMS07, PEMS08, PEMS\-BAY and METR\-LA\. Following previous work, we divide the PEMS04, PEMS07 and PEMS08 dataset into training, validation, and test sets in a ratio of 6:2:2\. For the remaining datasets, we adopt a split ratio of 7:1:2\. Detailed statistics of these datasets are shown in Table[1](https://arxiv.org/html/2607.13108#S5.T1)\.

Table 1\.Summary of Five Spatio\-temporal BenchmarksBaselines\.We compare 11 representative baselines with our proposed STKAN\. \(i\)Non\-spatial modeling\-based:STID\(Shaoet al\.,[2022](https://arxiv.org/html/2607.13108#bib.bib41)\), which adopts identity spatio\-temporal embeddings and avoids explicit spatial dependency modeling\. \(ii\)Static spatial\-based methods:STGCN\(Yuet al\.,[2018](https://arxiv.org/html/2607.13108#bib.bib32)\), GWNet\(Wuet al\.,[2019](https://arxiv.org/html/2607.13108#bib.bib36)\), AGCRN\(Baiet al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib38)\), GMAN\(Zhenget al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib37)\), MTGNN\(Wuet al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib39)\)and STDN\(Caoet al\.,[2025](https://arxiv.org/html/2607.13108#bib.bib57)\)combine pre\-defined or learned static graph structures with temporal modeling modules\. \(iii\)Dynamic spatial\-based methods:STAEformer\(Liuet al\.,[2023](https://arxiv.org/html/2607.13108#bib.bib56)\)and STWave\(Fanget al\.,[2023](https://arxiv.org/html/2607.13108#bib.bib54)\)capture time\-varying spatial dependencies through adaptive or attention\-based mechanisms\. \(iv\)Spatio\-temporal decomposition\-based:StemGNN\(Caoet al\.,[2020](https://arxiv.org/html/2607.13108#bib.bib53)\)and STNorm\(Denget al\.,[2021](https://arxiv.org/html/2607.13108#bib.bib40)\)decompose spatio\-temporal series into separate components for modeling, focusing on disentangling spatial and temporal patterns\.

Settings\.In the experiments, we use the traffic flow of the last 12 time steps to predict the traffic flow of the next 12 time steps, and record the prediction performance of the 3rd, 6th, 12th steps and the average\. We set the Adam optimizer with an initial learning rate of 0\.002, where the learning rate follows a step\-wise decay strategy, and the batch size is set as 64\. During the training phase, we employ the early stopping strategy with tolerance 30 for 200 epochs\. For performance evaluation, we adopt three widely used metrics to quantify the accuracy of traffic forecasting results: Mean Absolute Error \(MAE\), Root Mean Squared Error \(RMSE\), and Mean Absolute Percentage Error \(MAPE\)\. All baseline methods are implemented within the BasicTS\(Shaoet al\.,[2024](https://arxiv.org/html/2607.13108#bib.bib58)\)framework to ensure consistent training pipelines and fair performance comparison\.

Table 2\.Forecasting performance on the five benchmark datasets\. We bold the best results and underline the second\-best results\.
### 5\.2\.Performance Comparisons

Table[2](https://arxiv.org/html/2607.13108#S5.T2)reports the forecasting results on the five benchmark datasets\. STKAN achieves the best average MAE and RMSE on PEMS04, although its average MAPE is slightly higher than that of STWave\. On PEMS07 and PEMS08, STKAN obtains the best average results across the three reported metrics\. On PEMS\-BAY, STKAN ties with STAEformer in average MAE and obtains the lowest average RMSE and MAPE\. On METR\-LA, STKAN remains competitive but does not outperform the strongest baseline in average MAE or RMSE\. Overall, the results indicate that STKAN is particularly effective on the evaluated traffic\-flow datasets, whereas its advantage is less evident on the METR\-LA speed dataset\.

The gains over the best\-performing baselines are generally modest\. Since the current evaluation reports single\-run results, small numerical differences should be interpreted cautiously\. These results demonstrate the competitiveness of the overall STKAN architecture\. The comparison with the MLP variant further suggests that the adopted TaylorKAN token mixers are useful within the evaluated configuration, although the current experiments do not isolate function\-approximator capacity as the sole source of the observed gains\.

### 5\.3\.Ablation Study

Effectiveness of KAN Modules\.To evaluate the role of KAN components within STKAN, we design three model variants:

- •MLPs: replacing the TaylorKAN token mixers with MLP layers in the evaluated implementation\.
- •w/o SKAN: removing the spatial block while retaining TKAN to test the importance of inter\-node mixing\.
- •w/o TKAN: removing the temporal block while retaining the spatial modeling component\.

As shown in Figure[3](https://arxiv.org/html/2607.13108#S5.F3), replacing the TaylorKAN token mixers with the evaluated MLP implementation increases the forecasting errors on PEMS04 and PEMS08\. This result suggests that the Taylor\-polynomial mappings contribute positively within the current architecture and hyperparameter setting\. Removing either SKAN or TKAN also degrades at least part of the reported performance, indicating that spatial\-group mixing and temporal mixing provide complementary information\. These observations are specific to the evaluated model configurations and should not be interpreted as establishing the general advantage of KANs over all parameter\-matched MLP alternatives\.

![Refer to caption](https://arxiv.org/html/2607.13108v1/ablation.png)Figure 3\.KAN modules ablation on PEMS04 and PEMS08\.Effectiveness of Attention Mechanisms\.To evaluate the role of attention in STKAN, we conduct ablation studies on PEMS04 and PEMS08 with the following variants:

- •w/o S\-Attention: disabling spatial attention while preserving temporal modeling\.
- •w/o T\-Attention: removing temporal attention while keeping spatial modeling\.
- •w/o ST\-Attention: removing both attention modules, leaving only KAN\-based token and channel mixing\.

Table[3](https://arxiv.org/html/2607.13108#S5.T3)shows that the complete model achieves the best result for every reported metric on PEMS04 and PEMS08\. However, the relative influence of spatial and temporal attention varies across metrics\. For example, removing temporal attention produces the largest MAE degradation on PEMS08, whereas removing spatial attention produces the largest RMSE degradation\. Removing both branches is therefore not uniformly the worst variant for every metric\. The results indicate that the two attention branches contribute differently across datasets and evaluation criteria\.

The ablation results suggest that attention provides an additional refinement over the KAN\-based mixer blocks, but the current experiments do not establish a strict ranking between the contributions of attention and KAN components\.

Table 3\.Ablation study of the attention block\.
### 5\.4\.Hyper\-parameter Study

Table[4](https://arxiv.org/html/2607.13108#S5.T4)reports the sensitivity of STKAN to the number of spatial groupsGG\. The results exhibit a non\-monotonic pattern: using either fewer or more groups than the selected value leads to slightly higher forecasting errors\. Among the tested settings,G=16G=16performs best on PEMS04, whileG=20G=20performs best on PEMS07\. This observation suggests that the group number should be selected according to the dataset rather than increased monotonically with the number of nodes\.

Table 4\.Hyper\-parameter study on varyingGGvalues\.
### 5\.5\.Case Study

Figure[4](https://arxiv.org/html/2607.13108#S5.F4)provides a qualitative visualization of the learned soft node\-to\-group assignment matrix on PEMS\-BAY\. Several groups exhibit relatively concentrated assignment patterns, and some corresponding hard assignments appear to cover spatially contiguous road segments\. This observation suggests that the learned grouping mechanism may capture corridor\-level regularities in the sensor network\. However, the visualization should be interpreted as qualitative evidence rather than a formal validation of geographical consistency or model interpretability\.

![Refer to caption](https://arxiv.org/html/2607.13108v1/group1.png)Figure 4\.Adaptive grouping matrices visualization on PEMS\-BAY\.

## 6\.Conclusion

In this work, we introduced STKAN, a spatio\-temporal forecasting architecture that incorporates Taylor\-polynomial KAN mappings into spatial and temporal token\-mixing modules\. The model combines learnable soft node grouping, group\-wise spatial mixing, temporal mixing, and spatial–temporal attention to model traffic dynamics\. Experiments on five benchmark datasets show that STKAN achieves competitive forecasting performance, with particularly strong results on the evaluated traffic\-flow datasets\. The comparison with the tested MLP variant suggests that TaylorKAN token mixers can provide a useful alternative nonlinear parameterization within the proposed architecture\. These findings indicate that function\-approximator design is a relevant component of spatio\-temporal modeling, alongside spatial and temporal architectural design\. Future work may examine parameter\-matched comparisons, computational efficiency, statistical variability, and broader spatio\-temporal applications\.

## References

- A\. A\. Aghaei \(2025\)Fkan: fractional kolmogorov–arnold networks with trainable jacobi basis functions\.Neurocomputing623,pp\. 129414\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p1.1)\.
- L\. Bai, L\. Yao, C\. Li, X\. Wang, and C\. Wang \(2020\)Adaptive graph convolutional recurrent network for traffic forecasting\.Advances in neural information processing systems33,pp\. 17804–17815\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- Z\. Bozorgasl and H\. Chen \(2024\)Wav\-kan: wavelet kolmogorov\-arnold networks\.arXiv preprint arXiv:2405\.12832\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p1.1)\.
- D\. Cao, Y\. Wang, J\. Duan, C\. Zhang, X\. Zhu, C\. Huang, Y\. Tong, B\. Xu, J\. Bai, J\. Tong,et al\.\(2020\)Spectral temporal graph neural network for multivariate time\-series forecasting\.Advances in neural information processing systems33,pp\. 17766–17778\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p2.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- L\. Cao, B\. Wang, G\. Jiang, Y\. Yu, and J\. Dong \(2025\)Spatiotemporal\-aware trend\-seasonality decomposition network for traffic flow forecasting\.InProceedings of the AAAI Conference on Artificial Intelligence,Vol\.39,pp\. 11463–11471\.Cited by:[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- L\. Chen, Q\. Zhong, X\. Xiao, Y\. Gao, P\. Jin, and C\. S\. Jensen \(2018\)Price\-and\-time\-aware dynamic ridesharing\.In2018 IEEE 34th international conference on data engineering \(ICDE\),pp\. 1061–1072\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p1.1)\.
- Y\. Chen, S\. Li, N\. Zhao, R\. Zheng, and Y\. Li \(2024\)BiLSTM\-kan: a time series\-based traffic flow forecasting model\.InProceedings of the 2024 13th International Conference on Computing and Pattern Recognition,pp\. 314–319\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p3.1)\.
- K\. Cho, B\. Van Merriënboer, D\. Bahdanau, and Y\. Bengio \(2014\)On the properties of neural machine translation: encoder\-decoder approaches\.arXiv preprint arXiv:1409\.1259\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- M\. Defferrard, X\. Bresson, and P\. Vandergheynst \(2016\)Convolutional neural networks on graphs with fast localized spectral filtering\.Advances in neural information processing systems29\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- J\. Deng, X\. Chen, R\. Jiang, X\. Song, and I\. W\. Tsang \(2021\)St\-norm: spatial and temporal normalization for multi\-variate time series forecasting\.InProceedings of the 27th ACM SIGKDD conference on knowledge discovery & data mining,pp\. 269–278\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- J\. Deng, F\. Ye, D\. Yin, X\. Song, I\. Tsang, and H\. Xiong \(2024\)Parsimony or capability? decomposition delivers both in long\-term time series forecasting\.Advances in Neural Information Processing Systems37,pp\. 66687–66712\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p1.1)\.
- Y\. Fang, Y\. Qin, H\. Luo, F\. Zhao, B\. Xu, L\. Zeng, and C\. Wang \(2023\)When spatio\-temporal meet wavelets: disentangled traffic forecasting via efficient spectral graph attention networks\.In2023 IEEE 39th International Conference on Data Engineering \(ICDE\),pp\. 517–529\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p2.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- J\. Han, H\. Liu, H\. Xiong, and J\. Yang \(2022\)Semi\-supervised air quality forecasting via self\-supervised hierarchical graph neural network\.IEEE Transactions on Knowledge and Data Engineering35\(5\),pp\. 5230–5243\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p1.1)\.
- X\. Han, X\. Zhang, Y\. Wu, Z\. Zhang, and Z\. Wu \(2024\)Are kans effective for multivariate time series forecasting?\.arXiv preprint arXiv:2408\.11306\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p3.1)\.
- S\. Huang, Z\. Zhao, C\. Li, and L\. Bai \(2025\)Timekan: kan\-based frequency decomposition learning architecture for long\-term time series forecasting\.arXiv preprint arXiv:2502\.06910\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p3.1)\.
- G\. Jin, Y\. Liang, Y\. Fang, Z\. Shao, J\. Huang, J\. Zhang, and Y\. Zheng \(2023\)Spatio\-temporal graph neural networks for predictive learning in urban computing: a survey\.IEEE transactions on knowledge and data engineering36\(10\),pp\. 5388–5408\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p2.1)\.
- T\. N\. Kipf and M\. Welling \(2016\)Semi\-supervised classification with graph convolutional networks\.arXiv preprint arXiv:1609\.02907\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- G\. Lai, W\. Chang, Y\. Yang, and H\. Liu \(2018\)Modeling long\-and short\-term temporal patterns with deep neural networks\.InThe 41st international ACM SIGIR conference on research & development in information retrieval,pp\. 95–104\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- F\. Li, J\. Feng, H\. Yan, G\. Jin, F\. Yang, F\. Sun, D\. Jin, and Y\. Li \(2023\)Dynamic graph convolutional recurrent network for traffic prediction: benchmark and solution\.ACM Transactions on Knowledge Discovery from Data17\(1\),pp\. 1–21\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- Y\. Li, R\. Yu, C\. Shahabi, and Y\. Liu \(2017\)Diffusion convolutional recurrent neural network: data\-driven traffic forecasting\.arXiv preprint arXiv:1707\.01926\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- Z\. Li \(2024\)Kolmogorov\-arnold networks are radial basis function networks\.arXiv preprint arXiv:2405\.06721\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p1.1)\.
- Z\. Liang, R\. An, W\. Fan, Y\. Rao, and Y\. Liang \(2025\)ITFKAN: interpretable time series forecasting with kolmogorov\-arnold network\.arXiv preprint arXiv:2504\.16432\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p3.1)\.
- H\. Lin, Z\. Gao, Y\. Xu, L\. Wu, L\. Li, and S\. Z\. Li \(2022\)Conditional local convolution for spatio\-temporal meteorological forecasting\.InProceedings of the AAAI conference on artificial intelligence,Vol\.36,pp\. 7470–7478\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p1.1)\.
- H\. Liu, Z\. Dong, R\. Jiang, J\. Deng, J\. Deng, Q\. Chen, and X\. Song \(2023\)Spatio\-temporal adaptive embedding makes vanilla transformer sota for traffic forecasting\.InProceedings of the 32nd ACM international conference on information and knowledge management,pp\. 4125–4129\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p2.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- Z\. Liu, Y\. Wang, S\. Vaidya, F\. Ruehle, J\. Halverson, M\. Soljačić, T\. Y\. Hou, and M\. Tegmark \(2024\)Kan: kolmogorov\-arnold networks\.arXiv preprint arXiv:2404\.19756\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p1.1)\.
- Z\. Pan, Y\. Liang, W\. Wang, Y\. Yu, Y\. Zheng, and J\. Zhang \(2019\)Urban traffic prediction from spatio\-temporal data using deep meta learning\.InProceedings of the 25th ACM SIGKDD international conference on knowledge discovery & data mining,pp\. 1720–1730\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- Z\. Shao, F\. Wang, Y\. Xu, W\. Wei, C\. Yu, Z\. Zhang, D\. Yao, T\. Sun, G\. Jin, X\. Cao,et al\.\(2024\)Exploring progress in multivariate time series forecasting: comprehensive benchmarking and heterogeneity analysis\.IEEE Transactions on Knowledge and Data Engineering37\(1\),pp\. 291–305\.Cited by:[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p3.1)\.
- Z\. Shao, Z\. Zhang, F\. Wang, W\. Wei, and Y\. Xu \(2022\)Spatial\-temporal identity: a simple yet effective baseline for multivariate time series forecasting\.InProceedings of the 31st ACM international conference on information & knowledge management,pp\. 4454–4458\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p2.1),[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- X\. Shi, Z\. Chen, H\. Wang, D\. Yeung, W\. Wong, and W\. Woo \(2015\)Convolutional lstm network: a machine learning approach for precipitation nowcasting\.Advances in neural information processing systems28\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- C\. J\. Vaca\-Rubio, L\. Blanco, R\. Pereira, and M\. Caus \(2024\)Kolmogorov\-arnold networks \(kans\) for time series analysis\.arXiv preprint arXiv:2405\.08790\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p3.1)\.
- J\. Wang, J\. Ji, Z\. Jiang, and L\. Sun \(2022\)Traffic flow prediction based on spatiotemporal potential energy fields\.IEEE Transactions on Knowledge and Data Engineering35\(9\),pp\. 9073–9087\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p1.1)\.
- S\. Wang, J\. Cao, and S\. Y\. Philip \(2020\)Deep learning for spatio\-temporal data mining: a survey\.IEEE transactions on knowledge and data engineering34\(8\),pp\. 3681–3700\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p1.1)\.
- Z\. Wu, S\. Pan, G\. Long, J\. Jiang, X\. Chang, and C\. Zhang \(2020\)Connecting the dots: multivariate time series forecasting with graph neural networks\.InProceedings of the 26th ACM SIGKDD international conference on knowledge discovery & data mining,pp\. 753–763\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- Z\. Wu, S\. Pan, G\. Long, J\. Jiang, and C\. Zhang \(2019\)Graph wavenet for deep spatio\-temporal graph modeling\.arXiv preprint arXiv:1906\.00121\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p2.1),[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- K\. Xu, L\. Chen, and S\. Wang \(2024\)Kolmogorov\-arnold networks for time series: bridging predictive power and interpretability\.arXiv preprint arXiv:2406\.02496\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p3.1)\.
- H\. Yao, F\. Wu, J\. Ke, X\. Tang, Y\. Jia, S\. Lu, P\. Gong, J\. Ye, and Z\. Li \(2018\)Deep multi\-view spatio\-temporal network for taxi demand prediction\.InProceedings of the AAAI conference on artificial intelligence,Vol\.32\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1)\.
- B\. Yu, H\. Yin, and Z\. Zhu \(2018\)Spatio\-temporal graph convolutional networks: a deep learning framework for traffic forecasting\.InProceedings of the Twenty\-Seventh International Joint Conference on Artificial Intelligence,pp\. 3634–3640\.Cited by:[§1](https://arxiv.org/html/2607.13108#S1.p2.1),[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.
- S\. Yu, Z\. Chen, Z\. Yang, J\. Gu, B\. Feng, and Q\. Sun \(2025\)Exploring kolmogorov\-arnold networks for realistic image sharpness assessment\.InICASSP 2025\-2025 IEEE International Conference on Acoustics, Speech and Signal Processing \(ICASSP\),pp\. 1–5\.Cited by:[§2\.2](https://arxiv.org/html/2607.13108#S2.SS2.p1.1)\.
- C\. Zheng, X\. Fan, C\. Wang, and J\. Qi \(2020\)Gman: a graph multi\-attention network for traffic prediction\.InProceedings of the AAAI conference on artificial intelligence,Vol\.34,pp\. 1234–1241\.Cited by:[§2\.1](https://arxiv.org/html/2607.13108#S2.SS1.p1.1),[§5\.1](https://arxiv.org/html/2607.13108#S5.SS1.p2.1)\.

Similar Articles

Geometric Kolmogorov--Arnold Network (GeoKAN)

arXiv cs.LG

This paper introduces Geometric Kolmogorov-Arnold Networks (GeoKAN), a family of geometry-aware models that learn Riemannian metrics to adapt coordinates for improved function approximation and physics-informed learning.

Geometry-Aware R-Structured Kolmogorov-Arnold Networks

arXiv cs.LG

Proposes Geometry-aware R-Structured KAN (GRS-KAN), a hybrid neural architecture that integrates R-functions into KAN to encode geometric and logical constraints, achieving up to 67% RMSE reduction on regression benchmarks with discontinuities.

Nested Spatio-Temporal Time Series Forecasting

arXiv cs.LG

This paper proposes a nested spatiotemporal forecasting framework that uses spectral clustering to construct semantically coherent macro-level regions, which provide top-down guidance for fine-grained micro-level predictions. Experiments on high-dimensional datasets show consistent improvements over state-of-the-art baselines.