CNN-LSTM Base Station Traffic Prediction Based On Dual Attention Mechanism and Timing Application

• Published in: Computer journal Vol. 67; no. 6; pp. 2246 - 2256
• Main Authors: Jia, Hairong, Wang, Suying, Ren, Zelong
• Format: Journal Article
• Language: English
• Published: Oxford University Press 24.06.2024
• Subjects: intelligent timing; attention mechanism; base station; k-means algorithm; CNN-LSTM; traffic prediction
• ISSN: 0010-4620; 1460-2067
• DOI: 10.1093/comjnl/bxae003

Energy consumption in 5G base stations remains consistently high, even during periods of low traffic loads, thereby resulting in unnecessary inefficiencies. To address this problem, this paper presents a novel approach by proposing a convolutional neural network (CNN)-long short-term memory (LSTM) traffic prediction model with a dual attention mechanism, coupled with the particle swarm optimization k-means algorithm for intelligent switch timing. The proposed CNN-LSTM model leverages a dual channel attention mechanism to bolster key feature information for long-term traffic data predictions. Specifically, a temporal attention mechanism is added to the LSTM to enhance the importance of temporal information. Moreover, the particle swarm optimization K-Means algorithm is proposed in order to cluster the traffic prediction results, output the corresponding time points of the lower traffic value and to obtain the optimal switch-off periods of the base station. Extensive experiments across multiple base stations over an extended period of time have validated our approach. The results show that this method offers accurate traffic prediction with minimal average errors in traffic prediction and the on/off timings of the base stations are in line with the “tide effect” of traffic, thereby achieving the goal of energy savings.