Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

HyperBERT: Mixing Hypergraph-Aware Layers with Language Models for Node Classification on Text-Attributed Hypergraphs

About

Hypergraphs are characterized by complex topological structure, representing higher-order interactions among multiple entities through hyperedges. Lately, hypergraph-based deep learning methods to learn informative data representations for the problem of node classification on text-attributed hypergraphs have garnered increasing research attention. However, existing methods struggle to simultaneously capture the full extent of hypergraph structural information and the rich linguistic attributes inherent in the nodes attributes, which largely hampers their effectiveness and generalizability. To overcome these challenges, we explore ways to further augment a pretrained BERT model with specialized hypergraph-aware layers for the task of node classification. Such layers introduce higher-order structural inductive bias into the language model, thus improving the model's capacity to harness both higher-order context information from the hypergraph structure and semantic information present in text. In this paper, we propose a new architecture, HyperBERT, a mixed text-hypergraph model which simultaneously models hypergraph relational structure while maintaining the high-quality text encoding capabilities of a pre-trained BERT. Notably, HyperBERT presents results that achieve a new state-of-the-art on five challenging text-attributed hypergraph node classification benchmarks.

Adri\'an Bazaga, Pietro Li\`o, Gos Micklem• 2024

Related benchmarks

TaskDatasetResultRank
Node ClassificationCiteseer
Accuracy58.25
1037
Node ClassificationPhoto
Accuracy41.46
254
Node ClassificationComputers
Accuracy26.79
145
Node ClassificationCora
Accuracy47.03
134
Node ClassificationHistory
Accuracy62.12
29
Graph Classificationobgn-arXiv (test)
Accuracy58.5
28
Hyperedge predictionCiteseer
Accuracy71.84
18
Hyperedge predictionCora
Accuracy64.48
18
Vertex ClassificationArxiv-HG In-domain (test)
Accuracy59.1
18
Hyperedge predictionHistory
Accuracy69.52
15
Showing 10 of 17 rows

Other info

Follow for update