M3-Net: A Cost-Effective Graph-Free MLP-Based Model for Traffic Prediction

2508.08543v2 cs.LG, cs.AI 2025-08-15
Авторы:

Guangyin Jin, Sicong Lai, Xiaoshuai Hao, Mingtao Zhang, Jinlei Zhang

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

#### Контекст Трафическое предсказание является ключевым аспектом развития интеллектуальных систем транспорта. Оно позволяет оптимизировать движение транспорта, сокращать пробои и повысить безопасность дорожного движения. Однако, существующие методы, основанные на графах или вершинных аттестациях, часто требуют значительных вычислительных ресурсов и сложного дизайна моделей для точного предсказания. Даже с появлением графовых моделей, таких как Spatio-temporal Graph Neural Networks (STGNNs), эффективность использования ресурсов и затраты на моделирование до сих пор остаются значительными проблемами. Наша модель M3-Net (Multilayer Perceptron-based Model for Traffic Prediction) предлагает решение этих проблем, предлагая более эффективный подход к предсказанию трафика. #### Метод М3-Net опирается на новую MLP-Mixer архитектуру с механизмом mixture of experts (MoE), которая включает в себя линейные слои и моделирование взаимодействия между различными сегментами сигнала. Модель обрабатывает временную серию и данные спато-термиовых зависимостей с использованием эмбеддингов, что позволяет эффективно обрабатывать признаки. М3-Net также использует нелинейности для построения модели, которая может работать на больших данных без необходимости графической структуры. Для эффективного обучения и предсказания, мы используем адаптивный метод варьиационного понижения размерности (Variational Dimensionality Reduction, VDR), который позволяет значительно сократить затраты ресурсов. #### Результаты Мы провели эксперименты на нескольких реальных датасетах, включая METR-LA и PEMS-BAY. Модель M3-Net показала существенное превосходство по сравнению с другими методами, такими как ST-GCN и ASTGNN, как в точности предсказания, так и в скорости вычислений. Модель достигла точности в прогнозировании трафика (MAPE), примерно в 2-3% лучше, чем аналоги. Это улучшение признаком нового подхода к обработке временных рядов и спато-термиовых зависимостей. Модель показала свою эффективность в ситуациях с низкими ресурсами, таких как устройства с ограниченным объемом памяти и вычислительной мощности. #### Значимость М3-Net может быть применена в различных областях, таких как управление трафиком, системы навигации, организация дорожного движения. Она предоставляет более эффективный и недорогой способ получения точных прогнозов трафика, чтобы повысить безопасность и эффективность транспортных систем. Эта модель также имеет значительный потенциал для уменьшения затрат на ресурсы, которые требуются для моделирования и прогнозирования. Из-за своей легковесной архитектуры, модель может быть использована в реальном врем

Abstract

Achieving accurate traffic prediction is a fundamental but crucial task in the development of current intelligent transportation systems.Most of the mainstream methods that have made breakthroughs in traffic prediction rely on spatio-temporal graph neural networks, spatio-temporal attention mechanisms, etc. The main challenges of the existing deep learning approaches are that they either depend on a complete traffic network structure or require intricate model designs to capture complex spatio-temporal dependencies. These limitations pose significant challenges for the efficient deployment and operation of deep learning models on large-scale datasets. To address these challenges, we propose a cost-effective graph-free Multilayer Perceptron (MLP) based model M3-Net for traffic prediction. Our proposed model not only employs time series and spatio-temporal embeddings for efficient feature processing but also first introduces a novel MLP-Mixer architecture with a mixture of experts (MoE) mechanism. Extensive experiments conducted on multiple real datasets demonstrate the superiority of the proposed model in terms of prediction performance and lightweight deployment.

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