arXiv:2409.05402cs.LGcs.AI2024-09被引 1

针对超图学习中的类别不平衡问题,提出新型数据增强方法HyperSMOTE。

HyperSMOTE: A Hypergraph-based Oversampling Approach for Imbalanced Node Classifications

  • 基于少数类样本及其邻居生成新节点,模拟真实分布
  • 通过解码器自适应地将新增节点关联到超边,保持结构一致性
  • 在Cora等数据集上提升准确率平均3.38%,适合处理高阶关系数据

超图因其在建模节点间高阶关系方面的优势,被广泛应用于单模态和多模态数据场景。然而,当前超图模型在处理类别不平衡数据时仍存在偏差问题,导致对多数类的过度偏好。尽管已有如GraphSMOTE等方法在图数据中提升了少数类分类性能,但在超图结构下仍不充分。受SMOTE思想启发,本文提出HyperSMOTE,通过两步策略缓解超图学习中的类别不平衡:首先基于少数类样本及其邻接节点合成新节点;随后利用原始超图关联矩阵训练解码器,自适应地将新增节点融入超边。我们在多个单模态数据集(如Cora、Cora-CA、Citeseer)及多模态对话数据集MELD上进行了全面评估,结果显示在准确率上分别平均提升3.38%和2.97%。

原文摘要 · Abstract (English)

Hypergraphs are increasingly utilized in both unimodal and multimodal data scenarios due to their superior ability to model and extract higher-order relationships among nodes, compared to traditional graphs. However, current hypergraph models are encountering challenges related to imbalanced data, as this imbalance can lead to biases in the model towards the more prevalent classes. While the existing techniques, such as GraphSMOTE, have improved classification accuracy for minority samples in graph data, they still fall short when addressing the unique structure of hypergraphs. Inspired by SMOTE concept, we propose HyperSMOTE as a solution to alleviate the class imbalance issue in hypergraph learning. This method involves a two-step process: initially synthesizing minority class nodes, followed by the nodes integration into the original hypergraph. We synthesize new nodes based on samples from minority classes and their neighbors. At the same time, in order to solve the problem on integrating the new node into the hypergraph, we train a decoder based on the original hypergraph incidence matrix to adaptively associate the augmented node to hyperedges. We conduct extensive evaluation on multiple single-modality datasets, such as Cora, Cora-CA and Citeseer, as well as multimodal conversation dataset MELD to verify the effectiveness of HyperSMOTE, showing an average performance gain of 3.38% and 2.97% on accuracy, respectively.

超图学习类别不平衡数据增强节点分类

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。