https://github.com/hkuds/graphst

[ICML'2023] "GraphST: Spatial-Temporal Graph Learning with Adversarial Contrastive Adaptation"
https://github.com/hkuds/graphst

contrastive-learning graph-neural-networks graph-representation-learning spatio-temporal-prediction urban-computing

Last synced: 9 months ago
JSON representation

[ICML'2023] "GraphST: Spatial-Temporal Graph Learning with Adversarial Contrastive Adaptation"

• Host: GitHub
• URL: https://github.com/hkuds/graphst
• Owner: HKUDS
• Created: 2023-05-25T06:43:11.000Z (about 3 years ago)
• Default Branch: main
• Last Pushed: 2024-12-24T08:40:15.000Z (over 1 year ago)
• Last Synced: 2024-12-24T09:33:13.895Z (over 1 year ago)
• Topics: contrastive-learning, graph-neural-networks, graph-representation-learning, spatio-temporal-prediction, urban-computing
• Language: Python
• Homepage: https://arxiv.org/abs/2306.10683
• Size: 2.08 MB
• Stars: 49
• Watchers: 4
• Forks: 2
• Open Issues: 3
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
![Framework](./pictures/framework.png)

# Initial Data link

https://drive.google.com/drive/folders/1heuy-28olym0Tn4bkMaJTABVF9kXMZNe?usp=sharing

# Spatial-Temporal Graph Learning with Adversarial Contrastive Adaptation #

This is the implementation of Spatial-Temporal Graph Learning with Adversarial Contrastive Adaptation (ICML'23) in the following paper:

## Requirements ##

Pytorch = 1.7.0 and Tensorflow = 1.15.3 (crime prediction task (ST-SHN))

## Data ##

Moblity data, crime data, census block data, POI data, house price of Chicago ([https://data.cityofchicago.org/](https://data.cityofchicago.org/)) and New York([https://opendata.cityofnewyork.us/](https://opendata.cityofnewyork.us/)).

Also, we provide the process data and processing code for data preprocessing.

## Data Preocessing ##

    cd data_pre

    pre_s1.py # collecing positions of 180 (234 regions) in Manhatton (a certain district) in New York (Chicago).

    pre_s2.py # collecting traffic data

    pre_s3.py # preprocessing the traffic data on correspongding regions

    pre_s4.py # preprocessing the POI data and obtain POI-aware Region Graph

    pre_poi_transformer.py # obatining the features of nodes by Transformer and Skip-gram

    pre_spatial_graph.py # constructing the trajectory-based Regoin Graph and Distance-based Region Graph

    pre_s5.py # constructing a hierarical graph

    pre_s6_dataloader.py # obtaining the dataloder (dataset.pt file) for next step to get region embeddings

    cd house

    pre_s7.py # processing house data

## Hyperparameters ##

The dimensionality d of region representation is set as 96 . The depth of convolutional layers in GCN is set as 2. The learning rate is initialized as 0.001 with the weight decay of 1e-5. The number of hidden units is 256. The number of projection hidden is 256. The eps is set as 0.5. And alpha is set as.1, beta is set as 0.1; the lamb is set as 0.05. For the crime prediction backbone model, ST-SHN is configured with the learning rate of 0.001 and the weight decay of 0.96. The depth of the spatial path aggregation layers is set as 2. For the traffic prediction backbone model ST-GCN, the historical time window of all tests are set as 60 minutes with 12 observed data points that are utilized to forecast traffic conditions in the next 15, 30, 45 minutes. The baselines are implemented with the source code released by their original papers. We further apply the grid search strategy for parameter tuning in baselines to achieve their best performance. 

## Spatial-Temporal Adversarial Graph Model (STAG) Training and Obtaining Region Representations ##

    train_edit_auto.py   # training for obatining region representations

## TASK 1: Crime Prediction Task ##

The code of the ST-SHN is [https://github.com/akaxlh/ST-SHN](https://github.com/akaxlh/ST-SHN)

Replace the area embedding with the region represenation vetcor and run `HG_ST_labcode.py`.

## TASK 2:  Traffic Prediction Task ##

The code of the ST-GCN is [https://github.com/VeritasYin/STGCN_IJCAI-18](https://github.com/VeritasYin/STGCN_IJCAI-18)

ST-GCN predicts traffic in one stage, namely 15 minutes.

Run `main.py`.

## TASK 3: House Prediction Prediction Task ##

We use Lasso Regression to perform house price evaluation task.

Just `cd house` and run `pre_s9.py` and got the regression result.

![Prediction Results](./pictures/result.png)

## Citing ##

If our paper benefits to your research, please cite our paper using the bitex below:

    @inproceedings{zhang2023spatial,

      title={Spatial-Temporal Graph Learning with Adversarial Contrastive Adaptation},

      author={Zhang, Qianru and Huang, Chao and Xia, Lianghao and Wang, Zheng and Yiu, Siu Ming and Han, Ruihua},

      booktitle={International Conference on Machine Learning},

      pages={41151--41163},

      year={2023},

      organization={PMLR}

    }