Prediction of Time and Distance of Trips Using Explainable Attention-based LSTMs
Preprint, 2023

distance the vehicle will travel. For this prediction task, we develop and investigate four methods.
In the first method, we use long short-term memory (LSTM)-based structures specifically designed
to handle multi-dimensional historical data of trip time and distances simultaneously. Using it, we
predict the future trip time and forecast the distance a vehicle will travel by concatenating the outputs
of LSTM networks through fully connected layers. The second method uses attention-based LSTM
networks (At-LSTM) to perform the same tasks. The third method utilizes two LSTM networks in
parallel, one for forecasting the time of the trip and the other for predicting the distance. The output
of each LSTM is then concatenated through fully connected layers. Finally, the last model is based
on two parallel At-LSTMs, where similarly, each At-LSTM predicts time and distance separately
through fully connected layers. Among the proposed methods, the most advanced one, i.e., parallel
At-LSTM, predicts the next trip’s distance and time with 3.99 % error margin where it is 23.89 %
better than LSTM, the first method. We also propose TimeSHAP as an explainability method for
understanding how the networks perform learning and model the sequence of information.

Author

Ebrahim Balouji

Energy Conversion and Propulsion Systems

Jonas Sjöblom

Energy Conversion and Propulsion Systems

Nikolce Murgovski

Chalmers, Electrical Engineering, Systems and control

Morteza Haghir Chehreghani

Data Science and AI 2

Modelling and optimization of energy management systems for plug-in hybrid vehicles

Swedish Energy Agency (2019-013262), 2019-10-01 -- 2022-12-31.

Areas of Advance

Information and Communication Technology

Transport

Driving Forces

Sustainable development

Subject Categories (SSIF 2025)

Transport Systems and Logistics

Computer Sciences

Computer Engineering

DOI

10.48550/arXiv.2303.15087

More information

Latest update

6/24/2025