Submitted:
13 April 2026
Posted:
14 April 2026
You are already at the latest version
Abstract
Keywords:
1. Introduction
2. Related Work
2.1. Deep Learning for Ship Trajectory Analysis
2.2. Tree-Based Ensemble Methods
2.3. Hybrid Deep-Tree Architectures
2.4. Explainability and Data Quality in AIS Processing
3. Proposed Method
3.1. Overall Framework
4. Dataset
| Class ID | Ship Type | Trajectories | Percentage (%) |
| 1 | Bulk Carrier | 227 | 10.34 |
| 2 | Cargo Ship | 242 | 11.02 |
| 3 | Chemical Tanker | 109 | 4.96 |
| 4 | Container Ship | 193 | 8.79 |
| 5 | Fishing Ship | 233 | 10.61 |
| 6 | LNG Carrier | 42 | 1.91 |
| 7 | Other Cargo | 31 | 1.41 |
| 8 | Passenger Ship | 185 | 8.42 |
| 9 | Pleasure Craft | 31 | 1.41 |
| 10 | Reefer Ship | 166 | 7.56 |
| 11 | Ro-Ro Ship | 165 | 7.51 |
| 12 | Tanker | 229 | 10.43 |
| 13 | Tug | 173 | 7.88 |
| 14 | Vehicle Carrier | 170 | 7.74 |


5. Experimental Results
5.1. Experimental Setup
5.2. Comparison with Baseline Models
5.3. Ablation Studies
5.4. Performance Analysis
5.5. Inference Efficiency
5.6. Discussion
6. Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Acknowledgments
Conflicts of Interest
References
- Sheng, K.; Liu, Z.; Zhou, D.; He, A.; Feng, C. Research on ship classification based on trajectory features. J Navig. 2018, 71(1), 100–116. [Google Scholar] [CrossRef]
- Rong, H.; Teixeira, A.P.; Soares, C.G. Maritime traffic probabilistic prediction based on ship motion pattern extraction. Reliab Eng Syst Saf. 2022, 217, 108061. [Google Scholar] [CrossRef]
- Zhang, C.; Liu, S.; Guo, M.; Liu, Y. A novel ship trajectory clustering analysis and anomaly detection method based on AIS data. Ocean Eng. 2023, 288, 116082. [Google Scholar] [CrossRef]
- Chondrodima, E.; Peleklis, N.; Pikrakis, A.; Theodoridis, Y. An efficient LSTM neural network-based framework for ship location forecasting. IEEE Trans Intell Transp Syst. 2023, 24(5), 4872–4888. [Google Scholar] [CrossRef]
- Xue, H.; Wang, S.; Xia, M.; Guo, S. G-trans: a hierarchical approach to ship trajectory prediction with GRU-based transformer. Ocean Eng. 2024, 300, 117431. [Google Scholar] [CrossRef]
- Liu, J.; Zhu, K.; Shu, Y.; Zhang, Y.; Zheng, P. Trajectory prediction model of a ship crossing under a bridge based on multi-head attention mechanism and Wasserstein GAN with gradient penalty. Ocean Eng. 2025, 334, 121551. [Google Scholar] [CrossRef]
- You, J.; Chen, Y.; Jiang, Z.; Liu, Z.; Huang, Z.; Ding, Y.; Ran, B. Exploring driving behavior for autonomous vehicles based on Gramian Angular Field Vision Transformer. IEEE Trans Intell Transp Syst. 2024, 25(11), 17493–17504. [Google Scholar] [CrossRef]
- Zhang, Y.; Chen, Y.; Li, Y.; et al. TPTrans: Ship Trajectory Prediction Model Based on Transformer Using AIS Data. ISPRS Int J Geo-Inf. 2024, 13(1), 400. [Google Scholar] [CrossRef]
- Zhou, H.; Zhang, S.; Peng, J.; et al. Informer: Beyond efficient transformer for long sequence time-series forecasting. Proc AAAI Conf Artif Intell., 2021; 35, pp. 11106–11115. [Google Scholar]
- Wu, H.; Xu, J.; Wang, J.; Long, M. Autoformer: Decomposition transformers with auto-correlation for long-term series forecasting. Adv Neural Inf Process Syst. 2021, 34, 22419–22430. [Google Scholar]
- Zhang, Y.; Liu, Y.; Zheng, Z.; et al. AIS data-driven maritime monitoring based on Transformer: A comprehensive review. arXiv 2025, arXiv:2505.07374. [Google Scholar] [CrossRef]
- Zhang, Z.; Huang, L.; Peng, X.; Wen, Y.; Song, L. Loitering behavior detection and classification of ship movements based on trajectory shape and convolutional neural networks. Ocean Eng. 2022, 258, 111852. [Google Scholar] [CrossRef]
- Hanafi, N.; Saadatfar, H. A fast DBSCAN algorithm for big data based on efficient density calculation. Expert Syst Appl. 2022, 203, 117501. [Google Scholar] [CrossRef]
- Ke, G.; Meng, Q.; Finley, T.; Wang, T.; Chen, W.; Ma, W.; et al. LightGBM: A highly efficient gradient boosting decision tree. Adv Neural Inf Process Syst. 2017, 30, 3146–3154. [Google Scholar]
- Ferreira, M.D.; Spadon, G.; Soares, A.; Matwin, S. A semi-supervised methodology for fishing activity detection using the geometry behind the trajectory of multiple ships. Sensors 2022, 22(16), 6063. [Google Scholar] [CrossRef]
- Ferreira, L.; Bifet, A.; Gama, J. CNN and XGBoost for time series classification: A comparative study. In 2020 International Joint Conference on Neural Networks (IJCNN); IEEE, 2020; pp. 1–8. [Google Scholar]
- Fu, H.; Gu, Z.; Wang, H.; Wang, Y. Ship motion prediction based on ConvLSTM and XGBoost variable weight combination model. In OCEANS 2022-Chennai; IEEE, 2022; pp. 1–8. [Google Scholar]
- Liu, Y.; Wu, H.; Wang, J.; Long, M. Non-stationary transformers: Exploring the stationarity in time series forecasting. Advances in Neural Information Processing Systems 2022, 35, 9881–9893. [Google Scholar]
- Pallotta, G.; Vespe, M.; Bryan, K. Ship pattern knowledge discovery from AIS data: A framework for anomaly detection and route prediction. Entropy 2013, 15(6), 2218–2245. [Google Scholar] [CrossRef]
- Zhang, L.; Wang, H.; Chen, Y.; et al. An interpretable ship risk model based on machine learning and SHAP interpretation technique. Ocean Eng. 2025, 319, 119392. [Google Scholar] [CrossRef]
- Liu, Y.; Chen, H.; Zhang, X.; et al. A method for ship trajectory repair based on feature correlation and SHAP model interpretability. In Proceedings of the 3rd International Conference on Spatial Data and Intelligence; 2025; Singapore Lecture Notes in Computer Science; Springer, 2025; pp. 102–115. [Google Scholar] [CrossRef]
- Wang, Y.; Liu, X.; Chen, Z. VISTA: Knowledge-driven interpretable ship trajectory imputation via large language models. arXiv 2026, arXiv:2601.06940. [Google Scholar]
- Chen, X.; Zhang, Y.; Liu, M. Physics-informed neural networks for ship trajectory reconstruction under uncertain conditions. Ocean Eng. 2024, 285, 115890. [Google Scholar] [CrossRef]
- Graves, A. Long short-term memory. In Supervised Sequence Labelling with Recurrent Neural Networks; Springer, 2012; pp. 37–45. [Google Scholar]
- Zhou, T.; Ma, Z.; Wen, Q.; et al. FEDformer: Frequency enhanced decomposed transformer for long-term series forecasting. Proc Mach Learn Res. 2022, 162, 27268–27286. [Google Scholar]
- Zhang, Y.; Chen, X.; Liu, M. Anomaly detection in maritime trajectories using Transformer-based variational autoencoders. IEEE Trans Intell Transp Syst. 2024, 25(8), 10234–10245. [Google Scholar]
- Breiman, L. Random forests. Mach Learn 2001, 45(1), 5–32. [Google Scholar] [CrossRef]
- Sánchez Pedroche, D.; Amigo, D.; Campaña, J.M.; Illán, Á; Mateos, J. Architecture for trajectory-based fishing ship classification with AIS data. Sensors 2020, 20(13), 3782. [Google Scholar] [CrossRef] [PubMed]
- Chen, T.; Guestrin, C. XGBoost: A scalable tree boosting system. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, 2016; ACM; pp. 785–794. [Google Scholar]
- Lundberg, S.M.; Lee, S.I. A unified approach to interpreting model predictions. Adv Neural Inf Process Syst. 2017, 30, 4765–4774. [Google Scholar]
- Vaswani, A.; Shazeer, N.; Parmar, N.; et al. Attention is all you need. Adv Neural Inf Process Syst. 2017, 30, 5998–6008. [Google Scholar]
- Kalman, R.E. A new approach to linear filtering and prediction problems. J Basic Eng. 1960, 82(1), 35–45. [Google Scholar] [CrossRef]





| Model | Test Accuracy | Macro-F1 | Inference (ms/batch) |
| LSTM | 0.6485 | 0.6213 | 29.43 |
| GRU | 0.6485 | 0.6198 | 51.51 |
| CNN-LSTM | 0.5909 | 0.5629 | 79.75 |
| ResNet + XGBoost | 0.7879 | 0.7347 | 0.53 |
| Feature-Transformer + LightGBM | 0.8242 | 0.7735 | 1.58 |
| Variant | Test Accuracy | Macro-F1 |
| Transformer only | 0.6121 | 0.5894 |
| LightGBM (tabular only) | 0.5909 | 0.5672 |
| Feature-Transformer + LightGBM (full) | 0.8242 | 0.7735 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.