GSMT: Graph Fusion and Spatiotemporal TaskCorrection for Multi-Bus Trajectory Prediction

2508.09227v1 cs.LG, cs.AI, cs.CE 2025-08-15
Авторы:

Fan Ding, Hwa Hui Tew, Junn Yong Loo, Susilawati, LiTong Liu, Fang Yu Leong, Xuewen Luo, Kar Keong Chin, Jia Jun Gan

Резюме на русском

## Контекст Городской транспорт, особенно автобусные линии, является ключевым компонентом урбанизированных транспортных систем. Однако точность прогноза маршрутов автобусов остается вызовом, особенно в районах с ограниченным доступом к многомодальным данным. Оставшийся вариант — использование данных GPS, но они имеют свои ограничения в точности и надежности. Наша мотивация заключается в разработке модели, которая использует доступные данные GPS и дополняет их с помощью расширенных методов машинного обучения для корректного прогнозирования маршрутов. ## Метод Мы предлагаем GSMT (Graph Fusion and Spatiotemporal Task Correction), комбинацию Graph Attention Network (GAT) и Sequence-to-Sequence Recurrent Neural Network (RNN). GAT извлекает специфические локальные зависимости данных, а RNN обрабатывает последовательности данных GPS. Более того, GSMT включает в себя Task Corrector, который выполняет детальный анализ и сортировку исторических данных путей, чтобы выявить различные модели движения и улучшить прогнозы. Этот двухэтапный подход позволяет объединять динамические и статические данные, улучшая точность и универсальность прогнозов маршрутов. ## Результаты Мы проверили GSMT на реальных данных города Куала-Лумпур. Модель была тестирована в условиях сложной тактической среды, где были проведены сравнения с другими популярными моделями. Результаты показали, что GSMT не только превосходит конкуренты в краткосрочном и долгосрочном прогнозировании маршрутов, но и достигает высокой точности в условиях проведения экспериментов, демонстрируя выигрыш в качестве прогнозов на несколько процентов. ## Значимость Наша модель может быть применима в различных ситуациях, где необходимо точно прогнозировать траектории движения, например, для управления транспортом, мониторинга дорожного движения и планирования маршрутов. Основные преимущества GSMT заключаются в ее универсальности, модифицируемости и высокой производительности в сложных транспортных средах. Это может привести к улучшению управления транспортом и к более эффективной планировке городских маршрутов. ## Выводы GSMT доказала свою эффективность в таске прогнозирования маршрутов автобусов в тяжелых городских условиях. Мы планируем расширить модель, внедрив более сложные механизмы обучения и выполнив эксперименты на более широком множестве данных, чтобы продемонстрировать его универсальность и надежность в разных областях.

Abstract

Accurate trajectory prediction for buses is crucial in intelligent transportation systems, particularly within urban environments. In developing regions where access to multimodal data is limited, relying solely on onboard GPS data remains indispensable despite inherent challenges. To address this problem, we propose GSMT, a hybrid model that integrates a Graph Attention Network (GAT) with a sequence-to-sequence Recurrent Neural Network (RNN), and incorporates a task corrector capable of extracting complex behavioral patterns from large-scale trajectory data. The task corrector clusters historical trajectories to identify distinct motion patterns and fine-tunes the predictions generated by the GAT and RNN. Specifically, GSMT fuses dynamic bus information and static station information through embedded hybrid networks to perform trajectory prediction, and applies the task corrector for secondary refinement after the initial predictions are generated. This two-stage approach enables multi-node trajectory prediction among buses operating in dense urban traffic environments under complex conditions. Experiments conducted on a real-world dataset from Kuala Lumpur, Malaysia, demonstrate that our method significantly outperforms existing approaches, achieving superior performance in both short-term and long-term trajectory prediction tasks.

Ссылки и действия