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

HyperAggregation: Aggregating over Graph Edges with Hypernetworks

About

HyperAggregation is a hypernetwork-based aggregation function for Graph Neural Networks. It uses a hypernetwork to dynamically generate weights in the size of the current neighborhood, which are then used to aggregate this neighborhood. This aggregation with the generated weights is done like an MLP-Mixer channel mixing over variable-sized vertex neighborhoods. We demonstrate HyperAggregation in two models, GraphHyperMixer is a model based on MLP-Mixer while GraphHyperConv is derived from a GCN but with a hypernetwork-based aggregation function. We perform experiments on diverse benchmark datasets for the vertex classification, graph classification, and graph regression tasks. The results show that HyperAggregation can be effectively used for homophilic and heterophilic datasets in both inductive and transductive settings. GraphHyperConv performs better than GraphHyperMixer and is especially strong in the transductive setting. On the heterophilic dataset Roman-Empire it reaches a new state of the art. On the graph-level tasks our models perform in line with similarly sized models. Ablation studies investigate the robustness against various hyperparameter choices. The implementation of HyperAggregation as well code to reproduce all experiments is available under https://github.com/Foisunt/HyperAggregation .

Nicolas Lell, Ansgar Scherp• 2024

Related benchmarks

TaskDatasetResultRank
Node ClassificationChameleon
Accuracy74.78
640
Node ClassificationSquirrel
Accuracy83.49
591
Node ClassificationPubmed
Accuracy76.31
396
Node ClassificationCiteseer
Accuracy67.12
393
Node ClassificationAmazon Photo
Accuracy91.63
191
Node ClassificationAmazon Computers
Accuracy82.12
120
Graph ClassificationCIFAR10
Accuracy59.83
110
Graph RegressionZINC
MAE0.337
105
Graph ClassificationMNIST
Accuracy97.24
95
Node ClassificationMinesweeper
Accuracy87.49
55
Showing 10 of 14 rows

Other info

Code

Follow for update