PLoS ONE (Jan 2024)

Multi-strategy modified sparrow search algorithm for hyperparameter optimization in arbitrage prediction models.

  • Shenjie Cheng,
  • Panke Qin,
  • Baoyun Lu,
  • Jinxia Yu,
  • Yongli Tang,
  • Zeliang Zeng,
  • Sensen Tu,
  • Haoran Qi,
  • Bo Ye,
  • Zhongqi Cai

DOI
https://doi.org/10.1371/journal.pone.0303688
Journal volume & issue
Vol. 19, no. 5
p. e0303688

Abstract

Read online

Deep learning models struggle to effectively capture data features and make accurate predictions because of the strong non-linear characteristics of arbitrage data. Therefore, to fully exploit the model performance, researchers have focused on network structure and hyperparameter selection using various swarm intelligence algorithms for optimization. Sparrow Search Algorithm (SSA), a classic heuristic method that simulates the sparrows' foraging and anti-predatory behavior, has demonstrated excellent performance in various optimization problems. Hence, in this study, the Multi-Strategy Modified Sparrow Search Algorithm (MSMSSA) is applied to the Long Short-Term Memory (LSTM) network to construct an arbitrage spread prediction model (MSMSSA-LSTM). In the modified algorithm, the good point set theory, the proportion-adaptive strategy, and the improved location update method are introduced to further enhance the spatial exploration capability of the sparrow. The proposed model was evaluated using the real spread data of rebar and hot coil futures in the Chinese futures market. The obtained results showed that the mean absolute percentage error, root mean square error, and mean absolute error of the proposed model had decreased by a maximum of 58.5%, 65.2%, and 67.6% compared to several classical models. The model has high accuracy in predicting arbitrage spreads, which can provide some reference for investors.