LLM2D
AutoSTF:面向成本效益的自动时空预测解耦神经架构搜索
AutoSTF: Decoupled Neural Architecture Search for Cost-Effective Automated Spatio-Temporal Forecasting
作者: Tengfei Lyu, Weijia Zhang, Jinliang Deng, Hao Liu
发布日期: 9/26/2024
arXiv ID: oai:arXiv.org:2409.16586v1

摘要

时空预测是各种智慧城市应用的关键组成部分,例如交通优化、能源管理和社会经济分析。近年来,人们提出了几种自动时空预测方法,以自动搜索最优的神经网络架构来捕捉复杂的时空依赖关系。然而,现有的自动方法存在神经架构搜索开销过高的缺陷,这阻碍了它们的实际应用以及对更细粒度的各种时空算子的进一步探索。本文提出了一种名为AutoSTF的解耦自动神经架构搜索框架,用于实现经济高效的自动时空预测。从效率的角度来看,我们首先将混合搜索空间解耦为时间空间和空间空间,并分别设计了表示压缩和参数共享方案来缓解参数爆炸。解耦时空搜索不仅加快了模型优化过程,而且为更有效的时空依赖建模留下了新的空间。从有效性的角度来看,我们提出了一种多块迁移模块,以联合捕捉多粒度时间依赖关系,并扩展了空间搜索空间,以实现更细粒度的逐层空间依赖搜索。在八个数据集上的大量实验表明了AutoSTF在准确性和效率方面的优越性。具体而言,我们提出的方法与最先进的自动时空预测方法相比,速度提高了高达13.48倍,同时保持了最佳的预测精度。