Skip to content
/ HIC Public

Source code of IEEE TPAMI 2024 "Hypergraph Isomorphism Computation"

License

Notifications You must be signed in to change notification settings

iMoonLab/HIC

Repository files navigation

Hypergraph Isomorphism Computation

This repository contains the code for the paper "Hypergraph Isomorphism Computation" published in IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI) 2024 by Yifan Feng, Jiashu Han, Shihui Ying, and Yue Gao*. The paper is available at here.

algorithm

Introduction

In this repository, we provide our implementation of Hypergraph WL Subtree Kernel, Hypergraph WL Hyperedge Kernel and some compared methods including Graph Subtree Kernel, Graphlet Kernel, Hypergraph Directed Line Kernel, and Hypergraph Rooted Kernel. The implementation is based on the following libraries:

Installation

  1. Clone this repository.
  2. Install the required libraries.
pip install -r requirements.txt

Usage

Modify the root path in ml_config.yaml to the absolute path of the data folder in this repository. Then, run the following command to reproduce the results in the paper:

python ml_main.py

You can change the name of model and dataset in ml_config.yaml to reproduce the results of other models and datasets. All available models and datasets are listed in the following:

Models

Datasets

  • Graph Classification Datasets: RG_macro, RG_sub, IMDBBINARY, IMDBMULTI, MUTAG, NCI1, PROTEINS
  • Hypergraph Classification Datasets: RHG_3, RHG_10, RHG_table, RHG_pyramid, IMDB_dir_form, IMDB_dir_genre, IMDB_wri_form, IMDB_wri_genre, IMDB_dir_genre_m, IMDB_wri_genre_m, stream_player, twitter_friend

Citation

If you find this repository useful in your research, please cite our following papers:

@article{feng2024hypergraph,
  title={Hypergraph Isomorphism Computation},
  author={Feng, Yifan and Han, Jiashu and Ying, Shihui and Gao, Yue},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
  year={2024},
  publisher={IEEE}
}

@article{gao2022hgnn+,
  title={HGNN+: General hypergraph neural networks},
  author={Gao, Yue and Feng, Yifan and Ji, Shuyi and Ji, Rongrong},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
  volume={45},
  number={3},
  pages={3181--3199},
  year={2022},
  publisher={IEEE}
}

About

Source code of IEEE TPAMI 2024 "Hypergraph Isomorphism Computation"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages