8000 GitHub - usail-hkust/t-PatchGNN: Official implementation for ICML24 paper "Irregular Multivariate Time Series Forecasting: A Transformable Patching Graph Neural Networks Approach"
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Official implementation for ICML24 paper "Irregular Multivariate Time Series Forecasting: A Transformable Patching Graph Neural Networks Approach"

Notifications You must be signed in to change notification settings

usail-hkust/t-PatchGNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

54 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is an official implementation of ICML 24 paper Irregular Multivariate Time Series Forecasting: A Transformable Patching Graph Neural Networks Approach.

IMTS Forecasting

As illustrated in the figure, given a set of historical Irregular Multivariate Time Series (IMTS) observations and forecasting queries, the IMTS forecasting problem aims to accurately forecast the values in correspondence to these queries.

problem

News 🎉

🌟 2025-04: Good news! The MIMIC dataset used in this work has been published in the official PhysioNet databases. You can directly apply the processed dataset at MIMIC-III-Ext-tPatchGNN to replicate the experiments or for your research.

New Benchmark

We build a new comprehensive benchmark for the problem of Irregular Multivariate Time Series Forecasting, including four scientific datasets covering areas of healthcare, biomechanics, and climate science.

For Physionet, Human Activity, and USHCN, we have provided the processed datasets to ease the reproduction.

For MIMIC, because of the PhysioNet Credentialed Health Data License, we have published the dataset in the official PhysioNet databases. You can directly apply the processed dataset at MIMIC-III-Ext-tPatchGNN to replicate the experiments.

The main results:

results

t-PatchGNN

The overview of t-PatchGNN, which initially divides each univariate irregular time series into a series of transformable patches with varying number of consecutive observations but maintains a unified time horizon resolution. Then the patching outcomes can be seamlessly modeled by Transformer and time-adaptive GNNs, which incorporate the time-varying adaptive graph structure learning (GSL), to realize an effective intra- and inter-time series modeling for IMTS.

model

Requirements

t-PatchGNN has been tested using Python 3.9 and CUDA Version: 12.2

To have consistent libraries and their versions, you can install the needed dependencies for this project by running the following command:

pip install -r requirements.txt

Run the Model

For the specific configurations utilized to obtain the principal experimental outcomes presented in the paper, kindly refer to the script "run_all.sh". To replicate these results, please execute the below command.

sh ./tPatchGNN/scripts/run_all.sh

Example:

python run_models.py \
    --dataset {dataset} --state {def_or_debug} --history {length_of_observation}\
    --patience {patience_of_earlystopping} --batch_size {batch_size} --lr {learning_rate} \
    --patch_size {window_size_for_a_patch} \
    --stride {period_stride_for_patch_sliding} \
    --nhead {heads_in_Transformer} \
    --tf_layer{number_of_layer_in_Transformer} \
    --nlayer {num_of_layer_in_Time_Series_Model} \
    --te_dim {number_of_units_for_time_encoding} \
    --node_dim {number_of_units_for_node_vectors} \
    --hid_dim {number_of_units_per_hidden_layer} \
    --outlayer {model_for_outlayer} --seed {seed} --gpu {gpu}
  • dataset: the dataset name, select from [physionet, mimic, activity, ushcn].
  • seed: the seed for parameter initialization.
  • history: the length of the time for observation, the rest will be used for forecasting. Please note that different datasets have varying time spans and levels of granularity.

Citation

@inproceedings{zhangirregular2024,
  title={Irregular Multivariate Time Series Forecasting: A Transformable Patching Graph Neural Networks Approach},
  author={Zhang, Weijia and Yin, Chenlong and Liu, Hao and Zhou, Xiaofang and Xiong, Hui},
  booktitle={Forty-first International Conference on Machine Learning},
  year={2024}
}

About

Official implementation for ICML24 paper "Irregular Multivariate Time Series Forecasting: A Transformable Patching Graph Neural Networks Approach"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published
0