Active learning for Graph Neural Networks (GNNs) aims to select valuable unlabeled samples for annotation with a limited budget to maximize the GNNs’ performance at a low cost. However, most methods often result in imbalanced class distributions, leading to a bias toward majority classes, which undermines minority class performance and overall model effectiveness. To tackle this issue, we develop the Class-Balanced Active Learning System for Graphs GraphCBAL-Sys. It learns an optimal policy through reinforcement learning to acquire class-balanced and informative nodes for annotation. Additionally, GraphCBAL-Sys is capable of visualizing the internal processes and results during our model’s training and testing phases. Our demonstration video can be found here: https://b23.tv/yCLOIPw .

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

GraphCBAL-Sys: A Class-Balanced Active Learning System for Graphs

  • Chengcheng Yu,
  • Wenqian Zhou,
  • Jiahui Wang,
  • Fangshu Chen,
  • Jiapeng Zhu,
  • Xiang Li

摘要

Active learning for Graph Neural Networks (GNNs) aims to select valuable unlabeled samples for annotation with a limited budget to maximize the GNNs’ performance at a low cost. However, most methods often result in imbalanced class distributions, leading to a bias toward majority classes, which undermines minority class performance and overall model effectiveness. To tackle this issue, we develop the Class-Balanced Active Learning System for Graphs GraphCBAL-Sys. It learns an optimal policy through reinforcement learning to acquire class-balanced and informative nodes for annotation. Additionally, GraphCBAL-Sys is capable of visualizing the internal processes and results during our model’s training and testing phases. Our demonstration video can be found here: https://b23.tv/yCLOIPw .