Feature-enhanced iTransformer: A two-stage framework for high-accuracy long-horizon traffic flow forecasting

功能增强型 iTransformer:用于高精度长期交通流量预测的两阶段框架

阅读:2

Abstract

Accurate and reliable long-horizon traffic flow prediction is a cornerstone of modern Intelligent Transportation Systems (ITS), yet it remains challenging due to the complex, non-linear, and dynamic spatio-temporal dependencies inherent in traffic data. While recent Transformer-based models have shown promise, they are typically end-to-end systems that couple feature extraction and sequence prediction, which can limit their ability to fully leverage multi-faceted domain information. To address this, we propose a two-stage framework, the Feature-Enhanced iTransformer (FE-iTransformer), founded on an extract-and-enhance philosophy. The framework first employs a comprehensive Feature Enhancement Module (FEM) to distill a global context vector from spatio-temporal dynamics, periodic patterns, and temporal context-without relying on a predefined graph structure. Subsequently, an innovative per-step feature enhancement mechanism uses this global vector to enrich the original input sequence, yielding an information-rich representation that is then processed by a strong iTransformer backbone for final prediction. The effectiveness of FE-iTransformer is validated through extensive experiments: ablation studies on two classic datasets (Freeway and Urban) provide compelling evidence for the efficacy of the two-stage design, demonstrating that introducing FEM significantly improves the pure iTransformer backbone; supplementary experiments on the large-scale PEMS08 benchmark further confirm scalability and long-horizon performance, reducing Mean Absolute Error (MAE) by 19.1% over the vanilla backbone in the 120-minute forecasting task. Importantly, this study targets no-graph/weak-graph settings and does not aim to surpass graph-prior models; rather, it offers a deployment-ready, graph-free alternative when the roadway graph is unavailable or unreliable.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。