IEEE Access (Jan 2021)

A Multi-Scale Electricity Consumption Prediction Algorithm Based on Time-Frequency Variational Autoencoder

  • Kaihong Zheng,
  • Peng Li,
  • Shangli Zhou,
  • Wenhan Zhang,
  • Sheng Li,
  • Lukun Zeng,
  • Yingnan Zhang

DOI
https://doi.org/10.1109/ACCESS.2021.3071452
Journal volume & issue
Vol. 9
pp. 90937 – 90946

Abstract

Read online

Accurate electricity consumption forecasting can be treated as a reliable guidance for power production. However, traditional electricity forecasting models suffer from simultaneously capturing the periodicity and the volatility of sequential electricity consumption data, while the periodicity and the volatility are important for electricity forecasting. In order to effectively model this sequential data and predict electricity consumption accurately, we propose a multi-scale prediction (Long Short Term Memory, LSTM) algorithm based on Time-Frequency Variational Autoencoder (TFVAE-LSTM). The proposed algorithm treats the sequential data as a superposition of data in different frequencies, it defines an encoder in frequency domain to extract frequency features to model the periodicity and volatility, and defines a decoder in time domain to capture the sequential features of data. Based on the extracted Time-Frequency features in a TFVAE, a multi-scale LSTM model is defined to further extract sequential features from different scales to predict electricity consumption. Experiments show the effectiveness of the proposed TFVAE-LSTM for electricity consumption forecasting tasks.

Keywords