KAN-Transformer Fusion with Mixture of Experts for Temporal Imputation of Spatiotemporal Air Pollution Data ()
1. Introduction
With the acceleration of global industrialization and urbanization, air pollution has become an escalating environmental challenge requiring urgent attention. In recent years, concentrations of key air pollutants—such as fine particulate matter (PM2.5), sulphur dioxide (SO2), and nitrogen oxides (NOx)—have shown upward trends in several urban areas across China. Multiple monitoring datasets indicate that, in certain regions, pollutant levels have exceeded the air quality guideline values recommended by the World Health Organization (WHO) [1]. This severe air pollution poses a direct threat to public health, increasing the incidence of chronic conditions including respiratory and cardiovascular diseases [2], while also exerting broader impacts on ecosystems, such as contributing to climate change and accelerating biodiversity loss. Against this backdrop, the accurate characterization, understanding, and modelling of air pollution have garnered increasing attention in the scientific community.
To comprehensively understand air quality, China has built a modern environmental monitoring network covering major cities and key regions for real-time dynamic observation of air pollutants. However, collected datasets often have missing values due to instrument malfunction, extreme weather, data transmission interruptions, and human errors [3]. Notably, data come from geographically distributed monitoring stations with substantial spatial heterogeneity, resulting in temporal and spatial missing data that impair analysis-reliable data integrity and continuity. Missing data severely limits accurate regional pollution assessment [4], hinders source attribution, early warning, forecasting, and environmental policy evaluation. Particularly for high-precision air quality prediction models, incomplete data causes insufficient or biased training, undermining output reliability [5]. Thus, effective data imputation is indispensable to reconstruct continuously missing spatio-temporal pollutant data.
Missing data in environmental data is generally unavoidable, and correct handling and filling of these gaps can enhance downstream research. Currently, such methods are divided into traditional methods and advanced methods based on deep learning. Traditional missing value processing methods can be broadly categorized into two main types: data deletion methods and statistical feature-based imputation techniques. However, these approaches exhibit substantial limitations when applied to air quality data, which often possess complex spatio-temporal characteristics. Although data deletion is straightforward to implement, it disrupts the continuity of time series, potentially leading the model to erroneously interpret discontinuous observations as continuous sequences, thereby introducing systematic bias [6]. Statistical imputation methods (e.g., mean/median imputation, linear imputation) consider only the statistical properties of individual variables, while neglecting key temporal dependencies such as trends and seasonality, as well as inter-variable relationships [7]. Moreover, these methods fail to capture spatial correlations, despite the fact that significant spatial autocorrelation typically exists between neighboring monitoring stations in air quality datasets.
In recent years, deep learning has shown great promise in missing value interpolation. Methods based on RNNs [8], CNNs [9], Transformers [10], and GANs [11] have achieved notable success in temporal data imputation. Transformers have become a focus due to their self-attention mechanism for long-range dependencies and inter-variable interactions, but they do not always outperform traditional models [12], as their permutation-invariant structure may lose temporal order information. To address this, improvements like RevIN [13] enhance adaptability via adaptive normalization. Meanwhile, the emerging KAN [14], based on function approximation theory, effectively captures temporal dynamics. The complementarity of Transformers (long-range dependence) and KAN (high-precision function learning) enables a more expressive spatio-temporal interpolation framework. Based on this, this paper proposes the KT model (KAN-Transformer) for air quality missing value imputation, integrating KAN and Transformer to reconstruct data from temporal dependence, inter-pollutant correlation, and sensor spatial distribution. Targeting air quality data’s unique characteristics (missing values, spatio-temporal structure, nonlinearity, high-dimensional dependencies), a targeted framework is designed: 1) Transformer for global variable dependencies and temporal feature extraction; 2) MoE mechanism integrated into KAN to handle nonlinear, multiscale pollution evolution; 3) CSPM to dynamically model variables independently or jointly based on spatial heterogeneity; 4) RevIN to improve generalization by stabilizing feature distributions. These components form a cohesive architecture for complex air quality imputation. The main contributions of this paper are as follows:
1) This is the first study, to the best of our knowledge, to integrate the Kolmogorov-Arnold Network (KAN) with a Transformer-based architecture for missing value imputation in air quality data. The proposed model jointly captures temporal, inter-variable, and spatial dependencies, and demonstrates competitive performance.
2) A hybrid expert mechanism, combined with Reversible Instance Normalization (RevIN), is introduced to guide the model in learning nonlinear transformation patterns among pollutants, while enhancing its adaptability to distributional shifts in real-world datasets.
3) Using a channel selection mechanism based on Spearman’s correlation coefficient, the channel input method is adaptively selected according to spatial correlation.
2. Related Work
In recent years, various imputation methods have been applied to missing values in air quality data, with the choice of methods often having a direct impact on the validity of subsequent analyses.
Traditional imputation approaches are primarily based on statistical principles, estimating missing values by analyzing the distributional characteristics of the data or the relationships among variables. Common techniques include mean imputation, median imputation, and mode imputation. These methods are easy to implement but typically focus solely on global statistical properties, thereby neglecting temporal dependencies among data points and latent correlation structures between variables within the time series.
With the advancement of deep learning technologies, significant breakthroughs have been achieved in the task of missing value imputation. Compared with traditional approaches, deep learning models exhibit superior nonlinear modeling capabilities and representation learning capacities, enabling them to better accommodate complex data structures and distributional shifts [15] [16]. Representative frameworks include models based on Recurrent Neural Networks (abbr. RNNs) [17], Convolutional Neural Networks (abbr. CNNs) [18], Graph Neural Networks (abbr. GNNs) [19] [20], and attention mechanisms [21]. In the context of time series imputation, the SAITS model [20] simulates real-world missing scenarios through a diagonal masking strategy and enhances imputation performance by integrating it with a reconstruction-based learning framework. ImputeFormer [22] combines a low-rank assumption with the self-attention mechanism of the Transformer architecture to efficiently impute high-dimensional spatiotemporal data. Meanwhile, multimodel fusion strategies have gained increasing attention due to their ability to integrate the strengths of different architectures, enabling more comprehensive extraction of the structural features in the data. For instance, TimesNet [23] introduces the chunked time-domain mixing (abbr. TSMix) mechanism, which fuses local convolution with global modeling capabilities to effectively capture multiscale temporal dependencies. ST-SILM [24] leverages a combination of Single Exponential Smoothing (abbr. SES) and Inverse Distance Weighting (abbr. IDW) with Long Short-Term Memory (abbr. LSTM) networks to achieve spatiotemporal correlation fusion. DSSE further improves imputation accuracy and robustness by integrating the global context modeling ability of the Transformer with the temporal feature extraction capability of Bidirectional LSTM (abbr. Bi-LSTM).
It is noteworthy that the Kolmogorov–Arnold Network (KAN), as a novel neural network architecture, has gradually demonstrated distinct advantages in time series modeling due to its highly flexible function approximation capabilities. For instance, TimeKAN [25] addresses the problem of multifrequency mixing by decomposing sequences into different frequency components, enabling more effective temporal modeling. WormKAN [26] introduces a concept-aware module that can efficiently detect and learn complex patterns such as concept drift. In addition, several studies have explored hybrid architectures that integrate Convolutional Neural Networks, Bidirectional Long Short-Term Memory networks (abbr. Bi-LSTMs), and KANs for tasks such as greenhouse gas emission prediction, demonstrating strong performance in capturing nonlinear dynamic changes [27]. Although KAN has shown promising results across various time series prediction tasks, its potential in missing value imputation has not yet been thoroughly explored yet. Therefore, investigating the applicability of KAN in this domain—particularly through its integration with attention mechanisms and hybrid expert architectures—has significant theoretical and practical value.
3. Materials and Methods
In this section, we first present a real-world air quality monitoring dataset. We then formulate the fundamental concepts and mathematical representation of the air quality data interpolation task by analyzing the inherent characteristics of the dataset. Finally, we present the overall architecture and functions of the proposed model in Section 3.5, and provide a detailed description of each module with the corresponding mathematical derivations.
3.1. Datasets
Experimental data were from the National Urban Air Quality Real-Time Release Platform (https://air.cnemc.cn:18007/), covering Jan 1, 2023, to Oct 12, 2024, including key pollutants (PM2.5, PM10, SO2) as a multivariate time series with temporal dynamics, inter-variable correlations, and spatial heterogeneity (see Figure 1(a)). Analysis of Beijing’s air quality data shows varying missing data across stations and pollutants (see Figure 1(a), Figure 1(b)). “Missing amount” refers to unavailable hours, “missing rate” to the proportion of missing hours. Most stations have an annual missing rate of 0.4% - 1.2% (up to 8.5% for individual stations), with an average of over 3% across pollutants. The dataset is split: test set (Mar, Jun, Sep, Dec), validation set (Apr, Jul), training set (remaining months). To evaluate the model, artificial missing masks are applied to observed data to simulate different missing patterns and rates, ensuring reliable assessment of imputation accuracy and robustness.
![]()
(a) Spatial distribution and missing rate of monitoring stations.
(b) Proportion of missing air quality monitoring indicators.
Figure 1. Missing data by site and by pollutant. (a) Spatial distribution and missing rate of monitoring stations: with green to red indicating a gradual increase in the rate of missing data (This map is produced based on the standard map with review number GS(2019)3333 downloaded from the Standard Map Service website of the Ministry of Natural Resources, and the map has not been modified). (b) Proportion of missing.
3.2. Problem Formulation
Based on the characteristics of air quality monitoring data, we provide the following formal definition. Given a multivariate time series
, where
represents observations at
time steps. Each observation
at time step
contains
features corresponding to the pollutant concentrations measured at various stations, denoted as
. Due to circumstances beyond our control in the real-world, the dataset contains missing values, and the objective is to accurately estimate these missing entries. To represent the pattern of missingness in
, we introduce a binary mask matrix
,
, define
and each
denotes the mask vector at time step
. The entries in
are defined as:
(1)
This mask is used to guide the model in distinguishing between observed and unobserved entries during the imputation process. Where the value of
is 0 if the value of the dth variable
is missing, and the value of
is 1 if
is observable.
3.3. Masking Strategy
Since the true values of the original missing data are unavailable, evaluating model performance requires an artificial masking scheme during training to simulate realistic missing patterns as closely as possible. We assume that the data distribution before and after missingness remains approximately consistent, allowing the estimation accuracy on artificially masked entries to serve as a proxy for evaluating the model’s ability to recover the original missing values. The formula is expressed as follows:
(2)
In this equation,
represents the original observation data,
represents the estimated values of the model,
represents the artificial masking matrix, and
represents a defined loss function.
To differentiate artificial missingness from natural missingness, we define the artificial masking task as follows:
(3)
Let
denote a binary mask matrix, where
marks artificially masked entries and
indicates retained data.
3.4. Methodology
The Kolmogorov-Arnold Network (abbr. KAN) is a novel neural network architecture inspired by the Kolmogorov-Arnold representation theorem, which demonstrates that any continuous multivariate function can be decomposed into a finite composition of univariate functions. This theoretical foundation is expressed mathematically as follows:
(4)
where both
and
are nonlinear functions.
Based on this theorem, complex dependencies in high-dimensional input data can be effectively represented by compositions of nonlinear univariate functions. Recent advances in lightweight time series architectures have shown a clear shift from linear models to MLP-based designs, highlighting the effectiveness of nonlinear modeling. While MLPs rely on fixed activation functions at the nodes, Kolmogorov-Arnold Networks (abbr. KANs) introduce learnable activations along the edges, which grants them richer nonlinear expressivity. This edge-based flexibility makes KANs particularly well-suited for capturing intricate dependencies in high-dimensional data. It positions them as a compelling alternative to MLPs in a wide range of applications. The composite mapping of a KAN with
layers can be formally expressed as follows:
(5)
where
corresponds to the function matrix of layer
and each matrix element is a trainable univariate function. The operator
denotes function composition across layers, applied in order from
to
.
KAN has demonstrated significant advantages in fields such as time series prediction and pattern recognition. In this study, it is first applied to the task of imputing missing values in spatiotemporal data, mainly based on the following considerations: 1) air quality data has complex spatiotemporal correlation, which requires strong nonlinear modeling capability; 2) the missing patterns of monitoring data often show non-uniform distribution, which requires the model to have the ability to capture local features.
3.5. Model Architecture
In response to the multidimensional spatiotemporal characteristics of air quality data, this study proposes a KAN-Transformer hybrid architecture, in short for KT designed to achieve high-precision imputation of missing values. KT simultaneously captures temporal dependencies, inter-variable correlations, and spatial heterogeneity by integrating three complementary modules: a hybrid expert KAN layer, a variable-wise Transformer module, and a channel-wise attention selection mechanism (ASPM). As illustrated in Figure 2, these components collectively form an end-to-end framework for spatio-temporal feature representation and reconstruction. To further enhance model robustness in non-stationary environments, a Reversible Instance Normalization (abbr. RevIN) strategy is applied prior to model training. This preprocessing step reduces the adverse effects of distributional shifts across variables and time periods, thereby enabling the model to operate on a more stable and homogeneous input space.
![]()
Figure 2. The overall architecture of our proposed model, as well as the (a) Hybrid expert temporal encoding, (b) Adaptive Dimension-aware Transformer, and (c) Channel-wise spatial attention.
3.5.1. Reversible Temporal Normalization
To address the pervasive non-stationarity in air quality data, this study introduces a Reversible Instance Normalization (abbr. RevIN) strategy prior to model training. This mechanism initially transforms the input sequences into a stationary distribution via instance normalization, effectively mitigating the impact of distributional shifts on model learning and enabling the network to better capture intrinsic temporal dependencies. During the output phase, the nonstationary components removed during normalization are seamlessly reintegrated through an inverse transformation, ensuring that the final predictions remain consistent with the original data distribution. As illustrated in Figure 3, this approach not only preserves the statistical characteristics of the raw data but also enhances the model’s capacity to learn meaningful temporal patterns.
Figure 3. RevIN structure.
We combine the original data
with the generated mask matrix
to obtain
as an input and first calculate the mean average and variance of the inputs and then normalise them:
(6)
(7)
(8)
where
and
are both learnable parameters,
is a constant, and
is the time step length.
Subsequently, the normalised data
is fed into the hybrid expert KAN model and the model output is inversely normalised mapped to the original distribution space to obtain the output
. This is done as follows:
(9)
Finally, the result of inverse normalization is linearly mapped back to the original feature dimensions and the missing is replaced with the resulting estimate to obtain the output
of the first module:
(10)
(11)
3.5.2. Hybrid Expert Temporal Encoding
As the initial processing layer of the model, the hybrid expert KAN layer effectively captures the complex distributions of pollutant concentrations through a collaborative multi-expert mechanism. This module, built upon the Kolmogoro v-Arnold Network hybrid expert framework, is specifically tailored to extract multilevel temporal features from preprocessed input data. Unlike traditional multilayer perceptrons (MLPs) that rely on fixed functional mappings, the KAN network leverages a combination of learnable univariate basis functions, enabling superior modeling of intricate nonlinear relationships. Inspired by [27], this study designs a cooperative learning architecture comprising four distinct types of expert functions (see Table 1) to accommodate the multiscale dynamics and heterogeneity inherent in air quality monitoring data.
The model consists of a KAN network with a hybrid layer of experts, which internally contains a gated network that assigns the KAN layer to the corresponding variables based on the read time series features, and the KAN internally contains different spline functions, i.e., each expert is responsible for processing the input data differently. Therefore, the KAN hybrid expert network with
experts can be represented as:
(12)
where
is the gating network.
is the KAN.
is the output. This hybrid expert structure can effectively address the heterogeneity of data distribution and learn features from different variables for better missing value estimation.
Each expert applies a different function for fitting, respectively.
The gating network is a critical component of the architecture, as it calculates the weights for each expert based on the input data. The Softmax function is then applied to normalize these weights into a probability distribution. This enables the dynamic selection of the most suitable expert network for computation, based on the varying characteristics of the data.
Table 1. Formulas for different fitting functions and their meanings.
Expert Name |
Function Form |
Application Scenario |
Mexican Hat Wavelet Expert |
|
It is sensitive to sudden pulses, amplifies local signal features, and provides a good fit to sudden increases and transient spikes in pollutant levels. |
Morlet Wavelet Expert |
|
Suitable for analyzing the periodic variations in air pollutant concentrations. |
Derivative of Gaussian (DOG) Wavelet Expert |
|
Rapid increase or decrease in concentration of applicable pollutant. |
B-spline |
|
Rapid increase or decrease in concentration of applicable pollutant. |
(13)
To learn the relevant features of the task more efficiently, the network adds Gaussian noise to the input time series during training and selects only the top
best matching experts for computation.
(14)
(15)
where
denotes standard normalization and
is Gaussian noise.
retains the top
largest elements of the input vector
. This ensures that the Softmax operation is computed only over the top
values, effectively enforcing sparsity and enabling selective attention.
The architecture achieves adaptive synergy among experts through a dynamic gating mechanism, which maintains the specialized feature extraction capability of each expert function and improves the generalization performance of the model through integrated learning. This design provides a new solution for the time-series feature analysis of complex pollution processes.
3.5.3. Adaptive Dimension-Aware Transformer
Traditional Transformer-based multivariate time series methods usually embed all variables’ data at one time step into a single representation [28], which captures temporal dependencies but assumes tight coupling of all variables. However, air quality datasets often have temporal misalignment among pollutants due to lagged pollution event responses, leading to mutual interference and degraded feature correlation capture when embedding multiple variables into one representation. Recent studies [29] show that input dimension transposition (treating each variable as an independent token and modeling the feature dimension explicitly) can mitigate this issue. Inspired by this, we propose a feature-oriented transposed Transformer, shifting from time-domain to feature-centric tokenization. Specifically, the input tensor is transposed to form an independent token sequence for each pollutant, projected to hidden dimensions via linear mapping to get refined representation
, and then attention mechanisms and feed-forward networks are applied along the feature dimension to capture inter-variable correlations while preserving temporal structure. This enhances global multivariate pattern extraction, facilitating more accurate missing value imputation.
(16)
(17)
(18)
where
is a learnable parameter.
Secondly, the multi-head self-attention of the mask is utilized with the N stacked layers of the feed-forward neural network to operate self-attention on the inputs to effectively capture the feature correlations, thus improving the representation and learning performance. To further learn the feature representation, we apply two linear projections and add ReLU activation between them to obtain
, as deeper structures can generally learn better.
(19)
(20)
where MHA denotes the multi-head self-attention mechanism. Finally, the learned feature association information is reintegrated with the original temporal structure through an inversion operation, the output of which is:
(21)
(22)
3.5.4. Channel-Wise Spatial Attention
The previous inverse Transformer adopted a fully channel-independent approach (processing each variable separately). While it effectively eliminates inter-variable interference during normalization and reduces noise, it sacrifices inter-variable dependency modeling, which is critical for capturing complex pollutant interactions. In air quality data, variable interdependencies reflect important spatial relationships and co-evolving pollution dynamics across monitoring sites; ignoring them limits the model’s ability to utilize data structure. Thus, we propose a selective channel-based spatial attention mechanism for spatio-temporal feature extraction. Unlike the inverse Transformer’s pure channel independence, it adaptively chooses channel-independent or mixing strategies by assessing inter-variable correlation via Spearman correlation coefficient. Its motivation is twofold: 1) Preserve noise robustness (maintain channel independence for weak correlations to avoid noise and overfitting); 2) Enhance information sharing (adopt channel mixing for strong correlations to improve spatial dependency modeling).
Patching Local embedding starts with transforming input
to
, then slicing in the time dimension to get
(
= window size,
= number of windows). Each variable sequence
is split into
; with step
,
. The split data is mapped to hidden dimensions to obtain
.
Channel Selection Channel selection is performed using Spearman correlation coefficient (a non-parametric metric for monotonic relationships, calculated via rank to reduce outlier impact). We first extract training dataset
, compute Spearman coefficients
between
and
(
), then count correlated sequences under different thresholds:
(high correlation) and
(non-negative correlation).
(23)
(24)
(25)
where
is a human-set threshold.
is an indicator function that takes the value of 1 when the condition is met and 0 otherwise. Then, the number of high correlations and the number of non-negative correlations are
and
, respectively. And calculate the high correlation ratio, defined as
. Selection of channel selection strategies based on high correlation ratios:
(26)
where if
we adopt a channel mixing strategy and vice versa a channel independent strategy.
The embedding of the data in front of the input encoder is different for different channel choices. If channel independence is chosen, the different channels are independent of each other, i.e., the model processes each channel separately without sharing or exchanging information, and the data is reshaped as:
, which will be obtained by encoding them separately in the encoder:
(27)
Instead the channel data can interact with each other by reshaping the data as:
, the variable dimensions are preserved, at which point multiple channels are fed together into the encoder, at which point the self-attention in the Encoder allows the information between different channels to interact with each other:
(28)
In order to map the results of the above spatial dimensionality processing to the target interpolation format, we exchange the dimensions of the encoder outputs to obtain
. If the channels are independent, each variable channel is interpolated individually, and different variables are mapped using separate linear layers with linear transformations, and different channels do not share weights. Finally, we stack all channels; the process is as follows:
(29)
(30)
(31)
where
is a channel
independent weight matrix and
is a bias term.
If the channels are mixed, all channels go through the same linear layer together and all parameters are shared, denoted as follows:
(32)
(33)
where
.
Finally, the final imputation results are generated:
(34)
4. Experiments
4.1. Experimental Setup
All experiments are conducted on workstations equipped with NVIDIA RTX 3060 GPUs. The model is trained using the Adam optimizer implemented in PyTorch. The learning rate was set to 0.001, and the hyperparameters are detailed in Table 2. To evaluate the model’s performance, three commonly used metrics are employed to quantify the performance: mean absolute error (MAE), mean relative error (MRE), and root mean square error (RMSE), with their respective formulas defined as follows:
(35)
(36)
(37)
where
denotes the estimated value of the air quality data,
denotes the initial value of the air quality data,
denotes the mask matrix,
denotes the variable, and
denotes the time.
Table 2. Hyper-parameters for model training.
Hyper-params |
Air-Quality |
Batch_size |
64 |
d_model |
256 |
n_head |
4 |
n_group |
2 |
4.2. Baseline Methodology
To evaluate the performance of the proposed model, we compare it against several deep learning-based models (i.e., SAITS, Transformer, TimesNet, GP-VAE, BRITS, MRNN, ImputeFormer, ITransformer, TEFN, TimeMixer) and statistical methods (i.e., Mean and Median) with similar methodological backgrounds. All methods used the same masking strategy, data partitioning approach, and evaluation protocol. 1) Mean 2) Median 3) SAITS [30]: A diagonal mask-based bi-level self-attention framework for missing value imputation. 4) Transformer [31]: Imputes missing values using the encoder module of the Transformer architecture. 5) TimesNet [23]: A method combining CNN and self-attention mechanisms that transforms time series into 2D tensors and applies 2D convolutional kernels for feature extraction. 6) GP-VAE [32]: Integrates the principles of variational autoencoders and Gaussian processes by embedding multivariate time series with missing values into a low-dimensional latent space, where Gaussian processes model the temporal dynamics. 7) BRITS [33]: Employs bidirectional recurrent neural networks (RNNs), leveraging both historical information and feature regression for missing value estimation. 8) MRNN [34]: Utilizes multidirectional RNNs to estimate missing values in time series data. 9) ImputeFormer [22]: A spatio-temporal imputation model that combines low-rank induction with Transformer-based attention mechanisms. 10) ITransformer [29]: Performs data imputation by learning feature dependencies through a feature-wise attention mechanism applied to transposed time series data. 11) TEFN [35]: Imputation missing values using a lightweight neural network model based on evidence theory and information fusion. 12) TimeMixer [36]: Captures complex temporal patterns for interpolation through a multi-scale, multi-resolution attention mechanism.
4.3. Imputation Performance
To assess the generalizability of the proposed model across diverse scenarios, we compare its performance against various baseline methods under different missing rates and missing patterns. The imputation results, evaluated using mean absolute error (MAE), mean relative error (MRE), and root mean square error (RMSE), are summarized as follows.
4.3.1. Imputation under MCAR
In real-world settings, sensor aging or failure often causes isolated missing values. We simulate such point-wise missingness with rates of 1%, 5%, and 10%. As shown in Table 3, statistical methods (Mean, Median) perform worst and degrade sharply as missingness increases, failing to capture air quality data dynamics. Deep learning methods are more stable. RNN-based models work reasonably at low missing rates but suffer from error accumulation under higher missingness. Among Transformer-based approaches, ImputeFormer and SAITS leverage self-attention for global temporal modeling; however, SAITS underperforms at 1% due to its focus on long sequences. TimesNet and TimeMixer show limited adaptability, while TEFN—designed for prediction—struggles with precise local reconstruction. GP-VAE exhibits robustness at high missing rates but still incurs notable errors. KT consistently outperforms all baselines, thanks to KAN’s strong nonlinear modeling of complex pollutant patterns and its integration of global attention without error accumulation. This advantage is especially pronounced under high missingness. For a broader view, Figure 4 illustrates performance trends across missing rates from 1% to 70%.
Table 3. In the MCAR missing data scenario, the model imputation results show missing rates of 1%, 5%, and 10%. The best results are indicated in bold, and the second-best results are indicated with an underline.
Model/Missing Rate |
1% |
5% |
10% |
Metric |
MAE/RMSE/MRE |
MAE/RMSE/MRE |
MAE/RMSE/MRE |
Mean |
0.8633/1.3401/1.0293 |
0.8723/1.4292/1.0277 |
0.8742/1.3941/1.0314 |
Median |
0.8164/1.3824/0.9734 |
0.8231/1.4350/0.9711 |
0.8301/1.4704/0.9789 |
Transformer |
0.2663/0.4321/0.2812 |
0.2763/0.4685/0.2901 |
0.2901/0.4871/0.3071 |
MRNN |
0.3684/0.7114/0.4408 |
0.3868/0.7846/0.4546 |
0.3906/0.7677/0.4689 |
TimeMixer |
0.3721/0.7901/0.5901 |
0.3918/0.8637/0.6117 |
0.4212/0.7306/0.5990 |
TEFN |
0.3911/0.7308/0.4907 |
0.4075/0.7558/0.5000 |
0.4209/0.7708/0.5204 |
GP-VAE |
0.3820/0.6142/0.4604 |
0.4151/0.7203/0.4970 |
0.4288/0.6798/0.5086 |
TimesNet |
0.3414/0.6521/0.4115 |
0.3546/0.7555/0.4228 |
0.3580/0.7622/0.4201 |
SAITS |
0.2272/0.4124/0.2763 |
0.2356/0.4584/0.2791 |
0.2353/0.4630/0.2727 |
Imputeformer |
0.1785/0.4306/0.2152 |
0.2844/0.5500/0.3401 |
0.2877/0.5770/0.3459 |
Itransformer |
0.2391/0.6236/0.2882 |
0.2573/0.6140/0.3077 |
0.2718/0.6597/0.3268 |
BRITS |
0.1902/0.3613/0.2293 |
0.2046/0.3849/0.2489 |
0.2279/0.4897/0.2754 |
2*KT |
0.1446/0.2757/0.1758 |
0.1335/0.3026/0.1582 |
0.1378/0.3118/0.1657 |
|
↓19%/↓24%/↓18% |
↓34%/↓21%/↓36% |
↓39%/↓32%/↓39% |
![]()
Figure 4. MCAR (missing rate from 1% to 70%).
4.3.2. Imputation under MNAR and MAR
Additionally, the results for simulated MNAR and MAR missingness are shown in Figure 5. Overall, deep learning methods consistently outperform statistical approaches across all missing rates. As the missing proportion increases, all methods exhibit varying degrees of performance degradation; however, deep learning models show a relatively milder decline, indicating stronger robustness to missing data. Among all methods, our model demonstrates the best overall performance.
(a) MNAR
(b) MAR
Figure 5. Performance of different models under MNAR and MAR missingness at various missing rates. (a) MNAR (b) MAR.
5. Ablation Experiment
In this section, we validate the effectiveness of KT architecture through four ablation experiments, each designed to validate the necessity of individual components. Specifically, we sequentially remove the KAN module, the transposed transformer module, and the channel-selection transformer module to assess their respective contributions. Furthermore, we verify the importance of our transposition operation by replacing the variable-dimension transformer with a time-dimension transformer, confirming that modeling along the variable dimension yields superior performance. As shown in Table 4.
1) Effectiveness of the KAN Architecture in Modeling Temporal Dependencies. We evaluate the capability of the KAN architecture in capturing temporal dependencies through comparative experiments. Specifically, we construct a unified input by applying a linear mapping to the transposed raw data and mask matrix, and feed it into the Transformer model operating along the variable dimension. The experimental results demonstrate that explicitly modeling temporal dependencies via the KAN architecture leads to a substantial improvement in interpolation performance, particularly under a 10% missing rate, where a 15.9% performance gain is observed. This enhancement may be attributed to the inherent periodic patterns in air quality data, which are effectively captured by the temporal modeling module embedded within the KAN architecture.
2) The Importance of Modeling Inter-Variable Dependencies. Table 4 presents a performance comparison after removing the variable-dimension Transformer module. The experimental results indicate that ignoring inter-variable interactions leads to a significant degradation in imputation performance, with RMSE increasing by 133%, even under conditions of a very low missing rate (1%). This finding highlights the strong correlations among variables in air quality monitoring data (such as PM2.5, SO2, NO2, etc.) and underscores the necessity of modeling cross-feature dependencies through our variable-dimension Transformer module.
3) Contribution of Spatial Information Modeling. In the ablation experiments targeting spatial information modeling, we remove the channel-selection self-attention mechanism and directly combined the outputs of the first two modules with the mask matrix to produce the final result. Although this simplified version exhibits relatively stable performance, its evaluation metrics are still notably inferior to those of the full model. This indicates that: 1) feature selection in the spatial dimension plays a critical role in imputation accuracy, and 2) relying solely on spatio-temporal features without explicitly modeling the spatial correlations within the sensor network leads to substantial information loss.
Table 4. Different versions of our method.
Model/Missing Rate |
1% |
5% |
10% |
NO_ONE |
0.1327/0.2582/0.1615 |
0.1546/0.3231/0.1832 |
0.1640/0.3603/0.1971 |
NO_TWO |
0.1468/0.3210/0.1764 |
0.1540/0.2901/0.1875 |
0.1640/0.3314/0.1944 |
NO_THREE |
0.1448/0.2626/0.1761 |
0.1498/0.3101/0.1776 |
0.1546/0.3231/0.1832 |
KT |
0.1292/0.2934/0.1372 |
0.1335/0.3026/0.1582 |
0.1378/0.3118/0.1657 |
6. Conclusion
In this study, an innovative imputation method for air quality data is proposed, which significantly improves data reconstruction performance under complex missing scenarios by systematically integrating information across three dimensions: temporal dependency, inter-variable correlation, and spatial correlation. The model employs a three-level modeling framework: first, the KAN hybrid expert architecture is used to capture the periodic characteristics and non-stationary trends of each pollutant; second, the nonlinear correlations among pollutants are modeled using the variable-dimension Transformer; and third, a spatial information fusion strategy is dynamically built based on the Spearman correlation coefficient. Experimental results show that the proposed method demonstrates strong adaptability and achieves performance improvements of 1.1% to 17.1% compared to the best baseline state-of-the-art (SOTA) model. Moreover, the model effectively controls computational complexity while maintaining high performance, with a 71.9% reduction in parameter count compared to SATIS, the strongest among the baseline models. Comparative experiments with 13 baseline methods confirm that our model achieves substantial improvements across three key metrics: imputation accuracy, stability, and generalization capability. The primary contribution of this study is the introduction of a three-dimensional coupled modeling paradigm that simultaneously captures temporal, feature-wise, and spatial dependencies, offering a novel methodological approach for complex spatiotemporal data restoration.
Data Availability
Data will be made available on request.