基于改进的LSTM算法的时间序列流量预测

作者：郭佳丽; 邢双云^*; 栾昊; 贾艳婷

来源：南京信息工程大学学报(自然科学版), 2021, 13(05): 571-575.

DOI：10.13878/j.cnki.jnuist.2021.05.009

• 时间序列预测
• 长短期记忆网络
• 预测精度
• 泛化能力

时间序列流量的预测问题是近年来机器学习的一个热点问题,通过改变长短期记忆网络(LSTM)层数、网络层神经元的个数、网络层之间的连接方式,特殊网络层的应用等网络结构以及优化器和损失函数的选择可以极大地提高预测的精度.本文提出多层LSTM算法,该算法是在传统LSTM算法上进行改进的单一模型,模型设计的复杂度低,可以提高机器学习的效率.模型采用一个输入层、5个隐藏层、1个输出层,同时包含1个全连接层和1个Dropout层,Dropout层的作用是防止机器学习过拟合.选择adam为模型优化器、mlse为模型损失函数、relu作为模型的激活函数.实验结果表明,与传统模型相比,该模型具有较好的泛化能力.

• 单位
  沈阳建筑大学