arXivDaily arXiv每日学术速递 周一至周五更新
今晚才能更新了~~大家休息下再来吧
arXiv 2302.08643cs.LG
收藏

快速时序小波图神经网络

Fast Temporal Wavelet Graph Neural Networks

FPT Software AI Center(FPT软件人工智能中心);Indiana State University(印第安纳州立大学);University of Chicago(芝加哥大学)

Duc Thien Nguyen, Manh Duc Tuan Nguyen, Truong Son Hy, Risi Kondor

AI总结:

本文提出快速时序小波图神经网络FTWGNN,利用多分辨率矩阵分解构建稀疏小波基和快速小波卷积,在交通和脑电数据集上以低计算开销取得与最先进方法相当的预测性能。

摘要:

时空信号预测在众多领域中发挥着重要作用,尤其是在神经科学与交通运输领域。由于网络具有高度复杂的空间结构以及非线性的时序动态特性,该任务极具挑战性。为了对人类大脑和交通网络实现可靠且及时的预测,我们提出了快速时序小波图神经网络(FTWGNN),该方法在具有底层图结构的时序数据学习任务中兼具时间和内存效率,这得益于多分辨率分析理论以及离散空间上的小波理论。我们采用多分辨率矩阵分解(MMF)(Kondor et al., 2014)对高度稠密的图结构进行分解,并计算相应的稀疏小波基,从而能够构建快速小波卷积作为我们新型架构的主干。在真实世界的PEMS-BAY、METR-LA交通数据集和AJILE12 ECoG数据集上的实验结果表明,FTWGNN在保持较低计算开销的同时,与最先进的方法相比具有竞争力。我们的PyTorch实现已在 https://github.com/HySonLab/TWGNN 公开提供。

英文摘要:

Spatio-temporal signals forecasting plays an important role in numerous domains, especially in neuroscience and transportation. The task is challenging due to the highly intricate spatial structure, as well as the non-linear temporal dynamics of the network. To facilitate reliable and timely forecast for the human brain and traffic networks, we propose the Fast Temporal Wavelet Graph Neural Networks (FTWGNN) that is both time- and memory-efficient for learning tasks on timeseries data with the underlying graph structure, thanks to the theories of multiresolution analysis and wavelet theory on discrete spaces. We employ Multiresolution Matrix Factorization (MMF) (Kondor et al., 2014) to factorize the highly dense graph structure and compute the corresponding sparse wavelet basis that allows us to construct fast wavelet convolution as the backbone of our novel architecture. Experimental results on real-world PEMS-BAY, METR-LA traffic datasets and AJILE12 ECoG dataset show that FTWGNN is competitive with the state-of-the-arts while maintaining a low computational footprint. Our PyTorch implementation is publicly available at https://github.com/HySonLab/TWGNN

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑