An Adaptive Hybrid Short-Term Load Forecasting Framework Based on Improved Rime Optimization Variational Mode Decomposition and Cross-Dimensional Attention

Accurate Short-Term Load Forecasting (STLF) is paramount for the stable and economical operation of power systems, particularly in the context of high renewable energy penetration, which exacerbates load volatility and non-stationarity. The prevailing advanced “decomposition–ensemble” paradigm, however, faces two significant challenges when processing non-stationary signals: (1) The performance of Variational Mode Decomposition (VMD) is highly dependent on its hyperparameters (K, α), and traditional meta-heuristic algorithms (e.g., GA, GWO, PSO) are prone to converging to local optima during the optimization process; (2) Deep learning predictors struggle to dynamically weigh the importance of multi-dimensional, heterogeneous features (such as the decomposed Intrinsic Mode Functions (IMFs) and external climatic factors). To address these issues, this paper proposes a novel, adaptive hybrid forecasting framework, namely IRIME-VMD-CDA-LSTNet. Firstly, an Improved Rime Optimization Algorithm (IRIME) integrated with a Gaussian Mutation strategy is proposed. This algorithm adaptively optimizes the VMD hyperparameters by targeting the minimization of average sample entropy, enabling it to effectively escape local optima. Secondly, the optimally decomposed IMFs are combined with climatic features to construct a multi-dimensional information matrix. Finally, this matrix is fed into an innovative Cross-Dimensional Attention (CDA) LSTNet model, which dynamically allocates weights to each feature dimension. Ablation experiments conducted on a real-world dataset from a distribution substation demonstrate that, compared to GA-VMD, GWO-VMD, and PSO-VMD, the proposed IRIME-VMD method achieves a reduction in Root Mean Square Error (RMSE) of up to 18.9%. More importantly, the proposed model effectively mitigates the “prediction lag” phenomenon commonly observed in baseline models, especially during peak load periods. This framework provides a robust and high-accuracy solution for non-stationary load forecasting, holding significant practical value for the operation of modern power systems.