ProtoHG

Implementation for paper: Prototype-Enhanced Hypergraph Learning for Heterogeneous Information Networks. International Conference on Multimedia Modeling(MMM) Oral 2024 [paper]

Pipeline

Requirements

The variety and complexity of relations in multimedia data lead to Heterogeneous Information Networks (HINs). Capturing the semantics from such networks requires approaches capable of utilizing the full richness of the HINs. Existing methods for modeling HINs employ techniques originally designed for graph neural networks, and HINs decomposition analysis, like using manually predefined metapaths. In this paper, we introduce a novel prototype-enhanced hypergraph learning approach for node classification in HINs. Using hypergraphs instead of graphs, our method captures higher-order relationships among nodes and extracts semantic information without relying on metapaths. Our method leverages the power of prototypes to improve the robustness of the hypergraph learning process and creates the potential to provide human-interpretable insights into the underlying network structure. Extensive experiments on three real-world HINs demonstrate the effectiveness of our method.

Requirements

2.1 Main python packages and their version

• Python3
• torch 1.13.1+cu116
• torch-scatter 2.1.1+pt113cu117
• torch-sparse 0.6.17+pt113cu117

Hardware

• 1 NVIDIA A100

Datasets

ACM, DBLP, WikiArt

Structure

|- data_hete
    |- ACM_hete
    |- DBLP_hete
    |- wikiart_hete
|- model
    networks.py
    layer.py
main_transformer.py
utils.py

You can download the dataset from HGB_hyper_data

Training

taking ACM dataset as an example:

python main_transformer.py --data HGB_hyper_data --dataset ACM_hete --num_hidden 64 --n_head 4 --n_layer 3 --reg_p 1e-6 --loss sim --lr 0.01 --seeds 0 1 2 3 4  --wandb

Acknowledgement

This repo is based on HyperSage and HEGEL: Hpyergraph Transformer, thanks for their excellent work.

If you find this repo useful, please consider cite:

@InProceedings{10.1007/978-3-031-53311-2_34,
author="Wang, Shuai
and Shen, Jiayi
and Efthymiou, Athanasios
and Rudinac, Stevan
and Kackovic, Monika
and Wijnberg, Nachoem
and Worring, Marcel",
title="Prototype-Enhanced Hypergraph Learning for Heterogeneous Information Networks",
booktitle="MultiMedia Modeling",
year="2024",
publisher="Springer Nature Switzerland",
}