Interpretable Kolmogorov-Arnold Network with Feature-Isolated Temporal Attention Mechanism for Electricity Load Forecasting
2026-06-22 • Machine Learning
Machine LearningComputational Engineering, Finance, and ScienceSymbolic Computation
AI summaryⓘ
The authors created LoadKAN, a new method to predict electricity demand that is easier to understand than typical deep learning models. They combined a special attention mechanism that looks at each feature over time with a Kolmogorov-Arnold network (KAN) to make predictions. Tested on data from three U.S. electricity markets, LoadKAN performed well and helped explain how human mobility patterns affect electricity use. This approach offers clearer insights into complex relationships that are usually hidden in black-box models.
electricity load forecastingdeep learninginterpretabilityKolmogorov-Arnold networktemporal attentiontime-series forecastinghuman mobility patternssensitivity analysisnon-linear relationshipsblack-box models
Authors
Jinhao Li, Hao Wang
Abstract
Accurate electricity load forecasting is a crucial prerequisite for stable power system operations. While prevalent deep learning models present competitive performance, they often operate as black boxes and lack interpretability. While the Kolmogorov-Arnold network (KAN) has emerged as a promising alternative because of its learnable activation function design, its direct application to time-series forecasting faces challenges in modeling complex temporal data patterns. Also, simple integration into existing architectures, such as serving as replacement of neural modules, cannot fully leverage KAN's interpretability strengths. To address these gaps, this study develops LoadKAN, a novel hybrid and interpretable framework for load forecasting that synergistically combines a specifically-designed feature-isolated temporal attention mechanism with a KAN module. The attention stage aims to extract temporal dynamics from each input feature independently, such as historical load and human mobility, providing distilled feature representations to the KAN module for interpretable predictions. When evaluated on datasets from three representative U.S. electricity markets, our LoadKAN remains highly competitive when compared to extensively-tuned, state-of-the-art, black-box deep learning benchmarks. More importantly, LoadKAN's interpretability enables a granular analysis of the learned non-linear relationships between six distinct mobility patterns and electricity load. Through KAN-learned activation functions, our quantitative sensitivity analyses on mobility features reveal complex and market-specific dependencies. These findings further demonstrate the ability of our LoadKAN to generate insights often obscured by opaque black-box neural forecasting models.