atlas

Research map / Label-efficient and robust learning / Self-supervised and contrastive

HypeBoy: Generative Self-Supervised Representation Learning on Hypergraphs

ICLR 2024 · Label-efficient and robust learning › Self-supervised and contrastive · approach: Invariance-based self-supervised learning · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes hyperedge filling, a generative SSL task for hypergraphs (predict a missing node from the rest of a hyperedge), with theory linking it to node classification, and the HypeBoy method with a feature-reconstruction warm-up.

Problem

Designing generative self-supervised learning for hypergraphs is hard: the space of possible hyperedges grows as 2^|V|, and it is unclear what generative SSL task should be used and how it relates to downstream node classification.

Approach

Reformulate generative hypergraph SSL as hyperedge filling: predict the missing node given the other nodes of a hyperedge, with a softmax over all nodes (size |V| rather than 2^|V|), using cosine similarity between projected node and set embeddings, plus augmentation, projection heads, and a masked feature-reconstruction warm-up.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
pre-train the UniGCNII encoder with hyperedge filling (HypeBoy) instead of using UniGCNII without SSL pre-trainingUniGCNII · node classification, 10 of 11 datasetsaccuracy: performance gains up to 12.5 pointsHelped
use HypeBoy rather than TriCL for pre-traininghypergraph SSL node classification · node classification, 11 benchmark datasetsaccuracy: accuracy gap is up to 5.0 pointsMixed
add feature reconstruction warm-up to hyperedge filling without projection heads (V1 to V4) (ablation)HypeBoy · node classification, 11 benchmark datasetsaverage ranking: 4.3 -> 2.5Helped
add projection heads to hyperedge filling without feature reconstruction warm-up (V1 to V2) (ablation)HypeBoy · node classification, 11 benchmark datasetsaverage ranking: 4.3 -> 3.3Helped
add projection heads to HypeBoy with feature reconstruction warm-up (V4 to HypeBoy) (ablation)HypeBoy · node classification, 11 benchmark datasetsaverage ranking: 2.5 -> 1.4Helped
add hyperedge filling to feature reconstruction only (V3 to HypeBoy) (ablation)HypeBoy · node classification, 11 benchmark datasetsaverage ranking: 3.6 -> 1.4Helped
add feature reconstruction warm-up to hyperedge filling with projection heads (V2 to HypeBoy) (ablation)HypeBoy · Cora node classificationaccuracy: 59.7 ({\scriptsize 9.2}) -> 62.3 ({\scriptsize 7.7})Mixed
use HypeBoy rather than TriCL for fixed-embedding node classificationhypergraph SSL representations · node classification, 11 benchmark datasetsaccuracy: accuracy gap is up to 6.3 pointsMixed
use HypeBoy rather than other SSL methods to learn fixed embeddingshypergraph representation learning · hyperedge prediction, 11 benchmark datasetsAUROCMixed

Surprising findings

Limitations

Experimental setup

Datasets: Citeseer Cora Pubmed Cora-CA DBLP-P DBLP-A AMiner IMDB MN-40 20News House

Benchmarks: Citeseer Cora Pubmed Cora-CA DBLP-P DBLP-A AMiner IMDB MN-40 20News House

Metrics: accuracy AUROC average ranking

Baselines: MLP HGNN HyperGCN HNHN UniGCN UniGIN UniGCNII AllSet ED-HNN PhenomNN GraphMAE2 MaskGAE

Builds on: UniGCNII GraphMAE2 TriCL Deep Sets alignment and uniformity

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy