Improved FA-BP Neural Network Traffic Flow Prediction Algorithm

doi:10.3969/j.issn.1006-2475.2025.10.013

Abstract

Abstract: Abstract: Traffic flow prediction is one of the important technical means to improve efficiency and reduce congestion in intelligent transportation systems. A BP neural network traffic flow prediction method based on improved Firefly Algorithm （FA） and Levenberg Marquardt （LM） algorithm is proposed to address the problems of slow convergence speed and low prediction accuracy in existing traffic flow prediction algorithms. This method utilizes an improved chaotic Firefly Algorithm to optimize the initial weights and thresholds of the BP neural network， and uses the LM algorithm instead of the traditional gradient descent method in the weight update stage to accelerate the convergence process and improve model accuracy. Finally， the LM-FA-BP algorithm is used to predict traffic flow. Based on real complex urban traffic data， multiple fusion models were compared through experiments. The prediction error of our model was significantly reduced compared to other models， with a 33.84% improvement in Mean Absolute Error （MAE） compared to the BP model and a 29.82% improvement compared to the FA-BP model. The model has been tested and implemented on actual roads， with a maximum accuracy of 98% （average absolute percentage error<2.0%）， reaching a high level. The improved LM-FA-BP model has higher accuracy and faster convergence speed in traffic flow prediction. The research results indicate that the model has broad application prospects， especially in intelligent transportation systems where it can effectively improve prediction accuracy.

Key words: Key words: traffic flow prediction, neural networks, firefly algorithm, Levenberg-Marquardt algorithm

CLC Number:

中图分类号：TP 301.6

WANG Yuanrui, JIANG Lingyun. Improved FA-BP Neural Network Traffic Flow Prediction Algorithm[J]. Computer and Modernization, 2025, 0(10): 80-88.

References

［1］ GEBRE T S， BENI L， WASEHUN， E T， et al. AI-integrated traffic information system: A synergistic approach of physics informed neural network and GPT-4 for traffic estimation and real-time assistance［J］. IEEE Access， 2024，12:65869-65882.
［2］ SAHA N， SWETAPADMA M， Mondal M. A brief review on artificial neural network: Network structuresand applications［C］// 2023 9th International Conference on Advanced Computing and Communication Systems（ICACCS）. IEEE， 2023:1974-1979.
［3］ LI F D， SHI D S， WANG N N， et al. Research on stock prediction based on PCA-GA-BP modeling［C］// 2024 6th International Conference on Electronics and Communication， Network and Computer Technology （ECNCT）. IEEE， 2024:262-265.
［4］ HOSSAIN M N， AHMED N， Traffic flow forecasting in intelligent transportation systems prediction using machine learning［J］. // International Conference on Futuristic Technologies （INCOFT）. IEEE， 2022:1-5.
［5］ LUO X L， LI D Y， YANG Y， et al. Spatiotemporal traffic flow prediction with KNN and LSTM［J］. Journal of Advanced Transportation， 2019（1）:537-546.
［6］ ZHAO F X， ZENG G Q， LU K D. EnLSTM-WPEO: Short-term traffic flow prediction by ensemble LSTM， NNCT weight integration， and population extremal optimization［J］. IEEE Transactions on Vehicular Technology， 2020，69（1）:101-113.
［7］ OSIPOV V， NIKIFOROV V， ZHUKOVA N， et al. Urban traffic flows forecasting by recurrent neural networks with spiral structures of layers［J］. Neural Computing and Applications， 2020，32（18）:14885-14897.
［8］ HE P L， JIANG G Y， LAM S-K， et al. Learning heterogeneous traffic patterns for travel time prediction of bus journeys［J］. Information Sciences， 2020，512:1394-1406.
［9］ GUO S N， LIN Y F， LI S J， et al. Deep spatial–temporal 3D convolutional neural networks for traffic data forecasting［J］. IEEE Transactions on Intelligent Transportation Systems， 2019，20（10）:3913-3926.
［10］ LI M Y， JI X， WANG X， et al. Study on load prediction based on the MPGA-BP neural network model［C］// 2023 China Automation Congress （CAC）. IEEE， 2023:6109-6114.
［11］ DING C L， JUANATAS R A. A short-term traffic flow forecasting method based on SVM-PSO-BP combined model［J］. Robotics， Intelligent Control and Artificial Intelligence， 2023，5:869-873.
［12］ PENG H， WANG H F， DU B W， et al. Spatial temporal incidence dynamic graph neural networks for traffic flow forecasting［J］. Information Science， 2020，521:277-290.
［13］ ZHAO L， ZHOU Y H， LU H P， et al. Parallel computing method of deep belief networks and its application to traffic flow prediction［J］. Knowledge-Based Systems， 2019，163:972-987.
［14］ LI L C， QIN L Q， QU X， et al. Day-ahead traffic flow forecasting based on a deep belief network optimized by the multi-objective particle swarm algorithm［J］. Knowledge-Based Systems， 2019，172:1-14.
［15］ WANG T， ZHANG B B， WEI W， et al. Traffic flow prediction based on BP neural network［C］// 2021 IEEE International Conference on Artificial Intelligence and Industrial Design （AIID）. IEEE， 2021：312-315.
［16］ ZHANG L Z， LIU W C， FENG L P. Short-term traffic flow prediction based on improved neural network with GA［C］// 2021 IEEE International Conference on Artificial Intelligence and Computer Applications （ICAICA）. IEEE， 2021:28-30.
［17］ YANG X F， ZHANG L W， XIE W. Forecasting model for urban traffic flow with BP neural network based on genetic algorithm［C］// 2019 Chinese Control and Decision Conference （CCDC）. IEEE， 2019:4395-4399.
［18］ LIU J G， HUANG J， SUN R， et al. Data fusion for multi-source sensors using GA-PSO-BP neural network［J］. IEEE Transactions on Intelligent Transportation Systems， 2020，22（10）:6583-6598.
［19］ LU L X. Power load forecasting based on improved BP neural network and LM algorithm［C］// 2023 IEEE International Conference on Sensors， Electronics and Computer Engineering （ICSECE）. IEEE， 2023:1020-1024.
［20］ HAGAN M T， MENBAJ M B. Training feedforward networks with the marquardt algorithm［J］. IEEE Transactions on Neural Networks， 1994，5（6）:295-301.
［21］魏学诚，江凌云，李研，等. 改进YOLOv5的路侧单目视角小目标检测算法［J］. 计算机与现代化， 2024（10）: 27-34.
［22］毛明扬，徐胜超. 面向粒子群优化BP神经网络的粗糙集连续属性离散化算法［J］. 计算机与现代化， 2023（9）: 115-119.
［23］ ZHANG J Y， CUI X M， L L J， et al. Research on emergency parcel mobilization in e-commerce logistics network based on BP neural network and Bayesian network prediction model［C］// 2024 IEEE 4th International Conference on Electronic Technology. IEEE，2024:834-839.
［24］孟闯，王慧，林浩，等. 道路交通流数据预测方法研究综述［J］. 计算机工程与应用， 2023，59（14）:51-61.
［25］ ZHU Y， ZHANG G H， QIU J. Network traffic prediction based on particle swarm BP neural network［J］. Journal of Netwroks， 2013，8（11）:2685-2691.
［26］ YUAN X. BP neural network optimization model based on nonlinear function transformation approach［C］// 2015 International Conference on Intelligent Transportation， Big Data and Smart City. IEEE， 2015:184-187.
［27］ KONG Q J， LI Z P， CHEN Y K， et al. An Approach to Urban Traffic State Estimation by Fusing Multisource Information［J］. IEEE Transactions on Intelligent Transportation Systems， 2009，10（3）:499-511.

[1]	WEI Qingsong1, WANG Xiaojun2, WEI Yuan1, ZENG Shangqi1, FANG Yuan1. HGAT: Multivariate Time Series Anomaly Detection Based on Hybrid Graph Attention Network [J]. Computer and Modernization, 2025, 0(07): 97-105.
[2]	JI Zhengjie, WEI Linjing. ICS-ResNet： A Lightweight Network for Maize Leaf Disease Classification [J]. Computer and Modernization, 2025, 0(04): 19-28.
[3]	XIE Haiqing, LING Jiaqi, YI Xinbo. Tongue Constitution Classification Method Based on Deep Learning [J]. Computer and Modernization, 2025, 0(03): 99-105.
[4]	LIU Baobao, YANG Jingjing, TAO Lu, WANG Heying . DSMSC Based on Attention Mechanism for Remote Sensing Image Scene Classification [J]. Computer and Modernization, 2024, 0(12): 72-77.
[5]	CAO Ning1, YAN Xin’e1, XU Genqi2, XU Youwen1, ZHANG Zhengbo2, DU Qianyun2. Prediction Model of Hydraulic Engineering Slope Mechanical Parameters Based on DEFA-LSSAR [J]. Computer and Modernization, 2024, 0(07): 106-111.
[6]	HUANG Zheng-lin, DONG Bao-liang. Temporal Knowledge Graph Question Answering Method Based on#br# Semantic and Structural Enhancement [J]. Computer and Modernization, 2024, 0(03): 15-23.
[7]	WANG Yu-hang, DONG Bao-liang, GONG Chao, SHANG Zhen-zhen, YAO Kang-ning. Dynamic Threat Assessment of Air Swarm Targets Based on Intent Recognition [J]. Computer and Modernization, 2023, 0(12): 100-104.
[8]	GU Ming-xuan, FAN Bing-bing. Feature-level Multimodal Fusion for Depression Recognition [J]. Computer and Modernization, 2023, 0(10): 17-22.
[9]	LI Shi-da, XIANG Jian-wen. A Weakened Joint Reinforcement Method to Improve Robustness of Image Recognition Models [J]. Computer and Modernization, 2023, 0(10): 70-76.
[10]	LIU Fu-qi, ZHANG Da, SONG Jian-hua, WANG Hai-dong. Fault Diagnosis of Hydraulic Systems Based on CNN-BiLSTM [J]. Computer and Modernization, 2023, 0(09): 10-19.
[11]	XU Ye-tong, GENG Xin-zhe, ZHAO Wei-qiang, ZHANG Yue, NING Hai-long, LEI Tao. A Remote Sensing Image Change Detection Model Based on CNN-Transformer Hybrid Structure [J]. Computer and Modernization, 2023, 0(07): 79-85.
[12]	NIU Yu-heng, LI Yong-ke, CHEN Yan-hong, JANG Ping-an. Image Segmentation Method of Residual Film on Cotton Field Surface Based on Improved SegFormer Model#br# [J]. Computer and Modernization, 2023, 0(07): 93-98.
[13]	HAN Mu-feng. Multi-time Scale Scheduling Simulation of Microgrid Considering Demand Response [J]. Computer and Modernization, 2023, 0(03): 102-106.
[14]	WU Song-lin, ZHANG Guo-wei, LU Qiu-hong, SHI Jian-zhuang, HUANG Wei. Laser SLAM Mapping Method Based on Visual Information [J]. Computer and Modernization, 2023, 0(02): 17-23.
[15]	HUANGFU Xiao-ying, QIAN Hui-min, HUANG Min. A Review of Deep Neural Networks Combined with Attention Mechanism [J]. Computer and Modernization, 2023, 0(02): 40-49.