PL EN
RESEARCH PAPER
Multi-variable Demand Power Prediction for Heavy-duty Tractors Based on Multi-task Learning and Dynamic Weight Optimization
,
 
Yan Ma 1
,
 
,
 
,
 
,
 
 
 
 
More details
Hide details
1
State Key Laboratory of Engine and Powertrain System, Weichai Power Company Limited
 
2
State Key Laboratory of Engines, Tianjin University
 
 
Submission date: 2025-11-26
 
 
Final revision date: 2026-02-26
 
 
Acceptance date: 2026-03-04
 
 
Publication date: 2026-03-31
 
 
Corresponding author
Xinfa Xu   

State Key Laboratory of Engine and Powertrain System, Weichai Power Company Limited
 
 
The Archives of Automotive Engineering – Archiwum Motoryzacji 2026;111(1):53-74
 
KEYWORDS
TOPICS
ABSTRACT
To improve the accuracy of power demand prediction for heavy-duty tractor trucks under complex working conditions, this study proposes a synchronous prediction model based on multi-task learning (MTL-A). This model innovatively synchronously predicts three key variables: vehicle speed, road slope, and acceleration, and takes driving style and road scenarios as labeled inputs to enhance the model's perception ability of complex working conditions. Firstly, a real vehicle data collection platform was established to obtain cumulative driving data of three drivers with distinct driving styles over more than 1800 kilometers in high-speed and national road scenarios. On this basis, the impact mechanism of each variable on power demand was quantitatively analyzed based on the vehicle longitudinal dynamics model, and the significant contributions of driving style and road conditions to power fluctuations were revealed. The proposed MTL-A model adopts the CNN-LSTM-Attention (CNN-LSTM-A) architecture and introduces the GradNorm algorithm to dynamically balance the loss weights in multi-task learning, thereby alleviating gradient conflicts and achieving precise multi-variable prediction in a short period. Experimental results show that compared with traditional single-task models, the average error of MTL-A model in predicting vehicle speed, slope, and acceleration is significantly reduced by 28.4%, 13.1%, and 16.3% respectively, thereby significantly improving the calculation accuracy of the final power demand. Particularly, in typical scenarios of high-speed and national roads, the power prediction error of MTL-A model is reduced by 58.40% and 41.68% respectively compared with traditional methods, fully verifying its excellent scene adaptability and prediction efficiency.
REFERENCES (18)
1.
Fan Y, Peng J, Yu S, Yan F, Wang Z, Li M, et al. Global optimization guided energy management strategy for hybrid electric vehicles based on generative adversarial network embedded reinforcement learning. Energy. 2025;322:135586. https://doi.org/10.1016/j.ener....
 
2.
Chen D, Chen T, Li Z, Liu Z, Sun C, Zhao H. Energy management strategy for plug-in hybrid electric vehicles based on vehicle speed prediction and limited traffic information. Energy. 2025:136292. https://doi.org/10.1016/j.ener....
 
3.
Dawei M, Yu Z, Meilan Z, Risha N. Intelligent fuzzy energy management research for a uniaxial parallel hybrid electric vehicle. Computers & Electrical Engineering. 2017;58:447–464. https://doi.org/10.1016/j.comp....
 
4.
Zhang F, Hu X, Langari R, Cao D. Energy management strategies of connected HEVs and PHEVs: Recent progress and outlook. Progress in Energy and Combustion Science. 2019;73:235–256. https://doi.org/10.1016/j.pecs....
 
5.
Xing J, Chu L, Hou Z, Sun W, Zhang Y. Energy management strategy based on a novel speed prediction method. Sensors. 2021;21(24):8273. https://doi.org/10.3390/s21248....
 
6.
Xue Q, Ma J, Zhao X, Liu R, Li H, Zhu X. Informer-FDR: A short-term vehicle speed prediction model in car-following scenario based on traffic environment. Expert Systems with Applications. 2025;262:125655. https://doi.org/10.1016/j.eswa....
 
7.
Chen B, Ma R, Zhou Y, Ma R, Jiang W, Yang F. Co-optimization of speed planning and cost-optimal energy management for fuel cell trucks under vehicle-following scenarios. Energy Conversion and Management. 2024;300:117914. https://doi.org/10.1016/j.enco....
 
8.
Pulvirenti L, Rolando L, Millo F. Energy management system optimization based on an LSTM deep learning model using vehicle speed prediction. Transportation Engineering. 2023;11:100160. https://doi.org/10.1016/j.tren....
 
9.
Guo J, He H, Sun C. ARIMA-based road gradient and vehicle velocity prediction for hybrid electric vehicle energy management. IEEE Transactions on Vehicular Technology. 2019;68(6):5309–5320. https://doi.org/10.1109/TVT.20....
 
10.
Zhang Q, Filev D, Szwabowski S, Langari R. A real-time fuzzy learning algorithm for Markov chain and its application on prediction of vehicle speed. 2019 IEEE International Conference on Fuzzy Systems (FUZZ-IEEE), 2019 June 23-26. New Orleans, LA, USA. https://doi.org/10.1109/FUZZ-I....
 
11.
Yue F, Liu Q, Kong Y, Zhang J, Xu N. A predictive energy management strategy for multi-energy source vehicles based on full-factor trip information. Sensors. 202;22(1):23. https://doi.org/10.3390/s22010....
 
12.
Zhou Y, Ravey A, Pera MC. A velocity prediction method based on self-learning multi-step Markov chain. IECON 2019-45th Annual Conference of the IEEE Industrial Electronics Society 2019;1:2598–2603. https://doi.org/10.1109/IECON.....
 
13.
Cao Y, Liang C, Cheng S, Yin X, Chen D, Liu Z, et al. Predictive Energy Management Strategy for Heavy-Duty Series Hybrid Electric Vehicles Based on Drive Power Prediction. World Electric Vehicle Journal. 2025;16(3). https://doi.org/10.3390/wevj16....
 
14.
Pang Z, Zhang J, Jiao X, Ren L. Energy management strategy combining double deep Q‐networks and demand torque prediction for connected hybrid electric vehicles. IEEJ Transactions on Electrical and Electronic Engineering. 2024;19(2):234–246. https://doi.org/10.1002/tee.23....
 
15.
Feng J, Han Z, Wu Z, Li M. Approximate optimal energy management with a high-precision vehicle speed prediction algorithm. Proceedings of the Institution of Mechanical Engineers, Part D: Journal of Automobile Engineering. 2024;238(4):774–787. https://doi.org/10.1177/095440....
 
16.
Wen M, Mei H, Yuan F. Survey of multi-task recommendation algorithms. Journal of Frontiers of Computer Science and Technology. 2024;2:363–377. https://doi.org/10.3778/j.issn....
 
17.
Ping M, Yan WJ, Jia X, Papadimitriou C, Yuen KV. Learning non-stationary model of prediction errors with hierarchical Bayesian modeling. Reliability Engineering & System Safety. 2025;260:111012. https://doi.org/10.1016/j.ress....
 
18.
Chen Z, Badrinarayanan V, Lee CY, Rabinovich A. Gradnorm: Gradient normalization for adaptive loss balancing in deep multitask networks. Proceedings of the 35th International Conference on Machine Learning. 2018;80:794–803. https://doi.org/10.48550/arXiv....
 
Declaration of availability
 
eISSN:2084-476X
Journals System - logo
Scroll to top