EMAGN: Efficient Multi-Attention Graph Network via Learned Clustering for Scalable Traffic Forecasting
EMAGN:基于学习聚类的高效多注意力图网络用于可扩展交通流量预测
机构 * Shanghai Jiao Tong University(上海交通大学) ; University of Michigan, Ann Arbor(密歇根大学安娜堡分校) ; University of California, Berkeley(加利福尼亚大学伯克利分校) ; Cornell University(康奈尔大学) ; Technische Universität Dresden(德累斯顿工业大学) ; Carnegie Mellon University(卡内基梅隆大学) ; The University of Texas at Austin(德克萨斯大学奥斯汀分校) ; University of Wisconsin–Madison(威斯康星大学麦迪逊分校)
AI总结 针对交通流量预测中自注意力机制扩展性有限的问题,提出EMAGN,通过学习聚类矩阵将空间注意力机制线性化,降低复杂度,实验表明其在准确性和效率上优于其他模型,扩展了可行模型配置。