Graph Neural Networks (GNNs) have demonstrated remarkable success in classification tasks on graphs, including multimedia applications such as image recognition, video analysis, and recommendation systems. However, most GNNs methods assume that the category of samples is balanced, which contradicts real-world class distribution. In practice, imbalanced category distribution often causes GNNs to neglect minority-class nodes during training, which in turn negatively impacts overall classification performance. Existing methods still face key challenges, including insufficient feature learning and inadequate generation of node homogeneity. To tackle these challenges, we propose GraphAFA, a novel Graph-based method that utilizes Attentional Feature Aggregation to generate a small number of synthetic class nodes, thereby promoting sample equilibrium. GraphAFA consists of two key components: attention-based feature extraction and neighbor-aware node aggregation. Firstly, GraphAFA constructs a feature space and utilizes an attention mechanism to extract node features, enabling effective learning higher-order relationships among nodes. Secondly, during the node generation process, GraphAFA aggregates information from neighboring nodes to capture shared features, ensuring the newly generated nodes are more homogeneous and reducing the risk of generating heterogeneous samples. Finally, GraphAFA connects edges to the newly generated nodes, integrating them into the graph for downstream classification. Comprehensive experiments on three benchmark datasets show that GraphAFA consistently outperforms state-of-the-art methods in class-imbalanced node classification.
扫码关注我们
求助内容:
应助结果提醒方式:
