Papers › A Time Series is Worth Five Experts: Heterogeneous Mixture of Experts for Traffic Flow...

A Time Series is Worth Five Experts: Heterogeneous Mixture of Experts for Traffic Flow Prediction

26 Sep 2024arXiv:2409.17440archive 2025-07-28

Guangyu Wang, Yujie Chen, Ming Gao, Zhiqiao Wu, Jiafu Tang, Jiabi Zhao

Accurate traffic prediction faces significant challenges, necessitating a deep understanding of both temporal and spatial cues and their complex interactions across multiple variables. Recent advancements in traffic prediction systems are primarily due to the development of complex sequence-centric models. However, existing approaches often embed multiple variables and spatial relationships at each time step, which may hinder effective variable-centric learning, ultimately leading to performance degradation in traditional traffic prediction tasks. To overcome these limitations, we introduce variable-centric and prior knowledge-centric modeling techniques. Specifically, we propose a Heterogeneous Mixture of Experts (TITAN) model for traffic flow prediction. TITAN initially consists of three experts focused on sequence-centric modeling. Then, designed a low-rank adaptive method, TITAN simultaneously enables variable-centric modeling. Furthermore, we supervise the gating process using a prior knowledge-centric modeling strategy to ensure accurate routing. Experiments on two public traffic network datasets, METR-LA and PEMS-BAY, demonstrate that TITAN effectively captures variable-centric dependencies while ensuring accurate routing. Consequently, it achieves improvements in all evaluation metrics, ranging from approximately 4.37\% to 11.53\%, compared to previous state-of-the-art (SOTA) models. The code is open at \href{https://github.com/sqlcow/TITAN}{https://github.com/sqlcow/TITAN}.

PaperPDFConference PDFCode

Code

sqlcow/TITAN officialmentioned in paperpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Mixture-of-ExpertsPredictionTime SeriesTraffic Prediction

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Traffic Prediction METR-LA TITAN 12 steps MAE 3.08 #1 of 20 Archive leaderboard report
Traffic Prediction METR-LA TITAN 12 steps MAPE 8.43 #1 of 20 Archive leaderboard report
Traffic Prediction METR-LA TITAN 12 steps RMSE 6.21 #1 of 20 Archive leaderboard report
Traffic Prediction METR-LA TITAN MAE @ 12 step 3.08 #1 of 20 Archive leaderboard report
Traffic Prediction METR-LA TITAN MAE @ 3 step 2.41 #1 of 20 Archive leaderboard report
Traffic Prediction PEMS-BAY TITAN MAE @ 12 step 1.69 #2 of 16 Archive leaderboard report
Traffic Prediction PEMS-BAY TITAN RMSE 3.79 #2 of 16 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections