Similar to node classification, edge classification is also a crucial research topic in graph learning. While existing studies predominantly focus on imbalanced node classification, research on imbalanced edge classification remains relatively scarce. Nevertheless, in practical applications, there are numerous tasks that require classifying different types of relationships, such as identifying illegal transactions. In such scenarios, the number of illegal transaction samples is often significantly smaller than that of normal transactions. Directly applying GNN classifiers in these cases can lead to inadequate feature learning for minority-class samples, thereby reducing overall model performance. Thus, there is a need to design algorithms specifically for imbalanced edge classification in GNN models. Drawing inspiration from research on imbalanced node classification, we propose the EdgeSMOTE algorithm. We evaluate our algorithm on two benchmark datasets, and the results demonstrate its superior performance compared to baseline methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Edge Classification on Imbalanced Multi-relational Graphs

  • Zhaojie Gong,
  • Yijun Duan,
  • Qiang Ma

摘要

Similar to node classification, edge classification is also a crucial research topic in graph learning. While existing studies predominantly focus on imbalanced node classification, research on imbalanced edge classification remains relatively scarce. Nevertheless, in practical applications, there are numerous tasks that require classifying different types of relationships, such as identifying illegal transactions. In such scenarios, the number of illegal transaction samples is often significantly smaller than that of normal transactions. Directly applying GNN classifiers in these cases can lead to inadequate feature learning for minority-class samples, thereby reducing overall model performance. Thus, there is a need to design algorithms specifically for imbalanced edge classification in GNN models. Drawing inspiration from research on imbalanced node classification, we propose the EdgeSMOTE algorithm. We evaluate our algorithm on two benchmark datasets, and the results demonstrate its superior performance compared to baseline methods.