Skip to main navigation Skip to search Skip to main content

Messages are Never Propagated Alone: Collaborative Hypergraph Neural Network for Time-Series Forecasting

  • Nan Yin
  • , Li Shen
  • , Huan Xiong
  • , Bin Gu
  • , Chong Chen
  • , Xian Sheng Hua
  • , Siwei Liu
  • , Xiao Luo*
  • *Corresponding author for this work
  • JD Explore Academy
  • Mohamed Bin Zayed University of Artificial Intelligence
  • Jilin University
  • Terminus Group Co., Ltd.
  • University of California at Los Angeles

Research output: Contribution to journalArticlepeer-review

Abstract

This paper delves into the problem of correlated time-series forecasting in practical applications, an area of growing interest in a multitude of fields such as stock price prediction and traffic demand analysis. Current methodologies primarily represent data using conventional graph structures, yet these fail to capture intricate structures with non-pairwise relationships. To address this challenge, we adopt dynamic hypergraphs in this study to better illustrate complex interactions, and introduce a novel hypergraph neural network model named CHNN for correlated time series forecasting. In more detail, CHNN leverages both semantic and topological similarities via an interaction model and hypergraph diffusion process, thereby constructing comprehensive collaborative correlation scores that effectively guide spatial message propagation. In addition, it incorporates short-term temporal information to generate efficient spatio-temporal feature maps. Lastly, a long-term temporal module is proposed to generate future predictions utilizing both temporal attention and a gated recurrent network. Comprehensive experiments conducted on four real-world datasets, i.e., Tiingo, Stocktwits, NYC-Taxi, and Social Network demonstrate that the proposed CHNN markedly outperforms a range of benchmark methods.

Original languageEnglish
Pages (from-to)2333-2347
Number of pages15
JournalIEEE Transactions on Pattern Analysis and Machine Intelligence
Volume46
Issue number4
DOIs
StatePublished - 1 Apr 2024

Keywords

  • Correlated time series forecasting
  • hypergraph
  • hypergraph neural network

Fingerprint

Dive into the research topics of 'Messages are Never Propagated Alone: Collaborative Hypergraph Neural Network for Time-Series Forecasting'. Together they form a unique fingerprint.

Cite this