• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

计算机工程与科学 ›› 2022, Vol. 44 ›› Issue (11): 2080-2090.

• 人工智能与数据挖掘 • 上一篇    

基于改进人工鱼群算法与RBF神经网络的股票价格预测

谢俊标1,江峰1,杜军威1,赵军2   

  1. (1.青岛科技大学信息科学技术学院,山东 青岛 266061;2.青岛科技大学化工学院,山东 青岛 266042)
  • 收稿日期:2020-08-16 修回日期:2021-01-26 接受日期:2022-11-25 出版日期:2022-11-25 发布日期:2022-11-25
  • 基金资助:
    国家自然科学基金(61973180,61671261);山东省自然科学基金(ZR2022MF326,ZR2018MF007)

Stock price prediction based on an improved artificial fish swarm algorithm and RBF neural network

XIE Jun-biao1,JIANG Feng1,DU Jun-wei1,ZHAO Jun2   

  1. (1.College of Information Science and Technology,Qingdao University of Science & Technology,Qingdao 266061;
    2.College of Chemical Engineering,Qingdao University of Science & Technology,Qingdao 266042,China)
  • Received:2020-08-16 Revised:2021-01-26 Accepted:2022-11-25 Online:2022-11-25 Published:2022-11-25

摘要: 股票价格受多种因素影响,这对股票价格预测造成了巨大挑战。近年来,机器学习方法被广泛用于股票价格预测的研究中,然而,现有方法存在相对误差较大、时间复杂度高等缺点。对此,提出基于引力搜索的改进人工鱼群算法AFSA_GS。该算法将引力搜索中计算质量和加速度的策略分别用于调节人工鱼的视野和步长,从而提高了人工鱼群算法在优化过程中的自适应能力;AFSA_GS算法还优化了RBF神经网络的相关参数,并使用优化后的网络预测股票价格。在多家上市公司股票数据上进行了实验,实验结果表明,相对于传统的优化算法,采用AFSA_GS算法优化的RBF神经网络,可以获得更好的股票预测性能。

关键词: 股票价格预测, 人工鱼群算法, 引力搜索算法, RBF神经网络, 视野, 步长

Abstract: Stock prices are affected by many factors, which poses a great challenge to stock index prediction. In recent years, machine learning has been widely used in the research of stock price prediction. However, the existing methods have some disadvantages such as large relative error and high time complexity. An improved artificial fish swarm algorithm based on gravity search (AFSA_GS) is proposed. AFSA_GS applies the gravity search strategy of calculating mass and acceleration to the visual and step size adjustment of artificial fish respectively, so as to improve the adaptive ability of artificial fish swarm algorithm in the optimization process. AFSA_GS is further used to optimize the relevant parameters of RBF neural network, and the optimized network is used to predict the stock price. Experiments were conducted on the stock data of a number of listed companies. The results show that, compared with the traditional optimization algorithm, AFSA_GS algorithm can be used to optimize RBF neural network, which can obtain better stock prediction performance.


Key words: stock price prediction, artificial fish swarm algorithm, gravitational search algorithm, RBF neural network, visual, step