An ADMM-LSTM Framework for Short-Term Load Forecasting

17 Pages Posted: 30 May 2023

See all articles by Shuo Liu

Shuo Liu

Wuhan University

Zhengmin Kong

Wuhan University

Tao Huang

James Cook University - College of Science and Engineering

Yang Du

James Cook University

Wei Xiang

La Trobe University - School of Engineering and Mathematical Sciences

Abstract

Accurate short-term load forecasting (STLF) is of paramount importance for ensuring the reliable and efficient operation of power systems. With the continuous increase in volume and variety of energy data provided by renewables, electric vehicles and other sources, long short-term memory (LSTM) has emerged as an attractive approach for STLF due to their superiorities in extracting the dynamic temporal information. Unfortunately, the conventional LSTM training methods rely on stochastic gradient methods that succumb to many limitations. In this paper, we present an innovative LSTM optimization framework via alternating direction method of multipliers (ADMM) for STLF, dubbed ADMM-LSTM. Explicitly, we train the LSTM network distributedly by ADMM algorithm. More specifically, we introduce a novel approach to update the parameters in the ADMM-LSTM framework, using a backward-forward order, which leads to a significant reduction in computational time. Furthermore, the solution of each subproblem in the proposed framework utilizes proximal point algorithm (PPA) or local linear approximation (LLA). As a benefit, the convergence rate is improved by the distributed optimization framework of ADMM-LSTM. Additionally, due to the gradient-free feature, ADMM-LSTM is able to avoid the exploding/vanishing gradient as well. Finally, we demonstrate that our proposed method exhibits superior performance compared to the benchmark methods in terms of both root mean square error (RMSE) and mean absolute percentage error (MSE).

Keywords: Alternating Direction Method of Multipliers, Long Short-term Memory Network, Short-term Load Forecasting, Gradient-free Feature

Suggested Citation

Liu, Shuo and Kong, Zhengmin and Huang, Tao and Du, Yang and Xiang, Wei, An ADMM-LSTM Framework for Short-Term Load Forecasting. Available at SSRN: https://ssrn.com/abstract=4455702 or http://dx.doi.org/10.2139/ssrn.4455702

Shuo Liu

Wuhan University ( email )

Wuhan
China

Zhengmin Kong (Contact Author)

Wuhan University ( email )

Wuhan
China

Tao Huang

James Cook University - College of Science and Engineering ( email )

Yang Du

James Cook University ( email )

Cairns, 4878
Australia

Wei Xiang

La Trobe University - School of Engineering and Mathematical Sciences ( email )

Do you have a job opening that you would like to promote on SSRN?

Paper statistics

Downloads
26
Abstract Views
148
PlumX Metrics