首页|基于DPSO-LSTM超参数调优的股市价格预测

基于DPSO-LSTM超参数调优的股市价格预测

扫码查看
长短期记忆网络(Long Short-Term Memory,LSTM)适合处理和预测时间序列中间隔和延迟较长的重要事件。由于其复杂的网络结构、不确定的超参数和耗时的网络训练,使得人工寻找高效的网络配置成为一项具有挑战性的工作。文中采用分布式粒子群算法(Distributed Particle Swarm Optimi-zation,DPSO)来有效解决LSTM的超参数调优问题,研究LSTM中最优的隐藏元个数、激活函数以及学习率等超参数的选择,寻找高性能的LSTM。基于沪深300历史交易数据进行价格预测,实验结果表明该方法是有效的,这为超参数调优与股市价格预测提供了新的思路和方法。
Stock market price forecasting based on DPSO-LSTM hyperparameter tuning
Long Short-Term Memory(LSTM)is suitable for processing and predicting important events with very long intervals and delays in time series.It is a challenging task to manually find an efficient network configuration due to its complex network structure,uncertain hyperparameters,and time-consuming network training.In this paper,Distributed Particle Swarm Optimization(DPSO)is used to effectively solve the hy-perparameter tuning problem of LSTM,and research the selection of the optimal number of hidden ele-ments,activation function and learning rate in LSTM to find a high-performance LSTM.Based on the histor-ical transaction data of CSI300 for price prediction,the experiment results show that the method is effective,which provides new ideas and methods for hyperparameter tuning and stock market price prediction.

Long Short-Term Memoryartificial neural network modeldistributed particle swarm optimi-zationhyperparameter tuningstock market prediction

张成军、李琪、王梅、乔译、陈亚当、余文斌

展开 >

南京信息工程大学,江苏大气环境和装备技术协同创新中心,南京 210044

南京信息工程大学计算机学院,南京 210044

南京信息工程大学数字取证教育部工程研究中心,南京 210044

长短期记忆网络 人工神经网络 分布式粒子群优化算法 超参数调优 股市预测

国家自然科学基金国家自然科学基金国家自然科学基金

615012476170321261802197

2024

信息技术
黑龙江省信息技术学会 中国电子信息产业发展研究院 中国信息产业部电子信息中心

信息技术

CSTPCD
影响因子:0.413
ISSN:1009-2552
年,卷(期):2024.(5)