iTransMamba时空预测网络

轻量级长时预测

一种结合Transformer编码器和选择性状态空间模型的轻量级时空网络,用频域通道混合增强多通道信息交互,以低于600万参数实现高精度长期交通流量预测。

Jianrong, Cao · Xing, Sheng · Jun-Zhe, Zhang · 段宗涛

Knowledge-Based Systems 2025

参数信息

参数量
<6 M
浮点运算成本
27 G

技术优势

通道混合更高效

频域通道混合模块利用Einstein矩阵增强通道相关性,实现最佳混合,解决了Transformer模型中通道混合策略效率低下的问题。

过滤冗余历史数据

解码器中的选择性状态空间模型(SSM)机制能够过滤掉冗余历史数据,高效压缩长期信息,避免无关信息对预测的干扰。

计算成本大幅降低

模型参数量低于6M,浮点运算成本为27G,显著低于其他SOTA模型,使得长期交通流量预测在资源受限环境下更可行。

应用场景