Organizations: Faculty of Electrical Engineering and Computer Science, Ningbo University, Ningbo, 315211, Zhejiang, China · Department of Computer Science and Technology, University of Cambridge, Cambridge, CB2 1TN, Cambridgeshire, United Kingdom
Hypergraph neural networks have achieved significant success in recent years. However, manual architecture crafting is labor-intensive and often fails to capture complex, higher-order relations, making the automation of hypergraph neural network structure design crucial. To improve the automation and adaptability of hypergraph learning, this paper proposes AutoHGNN, a neural architecture search framework tailored for hypergraph neural networks. First, we introduce a Hyper-Interaction Module (HIM) into the search space to address the mismatch between conventional graph neural network designs and hypergraph data. Second, we propose Hypergraph Stable Topological Distance (HyperSTD) as a structural selection criterion to identify architectures that best preserve the intrinsic structural affinities of the original hypergraph during differentiable search. Extensive experiments on various benchmark datasets demonstrate that AutoHGNN consistently outperforms manually designed and automatically searched baselines in classification accuracy and time efficiency, proving that the discovered architectures are significantly more effective.
Figures & tables
Method
Type
Message Passing
Complexity
HGNN [ 10 ]
Spectral
Laplacian
O(n2)
HyperGCN [ 45 ]
Spectral
Graph conversion
O(m⋅d)
HNHN [ 7 ]
Spectral
Hyperedge neurons
O(n⋅d)
HyperSAGE [ 1 ]
Spatial
Two-stage aggregation
O(sampling)
HGNN+ [ 14 ]
Spatial
Aggregation + update
O(nd+md)
UniG-Encoder [ 49 ]
Spatial
Projection-encoding-decoding
O(nd)
Table 1: Comparison of representative hypergraph neural network methodologies.
Figure 1: The overview of AutoHGNN. Green, orange and blue denote vertex-to-hyperedge aggregation, hyperedge-to-vertex aggregation and post-processing operations respectively. Varying depth of same color indicates different members of the same operation type that share the same weight ω . (a) The HGNN search space whose details are in Fig. 2 . (b) The search module with a weight-sharing supernet constructed from search space, a differentiable sampler and a supernet training modules. (c) Hypergraph neural architecture selection. In stage 1, promising candidates are selected from the j-th round of sampling and gradually merged into promising candidates set, so that the final architectures can be selected from them in stage 2.
Figure 2: Definition of search space in AutoHGNN. Each network layer consists of (a) a Hyper-Interaction Module layer with a vertex-to-hyperedge aggregator and a hyperedge-to-vertex aggregator and (b) a post-processing layer. All layers’ outputs are fused by the (c) feature aggregation operator F to get final output Z for downstream tasks. Xv(l) means vertex features of the l -th layer.
Table 6: Result on graph datasets. Bold and underlined texts indicate best and second-best models respectively. STPE indicates Search Time Per Epoch for NAS methods.
Method
Cora_CA
DBLP
Mean (%)
Std (%)
Mean (%)
Std (%)
GAT
66.95
1.00
83.06
0.48
GraphSAGE
72.38
0.95
84.69
0.09
SGC
66.56
0.47
83.05
0.32
GraphConv
70.71
0.40
83.17
0.37
GATv2
67.01
0.77
83.68
0.70
Table 7: Result on hypergraph datasets. Bold and underlined texts indicate best and second-best models respectively.
Figure 3: Hyperparameter sensitivity analysis of AutoHGNN Hyperparameter sensitivity analysis of AutoHGNN on (a) search epochs, (b) network depth, (c) temperature, and (d) training sample size. The X axis is the parameter value and Y axis is the average accuracy of 10 experiments on best architecture found by AutoHGNN.
Variant
Pubmed
Computers
Physics
Cora_CA
DBLP
AutoHGNN(mean+mean)
81.95 ± 0.12
84.92 ± 0.09
93.82 ± 0.08
67.26 ± 0.21
77.02 ± 0.11
AutoHGNN(w/o fusion)
79.98 ± 0.65
84.39 ± 0.37
93.97 ± 0.16
66.18 ± 0.12
86.47 ± 0.20
AutoHGNN(w/o HyperSTD)
82.01 ± 0.11
85.10 ± 0.09
94.85 ± 0.03
72.65 ± 0.44
88.09 ± 0.09
AutoHGNN
83.17 ± 0.23
87.02 ± 0.17
95.53 ± 0.04
73.76 ± 0.41
88.91 ± 0.07
Table 8: Ablation study results. Bold texts indicate the best model.
School of Computer Science and Engineering, University of New South Wales, Sydney, Australia · School of Artificial Intelligence, Shenzhen University, Shenzhen, China.