arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.12554cs.LG
收藏

超图神经网络的泛化性能

Generalization Performance of Hypergraph Neural Networks

University of Delaware(特拉华大学)

Yifan Wang, Gonzalo R. Arce, Guangmo Tong

AI总结:

本文基于PAC-Bayes框架为四类代表性超图神经网络开发了基于边界的泛化界,揭示了超图结构和权重谱范数对泛化的影响,并发现理论界与经验损失间存在强相关性。

摘要:

超图神经网络已成为处理涉及高阶数据学习任务的有力工具,在Web图中有显著应用,如建模多向超链接结构和复杂的用户交互。然而,它们在理论上的泛化能力尚不明确。本文旨在为四类代表性的超图神经网络开发基于边界的泛化界,包括基于卷积的方法(UniGCN)、基于集合的聚合(AllDeepSets)、不变和等变变换(M-IGN)以及基于张量的方法(T-MPHN)。通过PAC-Bayes框架,我们的结果揭示了超图结构和学习权重的谱范数如何影响泛化界,其中关键技术挑战在于为超图神经网络开发新的扰动分析,这为模型权重和超图结构的变异如何影响其泛化行为提供了严格理解。我们的实证研究考察了模型在合成和真实数据集上实际性能与理论界之间的关系。我们的主要观察之一是理论界与经验损失之间存在强相关性,在大多数情况下具有统计显著的致性。

英文摘要:

Hypergraph neural networks have been promising tools for handling learning tasks involving higher-order data, with notable applications in web graphs, such as modeling multi-way hyperlink structures and complex user interactions. Yet, their generalization abilities in theory are less clear to us. In this paper, we seek to develop margin-based generalization bounds for four representative classes of hypergraph neural networks, including convolutional-based methods (UniGCN), set-based aggregation (AllDeepSets), invariant and equivariant transformations (M-IGN), and tensor-based approaches (T-MPHN). Through the PAC-Bayes framework, our results reveal the manner in which hypergraph structure and spectral norms of the learned weights can affect the generalization bounds, where the key technical challenge lies in developing new perturbation analysis for hypergraph neural networks, which offers a rigorous understanding of how variations in the model's weights and hypergraph structure impact its generalization behavior. Our empirical study examines the relationship between the practical performance and theoretical bounds of the models over synthetic and real-world datasets. One of our primary observations is the strong correlation between the theoretical bounds and empirical loss, with statistically significant consistency in most cases.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑