Skip to content

[IJCAI 2024] This it the official github for IJCAI24 paper "Multi-Modality Spatio-Temporal Forecasting via Self-Supervised Learning"

Notifications You must be signed in to change notification settings

beginner-sketch/MoSSL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

61 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

[IJCAI 2024] Multi-Modality Spatio-Temporal Forecasting via Self-Supervised Learning

[IJCAI24] Jiewen Deng, Renhe Jiang, Jiaqi Zhang, Xuan Song, "Multi-Modality Spatio-Temporal Forecasting via Self-Supervised Learning", IJCAI, 2024.

Our research has been accepted for presentation at the main track of IJCAI 2024.

This implementation showcases our MoSSL model.

image

Preprint Link

arXiv

Citation

Citation details will be updated once the official proceedings for IJCAI 2024 are available online.

@inproceedings{ijcai2024p223,
  title     = {Multi-Modality Spatio-Temporal Forecasting via Self-Supervised Learning},
  author    = {Deng, Jiewen and Jiang, Renhe and Zhang, Jiaqi and Song, Xuan},
  booktitle = {Proceedings of the Thirty-Third International Joint Conference on
               Artificial Intelligence, {IJCAI-24}},
  publisher = {International Joint Conferences on Artificial Intelligence Organization},
  editor    = {Kate Larson},
  pages     = {2018--2026},
  year      = {2024},
  month     = {8},
  note      = {Main Track},
  doi       = {10.24963/ijcai.2024/223},
  url       = {https://doi.org/10.24963/ijcai.2024/223},
}

Multi-Modality Spatio-Temporal Dataset

NYC Traffic Demand dataset[1] is collected from the New York City, which consists of 98 nodes and four transportation modalities: Bike Inflow, Bike Outflow, Taxi Inflow, and Taxi Outflow. The timespan is from April 1st, 2016 to June 30th, 2016, and the time interval is set to half an hour.

BJ Air Quality dataset[2] is collected from the Beijing Municipal Environmental Monitoring Center, which contains 10 nodes and three pollutant modalities: $PM_{2.5}$, $PM_{10}$, and $SO_2$. The timespan is from March 1st, 2013 to February 28th, 2017, and the time interval is set to one hour.

1. ^ https://ride.citibikenyc.com/system-data; https://www1.nyc.gov/site/tlc/about/tlc-trip-record-data.page

2. ^ https://archive.ics.uci.edu/ml/datasets/Beijing+Multi-Site+Air-Quality+Data

Installation Dependencies

Python 3 (>= 3.6; Anaconda Distribution)

PyTorch (>= 1.6.0) 

Numpy >= 1.17.4

Pandas >= 1.0.3

Model Training and Testing

python traintest_MoSSL.py cuda_id

About

[IJCAI 2024] This it the official github for IJCAI24 paper "Multi-Modality Spatio-Temporal Forecasting via Self-Supervised Learning"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages