华东交通大学学报2024，Vol.41Issue(3)：119-126,8.
基于模糊超图神经网络的节点分类方法
Node Classification Method Based on Fuzzy Hypergraph Neural Network
摘要
Abstract
[Objective]Hypergraph neural networks(HGNN)have the ability to learn inter-class uniqueness and intra-class commonality,which can significantly improve learning performance.However,traditional HGNN methods are in lack of the strong relational induction which determines the way how low-dimensional data nodes interact with each other.In order to solve this problem,a fuzzy HGNN(FHGNN)classification algorithm based on fuzzy theory is proposed,and hypergraph structure is constructed according to the characteristic information of data nodes.[Method]FHGNN first adopts an edge-focused GNN to make edge prediction through iterative updates of edge labels.The fuzzy membership function is designed according to the output of edge prediction to achieve a more accurate representation of the connection relationship between nodes.Finally,the hypergraph is constructed by the relation representation.Then the nodes are classified again and the result is obtained.The edge label loss function and node label loss function are used in FHGNN and their parameters are trained and learned respectively.[Result]Experimental results prove the proposed FHGNN method is more suitable for small-scale data with low node information dimension,and performs well in node classification tasks.[Conclusion]For clas-sification tasks of different data sets,FHGNN can learn the relevant feature information of nodes more effective-ly and improve the learning effect.关键词
超图/超图神经网络/模糊系统/节点分类Key words
hypergraph/hypergraph neural network/fuzzy system/node classification分类
信息技术与安全科学引用本文复制引用
熊婉悦,徐峰,黄宇廷,韩星宇,范自柱..基于模糊超图神经网络的节点分类方法[J].华东交通大学学报,2024,41(3):119-126,8.基金项目
国家自然科学基金项目(61991401) （61991401）
衢州市科技计划项目(2023K265) （2023K265）
