Class-Imbalanced Graph Learning without Class Rebalancing
Abstract
Class imbalance is prevalent in real-world node classification tasks and poses great challenges for graph learning models. Most existing studies are rooted in a class-rebalancing (CR) perspective and address class imbalance with class-wise reweighting or resampling. In this work, we approach the root cause of class-imbalance bias from an topological paradigm. Specifically, we theoretically reveal two fundamental phenomena in the graph topology that greatly exacerbate the predictive bias stemming from class imbalance. On this basis, we devise a lightweight topological augmentation framework Bat to mitigate the class-imbalance bias without class rebalancing. Being orthogonal to CR, Bat can function as an efficient plug-and-play module that can be seamlessly combined with and significantly boost existing CR techniques. Systematic experiments on real-world imbalanced graph learning tasks show that Bat can deliver up to 46.27% performance gain and up to 72.74% bias reduction over existing techniques. Code, examples, and documentations are available at https://github.com/ZhiningLiu1998/BAT.
highlightrev \togglefalsehighlightrev
1 Introduction
![Refer to caption](extracted/5605652/figs/intro-a.png)
![Refer to caption](extracted/5605652/figs/intro-b.png)
Node classification stands as one of the most fundamental tasks in graph machine learning, holding significant relevance in various real-world applications (Akoglu et al., 2015; Tang & Liu, 2010). Graph Neural Networks (GNNs) have demonstrated great success in tackling related tasks due to their robust representation learning capabilities (Song et al., 2022b; Fu & He, 2021). However, real-world graphs are often inherently class-imbalanced, i.e., the sizes of unique classes vary significantly, and a few majority classes have overwhelming numbers in the training set. In Class-Imbalanced Graph Learning (CIGL), GNNs are prone to suffer from severe performance degradation on minority class nodes (Park et al., 2022). This results in a pronounced predictive bias characterized by a large performance disparity between the majority and minority classes.
Traditional imbalance-handling techniques rely on class rebalancing (CR) such as class reweighting and resampling (Chawla et al., 2002; Cui et al., 2019), which works well for non-graph data. Recent studies propose more graph-specific CR strategies tailored for CIGL, e.g., neighborhood-aware reweighting (Li et al., 2022; Huang et al., 2022) and oversampling (Zhao et al., 2021b; Park et al., 2022). Nonetheless, these works are restricted to the class-rebalancing paradigm. Parallel to class imbalance, another emerging line of research studies topology imbalance, characterized by “the asymmetric topological properties of the labeled nodes” (Chen et al., 2021). It is considered an orthogonal problem to class imbalance, and hence, few work theoretically investigates how topological structure affects the learning on class imbalanced graphs. To fill this gap, we conduct an in-depth analysis of the role that topology plays in class-imbalanced graph learning. We theoretically show that topological differences between minority and majority classes significantly amplify the class imbalance bias, imposing great challenges to CIGL. This reveal an unexplored avenue that limits the performance of existing CIGL techniques: mitigating class imbalance bias arising from imbalanced topology structure through topological operations. Following this novel perspective, we devise a lightweight practical solution for CIGL that can be seamlessly combined with and further boost existing CR techniques.
In this work, we formally define and theoretically investigate two fundamental local topological phenomena that greatly hinder CIGL: (i) ambivalent message-passing (AMP), i.e., high ratio of non-self-class neighbors in the node receptive field, and (ii) distant message-passing (DMP), i.e., poor connectivity with self-class labeled nodes. Intuitively, AMP leads to a higher influx of noisy information and DMP leads to poor reception of effective supervision signals in message-passing. Both result in lower signal-to-noise ratios and thus induce higher classification errors. Our theoretical finding reveals that the minority class is inherently more susceptible to both AMP and DMP (Theorem 2.1 & 2.2), which leads to a more pronounced predictive bias. Such bias induced by the graph topology escalates as the level of class imbalance increases. We emphasize that AMP/DMP is defined for all nodes based on the local neighborhood, while influence conflict has no formal definition and influence insufficiency is defined only on labeled nodes with global PageRank score (Chen et al., 2021). To distinguish from them, we use the terminology AMP/DMP instead. Further discussions can be found in Section 5. Fig. 1 visually illustrates the concepts of AMP and DMP, highlighting their distinct impacts on the predictive performance of majority and minority classes.
Following our theoretical and empirical findings, we devise Bat (BAlanced Topological augmentation), a model-agnostic and efficient technique to mitigate class imbalance bias in CIGL via topological augmentation. Bat dynamically locates and rectifies nodes critically influenced by AMP and DMP during learning, thereby effectively reducing the errors and biases in CIGL. Being orthogonal to class rebalancing, our solution is able to work hand-in-hand with existing techniques based on reweighting (Japkowicz & Stephen, 2002; Chen et al., 2021) and resampling (Zhao et al., 2021b; Park et al., 2022) and further boost their performance. Systematic experiments on real-world CIGL tasks show that Bat delivers significant performance boost (up to 46.27%) and bias reduction (up to 72.74%) over various CIGL baselines with diverse GNN architectures.
Our contributions: (i) Novel Perspective. We demonstrate the feasibility of taming class-imbalance bias without class rebalancing, which provides a new avenue that is orthogonal to the predominant class-rebalancing practice in CIGL. (ii) Theoretical Insights. We theoretically reveal the topological difference between minority and majority classes and its role in sha** predictive bias in CIGL, shedding light on future CIGL research. Empirical results validate our findings. (iii) Practical Solution. Motivated by theoretical and empirical finding, we devise a lightweight and versatile framework Bat to handle topological challenges in CIGL. Being complementary to class rebalancing, it can be seamlessly combined with and significantly boost existing CIGL techniques. (iv) Empirical Study. Systematic experiments and analysis across a diverse range of real-world tasks and GNN architectures show that Bat consistently demonstrates superior performance in both promoting classification and mitigating predictive bias.
2 Class Imbalance and Local Topology
![Refer to caption](extracted/5605652/figs/analysis_distr.png)
![Refer to caption](extracted/5605652/figs/analysis_acc.png)
In this section, we delve into the impact of graph topology on the predictive bias in class-imbalanced node classification. We theoretically unveil that compared to the majority class, the minority class is inherently more susceptible to both Ambivalent Message Passing (AMP) and Distant Message Passing (DMP). This significantly worsens minority-class performance and leads to a more pronounced predictive bias stemming from class imbalance. After that, we present an empirical analysis to validate our theoretical findings, and to provide insights on how to mitigate the bias induced by AMP and DMP in practice. Detailed proofs can be found in Appendix A.
Theoretical analysis on local topology. Consider a graph from a stochastic block model (Holland et al., 1983) , where is the total number of nodes, and are the intra- and inter-class node connection probability. To facilitate analysis, we call node homo-connected to node if there is a path where are of the same class, and let denote the set of -hop homo-connected neighbors of . For binary node classification, we denote the number of nodes of class as (); without loss of generality, let class / be the minority/majority class (thus ). We denote class ’s node set as and labeled node set as (). For asymptotic analysis, we adopt conventional assumptions: (i.e., is the average intra-class node degree of class 1); (Decelle et al., 2011).
We now give formal definitions of AMP and DMP. For a node from class , we define its (i) -hop AMP coefficient as the ratio of the expected number of non-self-class nodes to self-class nodes in its -hop neighborhood , i.e., ; (ii) -hop DMP coefficient as the indicator of whether all labeled nodes in its -hop neighborhood are NON-self-class, i.e., . For an intuitive example, the target node (marked by the dashed box) in Fig. 1(a) has and node in Fig. 1(b) has . Further, to characterize the level of AMP/DMP for different class, for class we define and , where is a node of class . Intuitively, a higher or indicates that class is more susceptible to AMP or DMP. Building on these metrics, we analyze the disparities in and between minority and majority classes, thereby providing insights into how the graph topology induces additional class-imbalance bias.
To simplify notation, we define imbalance ratio . The larger the is, the more imbalanced the dataset is. Then for AMP, we have the following Theorem 2.1.
Theorem 2.1 (AMP-sourced bias).
For a large , the ratio of AMP coefficients for the minority class to the majority class grows polynomially with the imbalance ratio and exponentially with :
(1) |
Proof.
Please see Appendix A.2.∎
Theorem 2.1 shows that the same-class neighbor proportion of minority-class nodes is significantly smaller than that of majority-class nodes, i.e., the minority class is more susceptible to AMP. As the imbalance ratio increases, this issue becomes even more pronounced and introduces a higher bias into the learning process. Moving on to DMP, we have the following theorem 2.2.
Theorem 2.2 (DMP-sourced bias).
Let denote the label rate of class . For a large , the ratio of DMP coefficients of the minority class over the majority class grows exponentially with :
(2) |
Proof.
Please see Appendix A.3.∎
Similarly, the result shows that the minority class exhibits a significantly higher susceptibility to DMP than the majority class. Theorem 2.2 also has several interesting implications: (i) The imbalance ratio greatly affects the bias induced by DMP, as grows exponentially with . (ii) Labeling more minority-class nodes can mitigate, but hardly solve the problem. Enlarging the minority-class label rate can linearly shrink , but it can hardly eliminate the bias induced by DMP (i.e., to have ) as is usually very large in practice. Take the Cora dataset (Sen et al., 2008) as an example (let class /class denote the smallest/largest class): eliminating the DMP bias requires the minority-class label rate , which is practically infeasible.
Our theoretical findings show that both AMP and DMP affect the minority and majority classes differently, and the difference is primarily determined by the imbalance ratio . However, directly manipulating is tricky in practice as it requires sampling new nodes and edges from an unknown underlying graph generation model, or at least, simulating the process by oversampling.
![Refer to caption](x1.png)
A closer look at AMP & DMP in practice. To verify the theoretical results, and to provide more insights on how to mitigate the bias brought about by AMP and DMP in practice, we conduct a fine-grained empirical analysis on a real-world task. Results are detailed in Fig. 2111Results obtained by training a GCN on the PubMed dataset.. Starting from Fig. 2(a), we can first observe that the minority class has a larger proportion of nodes with high or than the majority class , i.e., minority class has higher average and (specifically, , ), which is consistent with our theoretical findings. Further, Fig. 2(b) shows that both AMP and DMP significantly reduce the prediction accuracy, especially for the minority class that inherently has poorer data representation. This can be explained from a graph signal denoising perspective (Nt & Maehara, 2019): AMP introduces additional noise from dissimilar nodes, and DMP leads to less efficient label propagation/denoising, thus their impact is particularly significant on minority classes that are more susceptible to noise (Johnson & Khoshgoftaar, 2019) due to poor representation in the feature space. We further notice an intriguing fact that, at sample-level, the impact of AMP/DMP is concentrated on a small fraction of minority class nodes with large or (e.g., the / part in Fig. 2(a)). In other words, one can surrogate the tricky manipulation of and directly mitigate AMP/DMP by locating and rectifying a small number of critical nodes, and this exactly motivates our subsequent studies.
3 Handling Class Imbalance from a Topological Perspective
Armed with the findings from Section 2, we now discuss how to devise a practical strategy to mitigate the error and bias induced by graph topology in CIGL. Earlier analyses have shown that this can be achieved by identifying and rectifying the critical nodes that are highly influenced by AMP/DMP. This naturally poses two challenging questions: (i) How can critical nodes be located as the direct calculation of / using ground-truth labels is not possible? (ii) Subsequently, how can critical nodes be rectified and minimize the negative impact caused by AMP and DMP?
In answering the above questions, we devise a lightweight framework Bat (BAlanced Topological augmentation) for handling the topology-sourced errors and biases in CIGL. Specifically, for locating the misclassified nodes, Bat leverages model-based prediction uncertainty to assess the risk of potential misclassification caused by AMP/DMP for each node (§ 3.1). Then to rectify a misclassified node, we estimate a posterior likelihood of each node being in each class (§ 3.2) and dynamically augment the misclassified node’s topological context based on our risk scores and posterior likelihoods (§ 3.3) thereby mitigating the impact of AMP and DMP. An overview of the proposed Bat framework is shown in Fig. 3.
3.1 Node Misclassification Risk Estimation
We now elaborate on the technical details of Bat. As discussed earlier, our first step is to locate the critical nodes that are highly influenced by AMP/DMP. Given the unavailability of ground-truth labels, direct computation of AMP/DMP coefficient is infeasible in practice. Fortunately, recent studies have shown that conflicting or lack of information from the neighborhood can disturb GNN learning and the associated graph-denoising process for affected nodes (Nt & Maehara, 2019; Wu et al., 2019; Ma et al., 2021). This further yields high vacuity or dissonance uncertainty (Stadler et al., 2021; Zhao et al., 2020) in the prediction. This motivates us to exploit the model prediction uncertainty to estimate nodes’ risk of being misclassified due to AMP/DMP.
Uncertainty quantification. While there exist many techniques for uncertainty quantification (e.g., Bayesian-based (Zhang et al., 2019; Hasanzadeh et al., 2020), Jackknife sampling (Kang et al., 2022a)), they often either have to modify the model architecture, and/or impose additional computational overhead. In this study, we aim to streamline the design of Bat for optimal efficiency and adaptability. To this end, we employ an efficient and highly effective approach to uncertainty quantification. Formally, let be the number of classes. For a node , consider model ’s predicted probability vector , i.e., . Let be the predicted label. We measure the uncertainty score by the total variation (TV) distance:
|
(3) |
Intuitively, a node has higher uncertainty if the model is less confident about its current prediction. We remark that this metric can be naturally replaced by other uncertainty measures (e.g., information entropy or more complex ones) with additional computation cost, yet the impact on performance is marginal. Please refer to the ablation study provided in Appendix C.1 for more details.
Imbalance-calibrated misclassification risk. Due to the lack of training instances, minority classes generally exhibit higher uncertainty. Therefore, using directly as the risk score would treat most minority-class nodes as high-risk, which is contrary to our intention of rectifying the false negatives (i.e., minority nodes wrongly predicted as majority-class) that cause bias in CIGL. To cope with this, we propose imbalance-aware calibration for misclassification risk scores. For each class , let and . For node with predicted label , we define its risk as:
(4) |
Intuitively speaking, Eq. (4) calibrates ’s prediction uncertainty by a label imbalance score . Minority classes with smaller labeled sets will be discounted more.
Empirical validation. We validate the effectiveness of the proposed node risk assessment method, as shown in Fig. 4. The results indicate that our approach can accurately estimate node misclassification risk across various real-world CIGL tasks while enjoying computational efficiency.
![Refer to caption](extracted/5605652/figs/risk_acc.png)
3.2 Posterior Likelihood Estimation
With the estimated risk scores of being affected by AMP/DMP, we move to the next question: how to rectify high-risk nodes with topological augmentation? As high-risk nodes are prone to misclassification, their true labels are more likely to be among the non-predicted classes . This motivates us to investigate schemes to harness information from these non-predicted classes. Since uniformly drawing from all classes probably introduces noise to learning, we propose to estimate the posterior likelihood that a high-risk node belongs to each class after observing the current predictions. To estimate , we introduce a zeroth-order scheme and a first-order scheme with and time complexity, respectively. Please refer to §4 for a practical complexity analysis. We do not employ higher-order schemes due to the time complexity of the th-order scheme.
Zeroth-order estimation. A natural approach is to utilize the predicted probabilities . As we have shown, the predicted label of a high-risk node is very likely to be wrong. Thus, we define the posterior likelihood as the conditional probability given that the class is not , i.e.,
(5) | ||||
Intuitively, the zeroth-order posterior likelihoods are consistent with the predicted probabilities except for the wrongly predicted label . This can be computed efficiently on GPU in matrix form.
First-order estimation via random walk. We further explore the local topology for estimation. Since neighboring nodes on a homophily graph tend to share labels, we consider a 1-step random walk starting from node . Let be the neighboring node set of , and let denote the ending node of the random walk. We define as the conditional probability that is predicted as class given that is not predicted as class , i.e.,
(6) | ||||
With first-order estimation, is proportional to the label frequency among adjacent nodes. Different from the zeroth-order scheme, this scheme relies on both node-level predictions and local connectivity patterns. The computation can be done via sparse matrix operation with time complexity. As a remark, although this scheme can extend to -step random walks, we do not employ them due to the complexity of exact computation and the high variance of stochastic computation.
Empirical validation. Figure 5 compares the two schemes in practice. Results show that all high-risk () minority nodes are misclassified, and both schemes can effectively find alternatives with significantly higher chances to be the ground truth class for high-risk nodes.
![Refer to caption](extracted/5605652/figs/cand_acc_pubmed.png)
3.3 Virtual Topology Augmentation
Finally, we discuss how to mitigate AMP and DMP via topology augmentation using our node risk scores and the posterior likelihoods. The general idea is to augment the local topology of high-risk nodes so as to integrate information from nodes that share similar patterns (mitigate AMP), even if they are not closely adjacent to each other in the graph topology (mitigate DMP), thus achieving less-biased CIGL. A straightforward way is to connect high-risk nodes to nodes from high-likelihood classes in the original graph. However, this can be problematic in practice as a massive number of possible edges could be generated, greatly disturbing the original topology structure.
To achieve efficient augmentation without disrupting the graph topology, we create virtual nodes (one per class) as “shortcuts” connecting to high-risk nodes according to posterior likelihoods. These shortcuts aggregate and pass class information to high-risk nodes from nodes that exhibit similar patterns (even if they are distant in the original graph), thus mitigating both AMP and DMP. Formally, for each class , we build a virtual node with feature and label , and compute the average risk . Then for each node , we connect a virtual edge between and virtual node with probability proportional to the posterior likelihood . However, if the connection probability is exactly , there will be many unnecessary virtual edges for low-risk nodes. Hence, we introduce a discount factor based on risk scores and connect the virtual edge with probability . To design the optimal , we propose to solve the following constrained quadratic program:
(7) |
where the first term encourages virtual edges for high-risk nodes, and the second term is to minimize the number of virtual edges. The closed-form solution is (Antoniadis & Fan, 2001), which avoids virtual edges for low-risk nodes as we desire. We now summarize the procedure of Bat in Algorithm 1.
Complexity Analysis of Bat. Bat with 0th/1st-order estimation scales linearly with the number of nodes/edges (i.e., with or complexity). This makes Bat highly efficient and allows dynamically graph augmentation in each training step. Specifically, Bat introduces (the number of class, usally small) virtual nodes with edges. Because of the long-tail distribution of node uncertainty and the discount factor used to solve Eq. (7), only a small portion of nodes have positive risks with relatively few (empirically around 1-3%) virtual edges introduced. We provide the scalability results of Bat later in the experiment section. In short, Bat takes milliseconds for a single topological augmentation. Please refer to Table 3 and the corresponding discussion for further details. We also discuss how to further speedup Bat in practice in C.2.
Metric | Balanced Acc. | Macro-F1 | PerfStd | ||||||||||||
CIGL Baseline | ERM | RW | RN | RS | SM | GS | GE | Avg () | Best () | Avg () | Best () | Avg () | Best () | ||
Cora | GCN | base | 61.6 | 67.7 | 66.6 | 59.5 | 58.3 | 68.0 | 70.1 | 64.5 | 70.1 | 63.7 | 70.0 | 25.7 | 20.0 |
+Bat0 | 65.5 | 71.0 | 71.4 | 72.5 | 72.2 | 68.5 | 72.2 | 70.5 (+5.9) | 72.5 (+2.4) | 69.2 (+5.5) | 71.6 (+1.7) | 16.7 (-9.0) | 14.4 (-5.6) | ||
+Bat1 | 69.8 | 72.1 | 71.8 | 74.2 | 73.9 | 71.6 | 72.6 | 72.3 (+7.8) | 74.2 (+4.1) | 71.1 (+7.5) | 72.8 (+2.9) | 17.6 (-8.0) | 15.2 (-4.8) | ||
GAT | base | 61.5 | 66.9 | 66.8 | 57.8 | 58.8 | 64.7 | 69.8 | 63.8 | 69.8 | 63.1 | 70.0 | 26.0 | 20.1 | |
+Bat0 | 66.3 | 71.8 | 72.1 | 71.9 | 70.5 | 69.3 | 70.6 | 70.4 (+6.6) | 72.1 (+2.4) | 69.0 (+6.0) | 70.9 (+0.9) | 17.2 (-8.9) | 15.1 (-5.0) | ||
+Bat1 | 70.1 | 71.6 | 70.3 | 73.3 | 72.2 | 71.1 | 71.0 | 71.4 (+7.6) | 73.3 (+3.5) | 70.2 (+7.1) | 72.3 (+2.4) | 18.0 (-8.0) | 17.3 (-2.8) | ||
SAGE | base | 59.2 | 63.8 | 65.3 | 57.8 | 58.8 | 61.6 | 68.8 | 62.2 | 68.8 | 60.9 | 68.2 | 27.1 | 19.8 | |
+Bat0 | 66.2 | 70.1 | 71.3 | 71.2 | 70.3 | 69.9 | 69.8 | 69.8 (+7.7) | 71.3 (+2.5) | 68.9 (+8.0) | 70.4 (+2.2) | 16.4 (-10.7) | 13.3 (-6.5) | ||
+Bat1 | 66.5 | 71.1 | 71.5 | 73.0 | 73.0 | 72.3 | 71.9 | 71.4 (+9.2) | 73.0 (+4.2) | 70.1 (+9.2) | 71.7 (+3.5) | 16.6 (-10.5) | 14.9 (-4.9) | ||
CiteSeer | GCN | base | 37.6 | 42.5 | 42.6 | 39.2 | 39.3 | 45.1 | 56.0 | 43.2 | 56.0 | 36.1 | 54.5 | 27.0 | 16.9 |
+Bat0 | 52.7 | 57.9 | 57.5 | 57.9 | 60.1 | 57.7 | 60.6 | 57.8 (+14.6) | 60.6 (+4.6) | 56.9 (+20.8) | 59.9 (+5.4) | 17.6 (-9.4) | 13.8 (-3.1) | ||
+Bat1 | 55.4 | 58.4 | 59.3 | 58.8 | 62.0 | 57.6 | 62.7 | 59.2 (+16.0) | 62.7 (+6.7) | 58.4 (+22.3) | 62.5 (+8.0) | 19.3 (-7.7) | 13.9 (-3.0) | ||
GAT | base | 39.2 | 41.3 | 43.2 | 36.0 | 37.0 | 41.8 | 51.5 | 41.4 | 51.5 | 34.1 | 48.3 | 29.0 | 25.2 | |
+Bat0 | 55.7 | 59.3 | 58.3 | 60.1 | 60.6 | 56.1 | 60.9 | 58.7 (+17.3) | 60.9 (+9.4) | 58.1 (+23.9) | 60.0 (+11.7) | 16.3 (-12.6) | 10.7 (-14.5) | ||
+Bat1 | 60.3 | 61.2 | 59.1 | 60.3 | 62.4 | 57.7 | 63.5 | 60.6 (+19.2) | 63.5 (+12.0) | 59.9 (+25.8) | 62.5 (+14.2) | 17.7 (-11.3) | 13.2 (-12.0) | ||
SAGE | base | 43.0 | 45.9 | 48.6 | 39.4 | 38.4 | 42.2 | 52.6 | 44.3 | 52.6 | 37.9 | 51.0 | 27.1 | 19.8 | |
+Bat0 | 55.0 | 58.0 | 56.3 | 61.4 | 64.1 | 60.9 | 64.4 | 60.0 (+15.7) | 64.4 (+11.8) | 59.4 (+21.6) | 63.9 (+12.8) | 17.5 (-9.7) | 13.2 (-6.6) | ||
+Bat1 | 53.2 | 55.9 | 56.5 | 61.9 | 66.3 | 62.3 | 63.8 | 60.0 (+15.7) | 66.3 (+13.8) | 59.3 (+21.4) | 65.9 (+14.9) | 18.6 (-8.6) | 12.8 (-7.0) | ||
PubMed | GCN | base | 64.2 | 71.2 | 71.5 | 65.0 | 64.4 | 74.0 | 73.7 | 69.1 | 74.0 | 63.5 | 71.3 | 23.2 | 11.9 |
+Bat0 | 68.6 | 74.2 | 73.2 | 72.5 | 73.2 | 73.1 | 76.1 | 73.0 (+3.8) | 76.1 (+2.1) | 72.0 (+8.5) | 75.8 (+4.5) | 9.1 (-14.1) | 3.4 (-8.6) | ||
+Bat1 | 67.6 | 73.4 | 72.5 | 72.9 | 73.1 | 76.6 | 76.9 | 73.3 (+4.1) | 76.9 (+2.9) | 72.6 (+9.1) | 76.9 (+5.6) | 9.9 (-13.4) | 5.1 (-6.8) | ||
GAT | base | 65.5 | 68.4 | 71.2 | 65.1 | 64.8 | 68.7 | 73.1 | 68.1 | 73.1 | 62.4 | 71.8 | 24.8 | 10.3 | |
+Bat0 | 73.2 | 75.3 | 75.6 | 73.3 | 73.9 | 74.7 | 74.3 | 74.3 (+6.2) | 75.6 (+2.4) | 73.4 (+11.1) | 75.1 (+3.3) | 6.5 (-18.2) | 3.0 (-7.3) | ||
+Bat1 | 74.8 | 74.5 | 75.2 | 73.9 | 74.1 | 74.4 | 75.7 | 74.6 (+6.5) | 75.7 (+2.5) | 73.9 (+11.5) | 75.0 (+3.2) | 6.9 (-17.8) | 4.6 (-5.7) | ||
SAGE | base | 67.6 | 68.0 | 69.1 | 69.2 | 65.0 | 71.5 | 71.4 | 68.8 | 71.5 | 64.5 | 70.1 | 22.1 | 11.8 | |
+Bat0 | 75.3 | 74.6 | 74.2 | 74.9 | 74.6 | 74.7 | 75.9 | 74.9 (+6.1) | 75.9 (+4.3) | 74.2 (+9.7) | 75.3 (+5.3) | 7.6 (-14.4) | 3.4 (-8.4) | ||
+Bat1 | 77.4 | 75.4 | 75.3 | 75.8 | 77.3 | 76.1 | 76.5 | 76.3 (+7.4) | 77.4 (+5.8) | 75.8 (+11.3) | 76.9 (+6.9) | 6.8 (-15.3) | 4.1 (-7.7) |
-
*Bat0/Bat1: Bat with /-order posterior likelihood estimation. ERM: Empirical Risk Minimization (standard training), RW: Reweight (Japkowicz & Stephen, 2002), RN: ReNode (Chen et al., 2021), RS: Resampling (Japkowicz & Stephen, 2002), SM: SMOTE (Chawla et al., 2002), GS: GraphSMOTE (Zhao et al., 2021b), GE: GraphENS (Park et al., 2022).
4 Experiments
We carry out systematic experiments and analysis to validate Bat in the following aspects: (i) Effectiveness in both promoting imbalanced node classification and mitigating the prediction bias between different classes. (ii) Versatility in cooperating with and further boosting various CIGL techniques and GNN backbones. (iii) Robustness to extreme class imbalance. (iv) Efficiency in real-world applications.
Experiment Protocol. We validate Bat on five benchmark datasets for semi-supervised node classification, including the Cora, CiteSeer, PubMed from Plantoid graphs (Sen et al., 2008), and larger-scale CS, Physics from co-author networks (Shchur et al., 2018) with high-dimensional features. Following the same setting as prior studies (Park et al., 2022; Song et al., 2022a; Zhao et al., 2021b), we select half of the classes as minority. The imbalance ratio is the ratio between the size of the largest class to the smallest class, i.e., more imbalance higher IR. Detailed data statistics and class distributions can be found in Appendix B.1. We test Bat with six CIGL techniques (Park et al., 2022; Chen et al., 2021; Zhao et al., 2021b; Chawla et al., 2002; Japkowicz & Stephen, 2002) and three GNN backbones (Veličković et al., 2018; Hamilton et al., 2017; Welling & Kipf, 2016) under all possible combinations to fully validate Bat’s effectiveness and versatility in practice. Note that although there are other techniques available for CIGL (Hong et al., 2021; Kang et al., 2019; Shi et al., 2020), previous studies (Park et al., 2022; Song et al., 2022a) have shown they are generally outperformed by the baselines we use. Detailed settings can be found in Appendix B.2. To ensure a comprehensive evaluation, we employ three metrics to assess both the classification performance (Balanced Accuracy, Macro-F1) and the model predictive bias (PerfStd, i.e., the standard deviation of accuracy scores across all classes). Lower PerfStd indicates smaller performance gap between all majority and minority classes, and thus smaller predictive bias. For clarity, we use / to denote larger/smaller is better for each metric.
Bat significantly boosts various CIGL techniques. We report the main results in Table 1 (IR=10). In all settings (3 datasets3 backbones7 baselines3 metrics), Bat achieves significant and consistent performance improvements over other CIGL techniques, which also yields new state-of-the-art performance. Specifically: (1) By mitigating AMP and DMP, Bat further boosts the best CIGL baseline by a large margin, e.g., it boosts the best balanced accuracy score by 4.1/13.8/5.8 on Cora/CiteSeer/PubMed datasets. (2) In addition to better classification performance, Bat also greatly reduces the predictive bias in CIGL, with up to 10.7/14.5/18.2 average performance deviation reduction on Cora/CiteSeer/PubMed. (3) Compared with Bat0, Bat1 achieves better classification performance with first-order posterior likelihood estimation. But we also note that Bat0 performs better in terms of reducing predictive bias and is more computationally efficient, as we will discuss later in the scalability experiments (Table 3).
Dataset | Cora | CiteSeer | PubMed | CS | Physics | |||||
---|---|---|---|---|---|---|---|---|---|---|
Step IR | 10 | 20 | 10 | 20 | 10 | 20 | 10 | 20 | 10 | 20 |
Base | 61.6 | 52.7 | 37.6 | 34.2 | 64.2 | 60.8 | 75.4 | 65.3 | 80.1 | 67.7 |
+ Bat | 69.8 (+8.2) | 71.3 (+18.5) | 55.4 (+17.7) | 51.3 (+17.1) | 68.6 (+4.4) | 63.3 (+2.5) | 82.6 (+7.2) | 79.9 (+14.5) | 87.6 (+7.5) | 88.0 (+20.2) |
BestCIGL | 70.1 | 66.5 | 56.0 | 47.2 | 74.0 | 71.1 | 84.1 | 81.3 | 89.4 | 85.7 |
+ Bat | 74.2 (+4.1) | 71.6 (+5.1) | 62.7 (+6.7) | 62.5 (+15.3) | 76.9 (+2.9) | 75.7 (+4.6) | 86.3 (+2.2) | 85.6 (+4.3) | 91.2 (+1.9) | 90.9 (+5.2) |
Natural IR | 50 | 100 | 50 | 100 | 50 | 100 | 50 | 100 | 50 | 100 |
Base | 60.1 | 47.0 | 28.1 | 21.9 | 55.1 | 46.4 | 72.7 | 59.2 | 80.7 | 64.7 |
+ Bat | 68.7 (+8.6) | 69.6 (+22.6) | 54.9 (+26.9) | 48.9 (+27.0) | 67.2 (+12.1) | 60.7 (+14.3) | 78.6 (+5.9) | 74.7 (+15.5) | 88.8 (+8.1) | 87.8 (+23.2) |
BestCIGL | 70.0 | 66.2 | 54.5 | 45.0 | 71.3 | 68.9 | 83.9 | 80.9 | 89.5 | 86.2 |
+ Bat | 72.8 (+2.9) | 70.2 (+4.0) | 62.5 (+8.0) | 62.1 (+17.1) | 76.9 (+5.6) | 74.9 (+6.0) | 85.4 (+1.6) | 84.6 (+3.7) | 90.7 (+1.2) | 90.0 (+3.8) |
-
*Base: vanilla GCN model; Base+Bat: applying Bat without any other CIGL method; BestCIGL: best CIGL baseline w/o Bat; BestCIGL+Bat: best CIGL baseline w/ Bat;
Bat is robust even under extreme class imbalance. We further extend Table 1 and test Bat’s robustness to varying types and levels of imbalance, as reported in Table 2. In this experiment, we extend the step imbalance ratio from 10 (used in Table 1) to 20 to test Bat under even more challenging class imbalance scenarios. In addition, we consider the natural (long-tail) class imbalance (Park et al., 2022) that is commonly observed in real-world graphs with IR of 50 and 100. Datasets from (Shchur et al., 2018) (CS, Physics) are also included to test Bat on large-scale tasks. Results show that: (1) Bat is robust to extreme class imbalance, and it consistently boosts the CIGL performance by a significant margin under varying types and levels of imbalance. (2) The performance drop from increasing IR is significantly lowered by Bat, i.e., applying Bat improves model’s robustness to extreme class imbalance. (3) Bat’s advantage is even more prominent under higher class imbalance, e.g., on Cora with step IR, the performance gain of applying Bat on Base raised from 8.2 to 18.5 when IR increased from 10 to 20, and similar patterns can be observed in other settings.
Bat effectively alleviates both AMP and DMP. We further design experiments to verify to what extent Bat can effectively handle the topological challenges identified in this paper, i.e., ambivalent and distant message-passing. Specifically, we investigate whether Bat can improve the prediction accuracy of minority class nodes that are highly influenced by AMP/DMP, i.e., with high heterophilic neighbor ratio/long distance to supervision. Results are shown in Fig. 6 (5 independent runs with GCN classifier, IR=10). As can be observed, Bat effectively alleviates the negative impact of AMP and DMP and helps node classifiers to achieve better performance in minority classes.
![Refer to caption](extracted/5605652/figs/mitigate_ambivalent.png)
![Refer to caption](extracted/5605652/figs/mitigate_distant.png)
Bat is computationally efficient. As previously discussed in the complexity analysis, Bat with 0th/1st-order estimation scales linearly with the number of nodes/edges (i.e., with or complexity). Since all the operations can be executed in parallel in matrix form, Bat0/Bat1 has / time complexity, where is the number of available computational units and is usually large for modern GPUs. Table 3 reports the ratio of virtual nodes/edges to the original graph introduced and the running time of Bat. It can be observed that Bat only introduces a small number of virtual nodes/edges, and is highly efficient (taking milliseconds for augmentation) in practice.
Dataset | Nodes (%) | Edges (%) | Time (ms) |
---|---|---|---|
Cora | 0.258% | 2.842%/1.509% | 4.50/4.65ms |
CiteSeer | 0.180% | 3.715%/1.081% | 4.72/4.97ms |
PubMed | 0.015% | 3.175%/1.464% | 6.23/6.64ms |
CS | 0.082% | 1.395%/1.053% | 16.97/18.61ms |
Physics | 0.014% | 0.797%/0.527% | 30.68/31.91ms |
-
* Results obtained on an NVIDIA Tesla V100 32GB GPU.
5 Related Works
Imbalanced graph learning. Class imbalance is ubiquitous in many machine-learning tasks and has been extensively studied (He & Garcia, 2008; Krawczyk, 2016). However, most of the existing works focus on i.i.d. scenarios, which may not be tailored to the unique characteristics of graph data. To handle imbalanced graph learning, several techniques have been proposed in recent studies (e.g., by adversarial training (Shi et al., 2020; Qu et al., 2021), designing new GNN architectures (Wang et al., 2020; Liu et al., 2021) or loss functions (Song et al., 2022a)), we review the most closely related model-agnostic CR methods here. One of the early works GraphSMOTE (Zhao et al., 2021b) adopts SMOTE (Chawla et al., 2002) oversampling in the node embedding space to synthesize minority nodes and complements the topology with a learnable edge predictor. A more recent work GraphENS (Park et al., 2022) synthesizes the ego network through saliency-based ego network mixing to handle the neighbor-overfitting problem. Most studies are rooted in a class-rebalancing perspective and address the imbalance by node/class-wise reweighting or resampling.
Topology-imbalance in graphs. Topology imbalance is firstly discussed in Chen et al. (2021). They found that “the unequal structure role of labeled nodes” can cause influence conflict, and propose to re-weight the labeled nodes based on a conflict detection measure. Other works further discussed how to better address the issue via position-aware structure learning (Sun et al., 2022), and handle topology-imbalance in fake news detection (Gao et al., 2022) and bankruptcy prediction (Liu et al., 2023). These studies discussed concepts related to “influence conflict/insufficiency” (Chen et al., 2021), which motivated us to investigate AMP/DMP in this work.
6 Conclusion
In this paper, we study class-imbalanced graph learning from a novel topological perspective. We theoretically reveal that two fundamental topological phenomena, i.e., ambivalent and distant message-passing, can greatly exacerbate the predictive bias stemming from class imbalance. Our findings reveal an unexplored avenue that limits the performance of existing class-rebalancing-based CIGL techniques. In light of this, we propose Bat to handle the topological challenges in CIGL by dynamic topological augmentation. Bat is a swift and model-agnostic framework that can seamlessly complement other CIGL techniques, augmenting their performance and mitigating predictive bias. Systematic experiments validate Bat’s superior effectiveness, versatility, robustness, and efficiency across various CIGL tasks.
Acknowledgements
This work is supported by NSF (1947135), the NSF Program on Fairness in AI in collaboration with Amazon (1939725), NIFA (2020-67021-32799), DHS (17STQAC00001-07-00), the C3.ai Digital Transformation Institute, MIT-IBM Watson AI Lab, and IBM-Illinois Discovery Accelerator Institute. The content of the information in this document does not necessarily reflect the position or the policy of the Government or Amazon, and no official endorsement should be inferred. The U.S. Government is authorized to reproduce and distribute reprints for Government purposes notwithstanding any copyright notation here on.
Impact Statements
This paper presents work whose goal is to advance the field of Graph Data Mining. There are many potential societal consequences of our work, none which we feel must be specifically highlighted here.
References
- Akoglu et al. (2015) Akoglu, L., Tong, H., and Koutra, D. Graph based anomaly detection and description: a survey. Data mining and knowledge discovery, 29(3):626–688, 2015.
- Antoniadis & Fan (2001) Antoniadis, A. and Fan, J. Regularization of wavelet approximations. Journal of the American Statistical Association, 96(455):939–967, 2001.
- Bojchevski & Günnemann (2017) Bojchevski, A. and Günnemann, S. Deep gaussian embedding of graphs: Unsupervised inductive learning via ranking. arXiv preprint arXiv:1707.03815, 2017.
- Cai et al. (2021) Cai, L., Li, J., Wang, J., and Ji, S. Line graph neural networks for link prediction. IEEE Transactions on Pattern Analysis and Machine Intelligence, 44(9):5103–5113, 2021.
- Chawla et al. (2002) Chawla, N. V., Bowyer, K. W., Hall, L. O., and Kegelmeyer, W. P. Smote: synthetic minority over-sampling technique. Journal of artificial intelligence research, 16:321–357, 2002.
- Chen et al. (2021) Chen, D., Lin, Y., Zhao, G., Ren, X., Li, P., Zhou, J., and Sun, X. Topology-imbalance learning for semi-supervised node classification. Advances in Neural Information Processing Systems, 34:29885–29897, 2021.
- Chien et al. (2020) Chien, E., Peng, J., Li, P., and Milenkovic, O. Adaptive universal generalized pagerank graph neural network. arXiv preprint arXiv:2006.07988, 2020.
- Cui et al. (2019) Cui, Y., Jia, M., Lin, T.-Y., Song, Y., and Belongie, S. Class-balanced loss based on effective number of samples. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 9268–9277, 2019.
- Decelle et al. (2011) Decelle, A., Krzakala, F., Moore, C., and Zdeborová, L. Asymptotic analysis of the stochastic block model for modular networks and its algorithmic applications. Physical Review E, 84(6):066106, 2011.
- Fey & Lenssen (2019) Fey, M. and Lenssen, J. E. Fast graph representation learning with pytorch geometric. arXiv preprint arXiv:1903.02428, 2019.
- Fu & He (2021) Fu, D. and He, J. SDG: A simplified and dynamic graph neural network. In Diaz, F., Shah, C., Suel, T., Castells, P., Jones, R., and Sakai, T. (eds.), SIGIR ’21: The 44th International ACM SIGIR Conference on Research and Development in Information Retrieval, Virtual Event, Canada, July 11-15, 2021, pp. 2273–2277. ACM, 2021. doi: 10.1145/3404835.3463059. URL https://doi.org/10.1145/3404835.3463059.
- Fu et al. (2023) Fu, D., Zhou, D., Maciejewski, R., Croitoru, A., Boyd, M., and He, J. Fairness-aware clique-preserving spectral clustering of temporal graphs. In Ding, Y., Tang, J., Sequeda, J. F., Aroyo, L., Castillo, C., and Houben, G. (eds.), Proceedings of the ACM Web Conference 2023, WWW 2023, Austin, TX, USA, 30 April 2023 - 4 May 2023, pp. 3755–3765. ACM, 2023. doi: 10.1145/3543507.3583423. URL https://doi.org/10.1145/3543507.3583423.
- Fu et al. (2024) Fu, D., Hua, Z., Xie, Y., Fang, J., Zhang, S., Sancak, K., Wu, H., Malevich, A., He, J., and Long, B. Vcr-graphormer: A mini-batch graph transformer via virtual connections. CoRR, abs/2403.16030, 2024. doi: 10.48550/ARXIV.2403.16030. URL https://doi.org/10.48550/arXiv.2403.16030.
- Gao et al. (2022) Gao, L., Song, L., Liu, J., Chen, B., and Shang, X. Topology imbalance and relation inauthenticity aware hierarchical graph attention networks for fake news detection. In Proceedings of the 29th International Conference on Computational Linguistics, pp. 4687–4696, 2022.
- Gasteiger et al. (2018) Gasteiger, J., Bojchevski, A., and Günnemann, S. Predict then propagate: Graph neural networks meet personalized pagerank. arXiv preprint arXiv:1810.05997, 2018.
- Hamilton et al. (2017) Hamilton, W., Ying, Z., and Leskovec, J. Inductive representation learning on large graphs. Advances in neural information processing systems, 30, 2017.
- Hasanzadeh et al. (2020) Hasanzadeh, A., Hajiramezanali, E., Boluki, S., Zhou, M., Duffield, N., Narayanan, K., and Qian, X. Bayesian graph neural networks with adaptive connection sampling. In International conference on machine learning, pp. 4094–4104. PMLR, 2020.
- He & Garcia (2008) He, H. and Garcia, E. A. Learning from imbalanced data. IEEE Transactions on Knowledge & Data Engineering, (9):1263–1284, 2008.
- Holland et al. (1983) Holland, P. W., Laskey, K. B., and Leinhardt, S. Stochastic blockmodels: First steps. Social networks, 5(2):109–137, 1983.
- Hong et al. (2021) Hong, Y., Han, S., Choi, K., Seo, S., Kim, B., and Chang, B. Disentangling label distribution for long-tailed visual recognition. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 6626–6636, 2021.
- Hu et al. (2020) Hu, W., Fey, M., Zitnik, M., Dong, Y., Ren, H., Liu, B., Catasta, M., and Leskovec, J. Open graph benchmark: Datasets for machine learning on graphs. Advances in neural information processing systems, 33:22118–22133, 2020.
- Huang et al. (2022) Huang, Z., Tang, Y., and Chen, Y. A graph neural network-based node classification model on class-imbalanced graph data. Knowledge-Based Systems, 244:108538, 2022.
- Japkowicz & Stephen (2002) Japkowicz, N. and Stephen, S. The class imbalance problem: A systematic study. Intelligent data analysis, 6(5):429–449, 2002.
- Johnson & Khoshgoftaar (2019) Johnson, J. M. and Khoshgoftaar, T. M. Survey on deep learning with class imbalance. Journal of Big Data, 6(1):1–54, 2019.
- Kang et al. (2019) Kang, B., Xie, S., Rohrbach, M., Yan, Z., Gordo, A., Feng, J., and Kalantidis, Y. Decoupling representation and classifier for long-tailed recognition. arXiv preprint arXiv:1910.09217, 2019.
- Kang et al. (2020) Kang, J., He, J., Maciejewski, R., and Tong, H. Inform: Individual fairness on graph mining. In Proceedings of the 26th ACM SIGKDD international conference on knowledge discovery & data mining, pp. 379–389, 2020.
- Kang et al. (2022a) Kang, J., Zhou, Q., and Tong, H. Jurygcn: quantifying jackknife uncertainty on graph convolutional networks. In Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pp. 742–752, 2022a.
- Kang et al. (2022b) Kang, J., Zhu, Y., Xia, Y., Luo, J., and Tong, H. Rawlsgcn: Towards rawlsian difference principle on graph convolutional network. In Proceedings of the ACM Web Conference 2022, pp. 1214–1225, 2022b.
- Kingma & Ba (2014) Kingma, D. P. and Ba, J. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980, 2014.
- Krawczyk (2016) Krawczyk, B. Learning from imbalanced data: open challenges and future directions. Progress in Artificial Intelligence, 5(4):221–232, 2016.
- Li et al. (2022) Li, X., Wen, L., Deng, Y., Feng, F., Hu, X., Wang, L., and Fan, Z. Graph neural network with curriculum learning for imbalanced node classification. arXiv preprint arXiv:2202.02529, 2022.
- Liu et al. (2021) Liu, Y., Ao, X., Qin, Z., Chi, J., Feng, J., Yang, H., and He, Q. Pick and choose: a gnn-based imbalanced learning approach for fraud detection. In Proceedings of the Web Conference 2021, pp. 3168–3177, 2021.
- Liu et al. (2023) Liu, Y., Gao, Z., Liu, X., Luo, P., Yang, Y., and Xiong, H. Qtiah-gnn: Quantity and topology imbalance-aware heterogeneous graph neural network for bankruptcy prediction. In Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pp. 1572–1582, 2023.
- Liu et al. (2020) Liu, Z.-Y., Li, S.-Y., Chen, S., Hu, Y., and Huang, S.-J. Uncertainty aware graph gaussian process for semi-supervised learning. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 34, pp. 4957–4964, 2020.
- Ma et al. (2021) Ma, Y., Liu, X., Zhao, T., Liu, Y., Tang, J., and Shah, N. A unified view on graph neural networks as graph signal denoising. In Proceedings of the 30th ACM International Conference on Information & Knowledge Management, pp. 1202–1211, 2021.
- Nt & Maehara (2019) Nt, H. and Maehara, T. Revisiting graph neural networks: All we have is low-pass filters. arXiv preprint arXiv:1905.09550, 2019.
- Pandey et al. (2019) Pandey, B., Bhanodia, P. K., Khamparia, A., and Pandey, D. K. A comprehensive survey of edge prediction in social networks: Techniques, parameters and challenges. Expert Systems with Applications, 124:164–181, 2019.
- Park et al. (2022) Park, J., Song, J., and Yang, E. Graphens: Neighbor-aware ego network synthesis for class-imbalanced node classification. In International Conference on Learning Representations, 2022.
- Paszke et al. (2019) Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., Antiga, L., et al. Pytorch: An imperative style, high-performance deep learning library. Advances in neural information processing systems, 32, 2019.
- Qu et al. (2021) Qu, L., Zhu, H., Zheng, R., Shi, Y., and Yin, H. Imgagn: Imbalanced network embedding via generative adversarial graph networks. In Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining, pp. 1390–1398, 2021.
- Sen et al. (2008) Sen, P., Namata, G., Bilgic, M., Getoor, L., Galligher, B., and Eliassi-Rad, T. Collective classification in network data. AI magazine, 29(3):93–93, 2008.
- Shchur et al. (2018) Shchur, O., Mumme, M., Bojchevski, A., and Günnemann, S. Pitfalls of graph neural network evaluation. arXiv preprint arXiv:1811.05868, 2018.
- Shi et al. (2020) Shi, M., Tang, Y., Zhu, X., Wilson, D., and Liu, J. Multi-class imbalanced graph convolutional network learning. In Proceedings of the Twenty-Ninth International Joint Conference on Artificial Intelligence (IJCAI-20), 2020.
- Song et al. (2022a) Song, J., Park, J., and Yang, E. Tam: Topology-aware margin loss for class-imbalanced node classification. In International Conference on Machine Learning, pp. 20369–20383. PMLR, 2022a.
- Song et al. (2022b) Song, Z., Yang, X., Xu, Z., and King, I. Graph-based semi-supervised learning: A comprehensive review. IEEE Transactions on Neural Networks and Learning Systems, 2022b.
- Srivastava et al. (2014) Srivastava, N., Hinton, G., Krizhevsky, A., Sutskever, I., and Salakhutdinov, R. Dropout: a simple way to prevent neural networks from overfitting. The journal of machine learning research, 15(1):1929–1958, 2014.
- Stadler et al. (2021) Stadler, M., Charpentier, B., Geisler, S., Zügner, D., and Günnemann, S. Graph posterior network: Bayesian predictive uncertainty for node classification. Advances in Neural Information Processing Systems, 34:18033–18048, 2021.
- Sun et al. (2022) Sun, Q., Li, J., Yuan, H., Fu, X., Peng, H., Ji, C., Li, Q., and Yu, P. S. Position-aware structure learning for graph topology-imbalance by relieving under-reaching and over-squashing. In Proceedings of the 31st ACM International Conference on Information & Knowledge Management, pp. 1848–1857, 2022.
- Tang & Liu (2010) Tang, L. and Liu, H. Community detection and mining in social media. Synthesis lectures on data mining and knowledge discovery, 2(1):1–137, 2010.
- Veličković et al. (2018) Veličković, P., Cucurull, G., Casanova, A., Romero, A., Liò, P., and Bengio, Y. Graph attention networks. In International Conference on Learning Representations, 2018.
- Wang et al. (2020) Wang, Z., Ye, X., Wang, C., Cui, J., and Philip, S. Y. Network embedding with completely-imbalanced labels. IEEE Transactions on Knowledge and Data Engineering, 33(11):3634–3647, 2020.
- Welling & Kipf (2016) Welling, M. and Kipf, T. N. Semi-supervised classification with graph convolutional networks. In J. International Conference on Learning Representations (ICLR 2017), 2016.
- Wu et al. (2019) Wu, F., Souza, A., Zhang, T., Fifty, C., Yu, T., and Weinberger, K. Simplifying graph convolutional networks. In International conference on machine learning, pp. 6861–6871. PMLR, 2019.
- Wu et al. (2022) Wu, L., Xia, J., Gao, Z., Lin, H., Tan, C., and Li, S. Z. Graphmixup: Improving class-imbalanced node classification by reinforcement mixup and self-supervised context prediction. In Joint European Conference on Machine Learning and Knowledge Discovery in Databases, pp. 519–535. Springer, 2022.
- Xu et al. (2023) Xu, Z., Chen, Y., Zhou, Q., Wu, Y., Pan, M., Yang, H., and Tong, H. Node classification beyond homophily: Towards a general solution. In Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pp. 2862–2873, 2023.
- Yan et al. (2021a) Yan, Y., Liu, L., Ban, Y., **g, B., and Tong, H. Dynamic knowledge graph alignment. In Proceedings of the AAAI conference on artificial intelligence, volume 35, pp. 4564–4572, 2021a.
- Yan et al. (2021b) Yan, Y., Zhang, S., and Tong, H. Bright: A bridging algorithm for network alignment. In Proceedings of the web conference 2021, pp. 3907–3917, 2021b.
- Yan et al. (2024) Yan, Y., Chen, Y., Chen, H., Xu, M., Das, M., Yang, H., and Tong, H. From trainable negative depth to edge heterophily in graphs. Advances in Neural Information Processing Systems, 36, 2024.
- Yun et al. (2022) Yun, S., Kim, K., Yoon, K., and Park, C. Lte4g: Long-tail experts for graph neural networks. In Proceedings of the 31st ACM International Conference on Information & Knowledge Management, pp. 2434–2443, 2022.
- Zeng et al. (2023a) Zeng, Z., Zhang, S., Xia, Y., and Tong, H. Parrot: Position-aware regularized optimal transport for network alignment. In Proceedings of the ACM Web Conference 2023, pp. 372–382, 2023a.
- Zeng et al. (2023b) Zeng, Z., Zhu, R., Xia, Y., Zeng, H., and Tong, H. Generative graph dictionary learning. In International Conference on Machine Learning, pp. 40749–40769. PMLR, 2023b.
- Zeng et al. (2024) Zeng, Z., Du, B., Zhang, S., Xia, Y., Liu, Z., and Tong, H. Hierarchical multi-marginal optimal transport for network alignment. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pp. 16660–16668, 2024.
- Zhang et al. (2019) Zhang, Y., Pal, S., Coates, M., and Ustebay, D. Bayesian graph convolutional neural networks for semi-supervised classification. In Proceedings of the AAAI conference on artificial intelligence, volume 33, pp. 5829–5836, 2019.
- Zhao et al. (2021a) Zhao, J., Dong, Y., Ding, M., Kharlamov, E., and Tang, J. Adaptive diffusion in graph neural networks. Advances in Neural Information Processing Systems, 34:23321–23333, 2021a.
- Zhao et al. (2021b) Zhao, T., Zhang, X., and Wang, S. Graphsmote: Imbalanced node classification on graphs with graph neural networks. In Proceedings of the 14th ACM international conference on web search and data mining, pp. 833–841, 2021b.
- Zhao et al. (2022) Zhao, T., Luo, D., Zhang, X., and Wang, S. Topoimb: Toward topology-level imbalance in learning from graphs. In Learning on Graphs Conference, pp. 37–1. PMLR, 2022.
- Zhao et al. (2020) Zhao, X., Chen, F., Hu, S., and Cho, J.-H. Uncertainty aware semi-supervised learning on graph data. Advances in Neural Information Processing Systems, 33:12827–12836, 2020.
- Zhu et al. (2020) Zhu, J., Yan, Y., Zhao, L., Heimann, M., Akoglu, L., and Koutra, D. Beyond homophily in graph neural networks: Current limitations and effective designs. Advances in Neural Information Processing Systems, 33:7793–7804, 2020.
Appendix
- •
- •
- •
- •
Appendix A Proofs of Theoretical Results
Define random variables denote the number of class- -hop homo-connected neighbors of a node . Note that the results of both Theorems 2.1 & 2.2 depend only on the distributions of . Thus, we will first derive the limiting distributions of as a technical lemma, and then give the proofs of Theorems 2.1 & 2.2.
A.1 Limiting Distributions of
To count the number of homo-connected neighbors, consider the breadth-first search (BFS) tree rooted at node . By enumerating the numbers of -hop homo-connected neighbors in the BFS tree respectively, we can calculate the exact joint distribution of :
Thus, and are independent, and their marginal distributions are:
Now consider . Let
Then, the limiting distributions of and are:
Figure 7 shows that the limiting distributions are good approximation for finite .
![Refer to caption](x2.png)
A.2 Proof of Theorem 2.1
.
Note that for any ,
Thus,
Similarly,
It follows that
A.3 Proof of Theorem 2.2
.
For , note the identity:
It follows that (with and )
For , similarly,
In general, the result for has nested exponentiations, but we still have:
Similarly,
By the law of total probability and the independence of and ,
It follows that
Appendix B Reproducibility
In this section, we describe the detailed experimental settings including (§B.1) data statistics, (§B.2) baseline settings, and (§B.3) evaluation protocols. The source code for implementing and evaluating Bat and all the CIGL baseline methods will be released after the paper is published.
B.1 Data Statistics
As previously described, we adopt 5 benchmark graph datasets: the Cora, CiteSeer, and PubMed citation networks (Sen et al., 2008), and the CS and Physics coauthor networks (Shchur et al., 2018) to test Bat on large graphs with more nodes and high-dimensional features. All datasets are publicly available222https://pytorch-geometric.readthedocs.io/en/latest/modules/datasets.html.. Table 4 summarizes the dataset statistics.
Dataset | #nodes | #edges | #features | #classes |
---|---|---|---|---|
Cora | 2,708 | 10,556 | 1,433 | 7 |
CiteSeer | 3,327 | 9,104 | 3,703 | 6 |
PubMed | 19,717 | 88,648 | 500 | 3 |
CS | 18,333 | 163,788 | 6,805 | 15 |
Physics | 34,493 | 495,924 | 8,415 | 5 |
We follow previous works (Zhao et al., 2021b; Park et al., 2022; Song et al., 2022a) to construct and adjust the class imbalanced node classification tasks. For step imbalance, we select half of the classes () as minority classes and the rest as majority classes. We follow the public split (Sen et al., 2008) for semi-supervised node classification where each class has 20 training nodes, then randomly remove minority class training nodes until the given imbalance ratio (IR) is met. The imbalance ratio is defined as , i.e., more imbalanced data has higher IR. For natural imbalance, we simulate the long-tail class imbalance present in real-world data by utilizing a power-law distribution. Specifically, for a given IR, the largest head class have training nodes, and the smallest tail class have 1 training node. The number of training nodes of the -th class is determined by . We set the IR (largest class to smallest class) to 50/100 to test Bat’s robustness under natural and extreme class imbalance. We show the training data distribution under step and natural imbalance in Fig. 8.
![Refer to caption](extracted/5605652/figs/train_dist.png)
B.2 Baseline Settings
To fully validate Bat’s performance and compatibility with existing CIGL techniques and GNN backbones, we include six baseline methods with five popular GNN backbones in our experiments, and combine Bat with them under all possible combinations. The included CIGL baselines can be generally divided into two categories: reweighting-based (i.e., Reweight (Japkowicz & Stephen, 2002), ReNode (Chen et al., 2021)) and augmentation-based (i.e., Oversampling (Japkowicz & Stephen, 2002), SMOTE (Chawla et al., 2002), GraphSMOTE (Zhao et al., 2021b), and GraphENS (Park et al., 2022)).
-
•
Reweight (Japkowicz & Stephen, 2002) assigns minority classes with higher misclassification costs (i.e., weights in the loss function) by the inverse of the class frequency in the training set.
-
•
ReNode (Chen et al., 2021) measures the influence conflict of training nodes, and perform instance-wise node reweighting to alleviate the topology imbalance.
-
•
Oversample (Japkowicz & Stephen, 2002) augments minority classes with additional synthetic nodes by replication-base oversampling.
-
•
SMOTE (Chawla et al., 2002) synthesizes minority nodes by 1) randomly selecting a seed node, 2) finding its -nearest neighbors in the feature space, and 3) performing linear interpolation between the seed and one of its -nearest neighbors.
- •
-
•
GraphENS (Park et al., 2022) directly synthesize the whole ego network (node with its 1-hop neighbors) for minority classes by similarity-based ego network combining and saliency-based node mixing to prevent neighbor memorization.
Baseline implementation details. We use the public implementations333https://github.com/victorchen96/renode444https://github.com/TianxiangZhao/GraphSmote555https://github.com/JoonHyung-Park/GraphENS of the baseline methods for a fair comparison. For ReNode (Chen et al., 2021), we use its transductive version and search hyperparameters among the lower bound of cosine annealing and upper bound of the cosine annealing following the original paper. We set the teleport probability of PageRank as given by the default setting in the released implementation. As Oversample (Cui et al., 2019) and SMOTE (Chawla et al., 2002) were not proposed to handle graph data, we adopt their enhanced versions provided by GraphSMOTE (Zhao et al., 2021b), which also duplicate the edges from the seed nodes to the synthesized nodes in order to connect them to the graph. For GraphSMOTE (Zhao et al., 2021b), we use the version that predicts edges with binary values as it performs better than the variant with continuous edge predictions in many datasets. For GraphENS (Park et al., 2022), we follow the settings in the paper: distribution is used to sample , the feature masking hyperparameter and temperature are tuned among and , and the number of warmup epochs is set to 5.
Combining Bat and baseline CIGL techniques. Since Bat only manipulates the graph data and remains independent of the model architecture, it seamlessly integrates with the aforementioned CIGL techniques. During each training epoch, Bat enhances the original graph using the current model and yields the augmented graph . Subsequently, other CIGL methods operate on the augmented graph . Specifically, loss function engineering methods (Reweight and ReNode) perform loss computation and backpropagation based on , and data augmentation methods (Resampling, SMOTE, GSMOTE, GENS) carry out additional class-balancing operations on , generating new minority nodes based on its structure.
GNN backbone implementation details. We use pytorch (Paszke et al., 2019) and torch_geometric (Fey & Lenssen, 2019) to implement all five GNN backbones used in this paper, i.e., GCN (Welling & Kipf, 2016), GAT (Veličković et al., 2018), GraphSAGE (Hamilton et al., 2017), APPNP (Gasteiger et al., 2018), and GPRGNN (Chien et al., 2020). Most of our settings are aligned with prevailing works (Park et al., 2022; Chen et al., 2021; Song et al., 2022a) to obtain fair and comparable results. Specifically, we implement all GNNs’ convolution layer with ReLU activation and dropout (Srivastava et al., 2014) with a drop** rate of 0.5 before the last layer. For GAT, we set the number of attention heads to 4. For APPNP and GPRGNN, we follow the best setting in the original paper and use 2 APPNP/GPR_prop convolution layers with 64 hidden units. Note that GraphENS’s official implementation requires modifying the graph convolution for resampling and thus cannot be directly combined with APPNP and GPRGNN. The teleport probability = 0.1 and the number of power iteration steps K = 10. We search for the best architecture for other backbones from #layers and hidden dimension based on the average of validation accuracy and F1 score. We train each GNN for 2,000 epochs using Adam optimizer (Kingma & Ba, 2014) with an initial learning rate of 0.01. To achieve better convergence, we follow (Park et al., 2022) to use 5e-4 weight decay and adopt the ReduceLROnPlateau scheduler in Pytorch, which reduces the learning rate by half if the validation loss does not improve for 100 epochs.
B.3 Evaluation Protocol
To evaluate the predictive performance on class-imbalanced data, we use two balanced metrics, i.e., balanced accuracy (BAcc.) and macro-averaged F1 score (Macro-F1). They compute accuracy/F1-score for each class independently and use the unweighted average mean as the final score, i.e., BAcc. , Macro-F1 . Additionally, we use performance standard deviation (PerfStd) to evaluate the level of model predictive bias. Formally, let be the classification accuracy of class , the PerfStd is defined as the standard deviation of the accuracy scores of all classes, i.e., . All the experiments are conducted on a Linux server with Intel Xeon Gold 6240R CPU and NVIDIA Tesla V100 32GB GPU.
Appendix C Extended Discussions
In this section, we present an ablation study (§C.1) validate the effectiveness and efficiency of the key modules, then we discuss how to further speed up Bat in practice (§C.2); how to choose between Bat0 and Bat1 in practice (§C.3); and finally, the limitation and future works (§C.4).
C.1 Ablation Study
We present an ablation study to validate the effectiveness and efficiency of the key modules in Bat. Specifically, for node risk estimation, we compare our total-variation-distance-based uncertainty with (i) the naïve random assignment that drawn uncertainty score from a uniform distribution and (ii) the information entropy . We substitute the original uncertainty metric with these aforementioned methods in Bat0, and assess their impact on performance as well as the computational time required for uncertainty estimation. It is worth noting that in practical implementation, the computation is parallelized on GPU (assuming sufficient GPU memory). Therefore, the computational time of a given uncertainty measure remains consistent across the three datasets we employed (Cora, CiteSeer, PubMed with IR=10). The detailed results are presented in Table 5, revealing that: (i) Randomly assigned uncertainty scores significantly impede the performance of Bat, resulting in a large drop in both balanced accuracy and Marco-f1. (ii) In comparison to our approach, employing information entropy as the node uncertainty score necessitates 2.3x computation time, yet the influence on performance remains marginal.
Uncertainty | Cora | CiteSeer | PubMed | Computation | |||
---|---|---|---|---|---|---|---|
BAcc | Macro-F1 | BAcc | Macro-F1 | BAcc | Macro-F1 | Time(ms) | |
Random | 61.641.89 | 59.441.71 | 46.592.29 | 44.373.23 | 61.601.69 | 58.131.81 | 0.0249 |
Information Entropy | 65.181.68 | 63.111.91 | 51.872.96 | 50.363.43 | 67.721.27 | 67.191.57 | 0.1257 |
TVDistance (ours) | 65.541.25 | 63.281.07 | 52.651.08 | 51.551.28 | 68.620.77 | 67.161.53 | 0.0543 |
Further, we conduct an ablation study for our posterior likelihood estimation strategy by comparing our -order (Bat0) and -order (Bat1) likelihood estimation methods with the random method that assigns (unnormalized) node-class likelihood by drawing from a uniform distribution . Results are shown in Table 6. We can observe that the random method significantly worsens the predictive performance on all CIGL tasks. Altogether, the ablation study results confirm the effectiveness and efficiency of the design of Bat, showcasing its ability to deliver strong performance with minimal computational overhead.
Estimation | Cora | CiteSeer | PubMed | Computation | |||
---|---|---|---|---|---|---|---|
BAcc | Macro-F1 | BAcc | Macro-F1 | BAcc | Macro-F1 | Time(ms) | |
Random | 63.852.17 | 61.942.68 | 46.513.27 | 41.704.33 | 64.321.23 | 53.582.48 | 0.0883 |
0th-order (Bat0) | 65.541.25 | 63.281.07 | 52.651.08 | 51.551.28 | 68.620.77 | 67.161.53 | 0.1251 |
1st-order (Bat1) | 69.801.30 | 68.681.49 | 55.371.39 | 54.941.44 | 67.573.22 | 64.403.68 | 0.3030 |
C.2 On the Further Speedup of Bat
As stated in the paper, thanks to its simple and efficient design, Bat can be integrated into the GNN training process to perform dynamic topology augmentation based on the training state. By default, we run Bat in every iteration of GNN training, i.e., the granularity of applying Bat is 1, as described in Alg. 1. However, we note that in practice, this granularity can be increased to further reduce the cost of applying Bat. This operation can result in a significant linear speedup ratio: setting the granularity to reduces the computational overhead of Bat to of the original (i.e., x speedup ratio), with minor performance degradation. This could be helpful for scaling Bat to large-scale graphs in practice. In this section, we design experiments to validate the influence of different Bat granularity (i.e., the number of iterations per each use of Bat) in real-world CIGL tasks. We set the granularity to 1/5/10/50/100 and test the performance of BatT with a vanilla GCN classifier on the Cora/CiteSeer/PubMed dataset with an imbalance ratio of 10. Fig. 9 shows the empirical results from 10 independent runs. The red horizontal line in each subfigure represents the baseline (vanilla GCN) performance. It can be observed that setting a larger Bat granularity is an effective way to further speed up Bat in practice. The performance drop of adopting this trick is relatively minor, especially considering the significant linear speedup ratio it brings. The predictive performance boost brought by Bat is still significant even with a large granularity at 100 (i.e., with 100x Bat speedup).
![Refer to caption](extracted/5605652/figs/granularity_bacc.png)
![Refer to caption](extracted/5605652/figs/granularity_f1.png)
C.3 Choosing between Bat0 and Bat1
In this section, we summarize the strengths and limitations of Bat0 and Bat1, and give suggestions for choosing between them in practice. In short, we recommend using Bat1 to achieve better classification performance. But in case that computational resources are limited, Bat0 can serve as a more efficient alternative. The reasons are as follows:
Performance. We observe a noticeable performance gap between Bat0 and Bat1, wherein Bat1 consistently demonstrates superior classification performance due to its incorporation of local topological structure. Across the 15 scenarios outlined in Table 1 (the best scores for 3 datasets x 5 GNN backbones): (1) Bat1 outperforms Bat0 significantly in 12 out of 15 scenarios for BAcc/F1 scores, with an average F1 advantage of 1.692 in the 11 leading cases. (2) Conversely, Bat0 exhibits a less pronounced advantage in the instances where it outperforms Bat1, with an average advantage of 0.518 in the 4 leading scenarios666Despite Bat1 holding a relative performance edge, both Bat0 and Bat1 substantially enhance the performance of the best-performing CIGL baseline methods. Over the 15 scenarios, Bat0 yields an average improvement of 4.789/5.848 in the best BAcc/F1, while Bat1 brings an average improvement of 5.805/6.950..
Efficiency. On the other hand, it’s worth noting that Bat1 generally incurs higher time and space complexity compared to Bat0. Specifically, Bat0 demonstrates linear complexity concerning the number of nodes, whereas Bat1 exhibits linear growth in complexity with the number of edges. Given that real-world graph data often features a significantly larger number of edges than nodes, Bat0 is usually the more efficient option (especially for densely connected graphs).
C.4 Limitations and Future Works
One potential limitation of the proposed Bat framework is its reliance on exploiting model prediction for risk and likelihood estimation. This strategy may not provide accurate estimation when the model itself exhibits extremely poor predictive performance. However, this rarely occurs in practice and can be prevented by more careful fine-tuning of parameters and model architectures. In addition to this, as described in Section 3, we adopt several fast measures to estimate node uncertainty, prediction risk, and posterior likelihood for the sake of efficiency. Other techniques for such purposes (e.g., deterministic (Liu et al., 2020; Zhao et al., 2020)/Bayesian (Zhang et al., 2019; Hasanzadeh et al., 2020)/Jackknife (Kang et al., 2022a) uncertainty estimation) can be easily integrated into the proposed Bat framework, although the computational efficiency might be a major bottleneck. How to exploit alternative uncertainty/risk measures while retaining computational efficiency is an interesting future direction.
Beyond the class imbalance in the node label distribution, graph data can also exhibit multi-facet skewness in other aspects. For instance, class imbalance may also exist in edge-level (e.g., in edge classification/prediction (Cai et al., 2021; Pandey et al., 2019)) and graph-level (e.g., in graph classification/alignment (Zeng et al., 2023a, 2024; Yan et al., 2021b)). Handling class imbalance can be even more challenging on evolving/dynamic graphs (Fu & He, 2021; Yan et al., 2021a). Beyond the quantity imbalance among classes, skewness may also exists in the topological structure, such as degree imbalance (Kang et al., 2022b), and motif-level imbalance (Zhao et al., 2022; Zeng et al., 2023b). How to jointly consider the multi-facet node/edge/graph-level imbalance to benefit more graph learning tasks is an exciting future direction. Finally, recent work on fairness-aware graph learning (Kang et al., 2020; Fu et al., 2023) has observed that the topology of the graph can also introduce bias/discrimination towards certain groups: extending BAT’s concept to mitigate group unfairness through topology augmentation poses an intriguing future direction.
Appendix D Additional Experimental Results and Analysis
D.1 Results on Additional Large-scale Graphs
To further test the scalability of Bat and the baseline CIGL methods, we extended the experiment to two large-scale graph datasets, CoraFull (19,793 nodes, 126,842 edges) (Bojchevski & Günnemann, 2017) and arXiv (169,343 nodes, 1,166,243 edges) (Hu et al., 2020). To ensure comprehensive and fair comparisons, we followed the same protocol as Table 1 to evaluate six baseline methods and the performance gains brought by Bat. We used GCN as the backbone. It is worth noting that CIGL on these large graphs is a highly challenging task due to (i) the inherent task complexity (with 70/172 classes), (ii) the label scarcity for numerous minority classes, and (iii) the requirement for the algorithm’s scalability. This may be the reason why most existing works (ReNode (Chen et al., 2021), GraphSMOTE (Zhao et al., 2021b), GraphENS (Park et al., 2022), TAM (Song et al., 2022a), LTE4G (Yun et al., 2022), etc.) did not consider these large datasets. Nevertheless, we conduct the experiments report the results in Table 7. We can observe that:
-
•
Bat can scale to large-scale CIGL tasks, consistently and significantly enhancing the performance of various CIGL baselines (overall with 10% relative performance gain).
-
•
The linear complexity of Bat makes it scalable to large graphs, while GraphENS, ReNode, and GraphSMOTE face scalability issues when applied to the arXiv dataset (with 169,343 nodes) due to their space complexity.
Metric | Dataset | Setting | CIGL Baseline | ||||||
---|---|---|---|---|---|---|---|---|---|
ERM | Reweight | ReNode | Resample | SMOTE | GraphSMOTE | GraphENS | |||
BAcc. | CoraFull | Base | 36.12 | 40.93 | 40.10 | 35.71 | 35.59 | 39.25 | 43.76 |
+Bat0 | 39.40 (+3.28) | 43.81 (+2.88) | 42.78 (+2.68) | 40.77 (+5.05) | 40.31 (+4.72) | 43.71 (+4.46) | 46.20 (+2.44) | ||
+Bat1 | 40.88 (+4.76) | 43.77 (+2.84) | 42.82 (+2.72) | 40.67 (+4.95) | 41.04 (+5.46) | 43.70 (+4.45) | 47.19 (+3.44) | ||
OGBN-arXiv | Base | 32.20 | 35.69 | OOM | 32.24 | 32.18 | OOM | OOM | |
+Bat0 | 34.36 (+2.16) | 37.59 (+1.90) | OOM | 37.20 (+4.96) | 36.88 (+4.69) | OOM | OOM | ||
+Bat1 | 36.57 (+4.37) | 39.28 (+3.60) | OOM | 37.36 (+5.12) | 37.50 (+5.32) | OOM | OOM | ||
Macro-F1 | CoraFull | Base | 33.54 | 38.86 | 37.98 | 32.84 | 32.70 | 37.70 | 41.28 |
+Bat0 | 37.25 (+3.71) | 41.21 (+2.35) | 40.58 (+2.61) | 39.04 (+6.19) | 38.35 (+5.65) | 41.25 (+3.55) | 43.90 (+2.61) | ||
+Bat1 | 38.58 (+5.04) | 41.16 (+2.30) | 40.37 (+2.39) | 38.48 (+5.64) | 39.18 (+6.48) | 41.75 (+4.05) | 44.61 (+3.33) | ||
OGBN-arXiv | Base | 29.90 | 32.14 | OOM | 30.16 | 29.96 | OOM | OOM | |
+Bat0 | 32.42 (+2.52) | 34.51 (+2.38) | OOM | 34.50 (+4.34) | 34.44 (+4.48) | OOM | OOM | ||
+Bat1 | 33.99 (+4.08) | 34.78 (+2.64) | OOM | 34.55 (+4.38) | 34.69 (+4.74) | OOM | OOM |
-
*OOM: Out-Of-Memory on a NVIDIA Tesla V100 32GB GPU.
D.2 Comparison with Additional CIGL Baselines
In the main results, we chose representative model-independent CIGL baselines to test Bat’s ability to cooperate and boost various CIGL techniques. In this section, we compare Bat with other representative CIGL techniques that are either model-dependent or with design constraints making them incompatible with Bat. Specifically, we further include three CIGL baselines LTE4G (Yun et al., 2022), GraphMixup (Wu et al., 2022), TAM (Song et al., 2022a), and compare them with Bat on seven datasets (including the newly introduced large-scale graphs CoraFull and arXiv in Section D.1). We use GCN as the backbone, and follow the main experiment protocol used in Table 1 to ensure fair and comparable results.
The results are reported in Table 8. In short, we observe that: (i) Bat consistently demonstrates better performance and scalability compared to the additional baselines. (ii) Some methods (such as GraphMixup and LTE4G) exhibit higher space complexity and thus cannot scale to large datasets. (iii) Among the baseline methods, LTE4G achieves the overall best performance due to its three-stage training (encoder pre-training, expert training, student training) and divide-and-conquer strategy. We note that its complex training and inference strategies introduce additional computational overhead and make it challenging to integrate with other methods. In comparison, Bat has better compatibility and can further achieve better performance with existing class-rebalancing techniques.
Metric | Method | Dataset | ||||||
---|---|---|---|---|---|---|---|---|
Cora | CiteSeer | PubMed | CS | Physics | CoraFull | arXiv | ||
BAcc. | GCN | 61.6 | 37.6 | 64.2 | 75.4 | 80.1 | 32.2 | 36.1 |
GraphMixup | 64.1 | 50.8 | OOM | OOM | OOM | OOM | OOM | |
LTE4G | 68.0 | 51.7 | 63.7 | 77.2 | 81.7 | 36.6 | OOM | |
TAM | 65.3 | 50.1 | 65.7 | 77.0 | 82.8 | 35.0 | 36.3 | |
Bat (Ours) | 69.8 | 55.4 | 68.6 | 82.6 | 87.6 | 36.6 | 40.1 | |
Macro-F1 | GCN | 60.1 | 28.1 | 55.1 | 72.7 | 80.7 | 33.5 | 29.9 |
GraphMixup | 62.7 | 47.3 | OOM | OOM | OOM | OOM | OOM | |
LTE4G | 67.1 | 49.7 | 61.7 | 76.7 | 82.1 | 37.7 | OOM | |
TAM | 65.7 | 43.4 | 63.5 | 75.2 | 82.3 | 32.8 | 31.2 | |
Bat (Ours) | 68.7 | 54.9 | 67.2 | 78.6 | 88.8 | 38.6 | 34.0 |
-
*OOM: Out-Of-Memory on a NVIDIA Tesla V100 32GB GPU.
D.3 Full Empirical Results with Additional GNN Backbones
Due to space limitation, we report the key results of our experiments in Table 1 and 2. We now provide complete results for all settings with the standard error of 5 independent runs. Specifically, Table 9 & 10 & 11 complement Table 1, and Table 12 complements Table 2. We further include APPNP (Gasteiger et al., 2018) and GPRGNN (Chien et al., 2020) as additional GNN backbones for supported CIGL techniques. Note that the official codebase of (Park et al., 2022) only implemented GraphENS using modified (to enable saliency-based mixup) GCN, GAT, and GraphSAGE backbones, and extending it to other GNN backbones is difficult. The results indicate that Bat can consistently boost various CIGL baselines with all GNN backbones under different types and levels of class imbalance, which aligns with our conclusions in the paper.
Dataset (IR=10) | Cora | CiteSeer | PubMed | |||||||
---|---|---|---|---|---|---|---|---|---|---|
Metric: BAcc. | Base | + Bat0 | + Bat1 | Base | + Bat0 | + Bat1 | Base | + Bat0 | + Bat1 | |
GCN | Vanilla | 61.561.24 | 65.541.25 | 69.801.30 | 37.621.61 | 52.651.08 | 55.371.39 | 64.231.55 | 68.620.77 | 67.573.22 |
Reweight | 67.650.64 | 70.971.28 | 72.140.72 | 42.492.66 | 57.910.98 | 58.361.09 | 71.202.33 | 74.191.12 | 73.370.96 | |
ReNode | 66.601.33 | 71.370.62 | 71.841.25 | 42.571.05 | 57.470.62 | 59.280.59 | 71.522.16 | 73.200.71 | 72.531.62 | |
Resample | 59.481.53 | 72.510.68 | 74.240.91 | 39.152.05 | 57.900.33 | 58.781.44 | 64.971.94 | 72.530.85 | 72.871.16 | |
SMOTE | 58.271.05 | 72.160.53 | 73.891.06 | 39.271.90 | 60.060.81 | 61.971.19 | 64.411.95 | 73.170.84 | 73.130.77 | |
GSMOTE | 67.991.37 | 68.520.81 | 71.550.50 | 45.051.95 | 57.681.03 | 57.651.18 | 73.990.88 | 73.091.30 | 76.570.42 | |
GENS | 70.120.43 | 72.220.57 | 72.580.58 | 56.011.17 | 60.600.63 | 62.670.42 | 73.661.04 | 76.110.60 | 76.911.03 | |
Best | 70.12 | 72.51 | 74.24 | 56.01 | 60.60 | 62.67 | 73.99 | 76.11 | 76.91 | |
GAT | Vanilla | 61.531.13 | 66.270.83 | 70.131.07 | 39.251.84 | 55.661.23 | 60.341.66 | 65.460.69 | 73.190.86 | 74.751.18 |
Reweight | 66.941.24 | 71.800.48 | 71.610.85 | 41.293.39 | 59.330.51 | 61.230.99 | 68.371.41 | 75.301.07 | 74.521.14 | |
ReNode | 66.810.98 | 72.141.24 | 70.311.38 | 43.251.78 | 58.261.98 | 59.050.88 | 71.182.13 | 75.551.01 | 75.220.84 | |
Resample | 57.761.73 | 71.900.88 | 73.291.08 | 35.971.42 | 60.101.26 | 60.330.75 | 65.140.86 | 73.270.61 | 73.890.40 | |
SMOTE | 58.810.64 | 70.500.44 | 72.190.75 | 36.951.86 | 60.591.19 | 62.361.18 | 64.811.47 | 73.900.68 | 74.080.51 | |
GSMOTE | 64.681.02 | 69.291.82 | 71.140.96 | 41.821.14 | 56.111.23 | 57.712.58 | 68.721.69 | 74.650.65 | 74.411.57 | |
GENS | 69.760.45 | 70.630.40 | 71.021.22 | 51.502.21 | 60.951.51 | 63.490.75 | 73.131.18 | 74.340.35 | 75.650.82 | |
Best | 69.76 | 72.14 | 73.29 | 51.50 | 60.95 | 63.49 | 73.13 | 75.55 | 75.65 | |
SAGE | Vanilla | 59.171.23 | 66.240.92 | 66.530.80 | 42.960.28 | 54.992.51 | 53.182.90 | 67.560.84 | 75.310.93 | 77.380.68 |
Reweight | 63.760.89 | 70.151.15 | 71.140.84 | 45.912.05 | 57.950.73 | 55.900.93 | 68.031.69 | 74.560.41 | 75.390.38 | |
ReNode | 65.321.07 | 71.311.29 | 71.540.85 | 48.552.31 | 56.320.40 | 56.491.73 | 69.082.04 | 74.240.20 | 75.280.69 | |
Resample | 57.771.35 | 71.241.08 | 73.011.02 | 39.371.40 | 61.411.11 | 61.931.40 | 69.221.28 | 74.911.09 | 75.800.39 | |
SMOTE | 58.811.97 | 70.311.35 | 73.022.29 | 38.421.69 | 64.140.75 | 66.350.70 | 64.961.56 | 74.590.96 | 77.310.45 | |
GSMOTE | 61.571.78 | 69.880.96 | 72.281.48 | 42.212.12 | 60.911.33 | 62.321.06 | 71.550.64 | 74.740.81 | 76.140.21 | |
GENS | 68.840.41 | 69.781.18 | 71.920.71 | 52.571.78 | 64.360.68 | 63.840.68 | 71.380.99 | 75.891.17 | 76.461.29 | |
Best | 68.84 | 71.31 | 73.02 | 52.57 | 64.36 | 66.35 | 71.55 | 75.89 | 77.38 | |
APPNP | Vanilla | 55.371.65 | 58.131.69 | 61.711.66 | 35.690.14 | 35.680.15 | 36.020.25 | 59.300.50 | 55.620.31 | 57.820.29 |
Reweight | 72.620.47 | 73.620.89 | 72.510.87 | 50.883.64 | 63.541.02 | 65.571.11 | 72.000.81 | 72.150.60 | 71.221.10 | |
ReNode | 73.741.12 | 75.021.54 | 72.150.76 | 50.503.51 | 63.730.54 | 65.130.40 | 72.761.37 | 71.540.96 | 71.880.70 | |
Resample | 65.781.72 | 73.140.94 | 73.570.92 | 40.791.87 | 66.540.49 | 59.514.16 | 67.741.94 | 72.250.81 | 74.410.95 | |
SMOTE | 65.341.68 | 73.181.02 | 72.880.90 | 40.792.05 | 66.620.33 | 58.824.59 | 67.242.10 | 72.671.65 | 73.331.37 | |
GSMOTE | 71.130.72 | 73.370.82 | 73.780.71 | 45.372.75 | 64.950.11 | 62.952.58 | 69.572.20 | 73.370.95 | 74.901.27 | |
Best | 73.74 | 75.02 | 73.78 | 50.88 | 66.62 | 65.57 | 72.76 | 73.37 | 74.90 | |
GPRGNN | Vanilla | 67.970.51 | 71.991.14 | 73.381.18 | 42.312.16 | 55.850.89 | 58.821.91 | 67.041.82 | 57.920.45 | 77.491.15 |
Reweight | 72.150.57 | 72.901.33 | 73.220.55 | 53.222.89 | 59.780.76 | 61.001.82 | 73.351.07 | 75.221.02 | 76.860.76 | |
ReNode | 73.380.67 | 73.710.57 | 73.931.60 | 54.662.82 | 59.690.73 | 60.341.31 | 73.560.98 | 75.691.10 | 76.250.67 | |
Resample | 67.001.33 | 72.941.02 | 74.890.86 | 42.272.15 | 64.160.62 | 63.890.98 | 70.421.51 | 73.790.83 | 75.310.54 | |
SMOTE | 66.991.33 | 74.011.51 | 74.411.05 | 40.972.02 | 63.880.55 | 62.601.72 | 70.291.47 | 73.890.69 | 75.481.02 | |
GSMOTE | 70.940.57 | 73.631.25 | 74.020.90 | 48.013.28 | 63.030.92 | 61.680.86 | 71.511.91 | 72.160.58 | 74.770.83 | |
Best | 73.38 | 74.01 | 74.89 | 54.66 | 64.16 | 63.89 | 73.56 | 75.69 | 77.49 |
Dataset (IR=10) | Cora | CiteSeer | PubMed | |||||||
---|---|---|---|---|---|---|---|---|---|---|
Metric: Macro-F1 | Base | + Bat0 | + Bat1 | Base | + Bat0 | + Bat1 | Base | + Bat0 | + Bat1 | |
GCN | Vanilla | 60.101.53 | 63.281.07 | 68.681.49 | 28.052.53 | 51.551.28 | 54.941.44 | 55.092.48 | 67.161.53 | 64.403.68 |
Reweight | 67.850.62 | 69.411.01 | 70.310.82 | 36.593.66 | 56.841.06 | 57.541.08 | 67.073.42 | 72.940.81 | 73.240.90 | |
ReNode | 66.661.59 | 69.790.79 | 70.591.25 | 34.641.54 | 56.690.64 | 58.070.77 | 67.863.99 | 72.610.41 | 72.250.89 | |
Resample | 57.342.27 | 71.360.39 | 72.821.13 | 29.732.77 | 57.170.48 | 58.031.42 | 56.743.54 | 71.190.83 | 73.131.33 | |
SMOTE | 55.651.62 | 71.040.16 | 72.820.86 | 29.392.81 | 59.530.88 | 61.531.24 | 56.143.74 | 71.720.60 | 72.831.20 | |
GSMOTE | 67.601.67 | 68.011.00 | 70.280.48 | 40.073.02 | 56.641.09 | 56.251.50 | 70.601.17 | 72.951.39 | 75.700.35 | |
GENS | 69.960.29 | 71.620.64 | 72.280.65 | 54.451.69 | 59.890.68 | 62.460.43 | 71.281.84 | 75.770.55 | 76.860.93 | |
Best | 69.96 | 71.62 | 72.82 | 54.45 | 59.89 | 62.46 | 71.28 | 75.77 | 76.86 | |
GAT | Vanilla | 60.711.61 | 64.270.95 | 68.930.79 | 31.123.15 | 54.711.18 | 59.421.55 | 57.321.55 | 71.271.11 | 74.031.08 |
Reweight | 66.491.34 | 69.840.91 | 69.790.77 | 34.944.09 | 58.530.68 | 60.281.12 | 63.821.60 | 75.131.13 | 73.881.38 | |
ReNode | 67.271.23 | 70.610.83 | 68.241.48 | 37.722.61 | 57.642.11 | 58.570.75 | 67.383.22 | 74.880.99 | 74.961.18 | |
Resample | 55.362.47 | 70.870.94 | 72.311.07 | 25.711.97 | 59.771.31 | 59.660.95 | 57.241.54 | 72.530.66 | 73.090.83 | |
SMOTE | 57.490.60 | 69.680.66 | 71.741.03 | 26.052.30 | 59.831.33 | 61.751.30 | 55.662.76 | 73.331.00 | 73.300.16 | |
GSMOTE | 64.341.69 | 68.231.80 | 69.771.08 | 35.071.77 | 55.861.10 | 57.132.69 | 63.352.92 | 74.230.84 | 73.342.06 | |
GENS | 69.960.62 | 69.830.41 | 70.711.16 | 48.342.19 | 60.041.85 | 62.550.86 | 71.781.19 | 72.690.84 | 74.421.12 | |
Best | 69.96 | 70.87 | 72.31 | 48.34 | 60.04 | 62.55 | 71.78 | 75.13 | 74.96 | |
SAGE | Vanilla | 57.361.77 | 64.900.87 | 65.610.97 | 36.071.06 | 54.762.47 | 51.863.25 | 63.751.24 | 74.350.74 | 76.920.63 |
Reweight | 63.721.10 | 69.060.90 | 69.590.53 | 39.642.57 | 57.170.76 | 54.830.74 | 62.832.57 | 73.880.40 | 75.420.48 | |
ReNode | 65.591.44 | 69.991.35 | 69.861.27 | 44.203.68 | 55.410.48 | 55.781.63 | 64.973.00 | 74.330.20 | 74.880.53 | |
Resample | 55.292.12 | 70.401.11 | 71.490.79 | 30.142.20 | 60.711.25 | 61.291.48 | 65.232.26 | 74.280.96 | 75.480.44 | |
SMOTE | 56.722.69 | 69.421.29 | 71.711.94 | 29.222.33 | 63.610.87 | 65.910.68 | 57.603.22 | 72.980.69 | 76.450.77 | |
GSMOTE | 59.442.25 | 69.100.95 | 71.301.47 | 34.863.46 | 60.531.27 | 61.961.12 | 67.230.61 | 74.361.02 | 75.680.31 | |
GENS | 68.230.72 | 69.760.95 | 71.110.81 | 51.052.03 | 63.870.82 | 63.410.57 | 70.060.86 | 75.331.46 | 76.011.14 | |
Best | 68.23 | 70.40 | 71.71 | 51.05 | 63.87 | 65.91 | 70.06 | 75.33 | 76.92 | |
APPNP | Vanilla | 50.392.81 | 54.192.58 | 59.992.49 | 22.210.13 | 22.540.25 | 22.890.22 | 44.500.21 | 44.670.07 | 44.590.06 |
Reweight | 72.630.53 | 72.710.60 | 70.610.65 | 45.254.85 | 63.081.03 | 65.201.20 | 69.531.14 | 72.240.58 | 72.260.80 | |
ReNode | 73.670.98 | 73.671.18 | 69.790.72 | 44.914.99 | 62.970.78 | 64.470.40 | 70.651.66 | 72.330.90 | 72.180.55 | |
Resample | 65.202.08 | 72.250.82 | 72.720.97 | 31.042.76 | 66.060.54 | 54.576.08 | 62.423.62 | 72.320.93 | 74.271.08 | |
SMOTE | 64.702.06 | 72.900.83 | 72.310.94 | 30.902.86 | 66.180.37 | 53.906.26 | 61.833.65 | 72.551.61 | 73.871.37 | |
GSMOTE | 71.200.67 | 73.020.74 | 73.220.92 | 37.904.29 | 64.560.18 | 60.413.84 | 65.653.06 | 72.540.85 | 74.611.36 | |
Best | 73.67 | 73.67 | 73.22 | 45.25 | 66.18 | 65.20 | 70.65 | 72.55 | 74.61 | |
GPRGNN | Vanilla | 67.860.79 | 70.801.16 | 72.321.18 | 35.002.96 | 55.060.89 | 56.312.87 | 59.013.62 | 50.121.46 | 77.621.04 |
Reweight | 71.660.85 | 70.460.98 | 71.240.49 | 49.193.61 | 59.110.73 | 60.302.04 | 71.180.95 | 75.470.90 | 77.010.52 | |
ReNode | 73.080.66 | 71.520.50 | 71.721.51 | 50.343.18 | 59.100.75 | 58.941.36 | 71.451.19 | 75.081.06 | 75.760.84 | |
Resample | 66.421.65 | 71.700.86 | 73.540.83 | 32.602.71 | 63.590.65 | 63.121.06 | 66.582.08 | 73.660.86 | 75.420.35 | |
SMOTE | 66.431.74 | 72.891.23 | 73.471.12 | 31.382.70 | 63.410.55 | 61.232.59 | 66.781.97 | 73.980.70 | 75.630.91 | |
GSMOTE | 70.870.53 | 72.530.85 | 73.120.95 | 42.824.52 | 62.091.04 | 60.820.88 | 67.933.01 | 72.720.72 | 74.660.74 | |
Best | 73.08 | 72.89 | 73.54 | 50.34 | 63.59 | 63.12 | 71.45 | 75.47 | 77.62 |
Dataset (IR=10) | Cora | CiteSeer | PubMed | |||||||
---|---|---|---|---|---|---|---|---|---|---|
Metric: PerfStd | Base | + Bat0 | + Bat1 | Base | + Bat0 | + Bat1 | Base | + Bat0 | + Bat1 | |
GCN | Vanilla | 27.881.79 | 21.271.76 | 18.492.68 | 29.931.38 | 13.822.06 | 13.930.80 | 34.732.14 | 9.232.78 | 21.814.51 |
Reweight | 22.291.41 | 14.432.51 | 18.322.20 | 25.471.78 | 19.101.48 | 22.640.79 | 19.335.26 | 10.211.58 | 5.880.83 | |
ReNode | 22.881.64 | 14.652.07 | 17.002.13 | 30.311.51 | 20.220.88 | 22.991.09 | 18.145.79 | 12.991.57 | 10.961.92 | |
Resample | 31.571.85 | 15.132.14 | 15.252.79 | 31.001.32 | 16.301.89 | 20.790.43 | 30.905.67 | 11.633.20 | 7.820.80 | |
SMOTE | 33.321.38 | 16.331.12 | 17.952.50 | 32.611.45 | 17.270.86 | 18.250.89 | 31.795.21 | 10.561.82 | 11.662.58 | |
GSMOTE | 21.781.79 | 17.902.75 | 18.442.20 | 22.642.69 | 21.371.25 | 21.011.45 | 15.872.34 | 3.351.08 | 5.831.27 | |
GENS | 20.041.12 | 16.983.02 | 18.022.23 | 16.952.64 | 14.940.75 | 15.540.60 | 11.933.46 | 5.951.85 | 5.150.80 | |
Best | 20.04 | 14.43 | 15.25 | 16.95 | 13.82 | 13.93 | 11.93 | 3.35 | 5.15 | |
GAT | Vanilla | 27.381.71 | 19.230.80 | 17.972.65 | 28.322.07 | 15.620.77 | 15.900.95 | 30.941.27 | 10.772.04 | 8.512.48 |
Reweight | 22.901.67 | 16.442.71 | 17.322.83 | 30.271.26 | 18.641.30 | 20.831.06 | 24.921.88 | 3.010.96 | 5.441.33 | |
ReNode | 23.131.54 | 15.051.59 | 18.961.65 | 25.211.85 | 20.480.82 | 20.510.49 | 18.154.37 | 4.771.22 | 6.170.42 | |
Resample | 32.732.12 | 17.872.04 | 17.642.57 | 32.590.89 | 17.761.79 | 18.731.02 | 31.670.98 | 6.181.35 | 4.581.14 | |
SMOTE | 31.170.69 | 18.401.03 | 18.261.87 | 33.320.88 | 10.680.68 | 13.241.12 | 32.792.13 | 8.142.00 | 7.561.05 | |
GSMOTE | 24.841.60 | 15.482.08 | 18.232.00 | 26.741.53 | 18.341.64 | 19.760.42 | 24.502.78 | 5.121.38 | 8.542.03 | |
GENS | 20.081.56 | 17.752.40 | 17.882.50 | 26.491.18 | 12.890.77 | 15.090.95 | 10.292.75 | 7.832.27 | 7.552.38 | |
Best | 20.08 | 15.05 | 17.32 | 25.21 | 10.68 | 13.24 | 10.29 | 3.01 | 4.58 | |
SAGE | Vanilla | 29.941.75 | 18.622.13 | 19.491.67 | 26.751.58 | 14.561.16 | 18.131.32 | 21.093.43 | 10.961.99 | 4.091.17 |
Reweight | 25.611.60 | 15.242.66 | 17.542.45 | 29.951.83 | 19.051.60 | 22.940.49 | 25.473.49 | 3.350.72 | 8.090.19 | |
ReNode | 24.121.73 | 13.323.03 | 15.452.41 | 22.414.31 | 22.200.97 | 22.750.87 | 22.924.36 | 7.631.23 | 5.771.55 | |
Resample | 31.661.47 | 15.772.75 | 15.082.74 | 30.291.16 | 18.720.90 | 18.482.00 | 21.412.88 | 4.681.42 | 4.761.09 | |
SMOTE | 30.862.64 | 17.302.09 | 14.873.23 | 32.071.00 | 13.171.33 | 12.780.43 | 31.622.86 | 13.881.44 | 11.632.28 | |
GSMOTE | 27.711.86 | 17.282.25 | 16.102.94 | 28.772.61 | 18.690.76 | 18.051.21 | 20.100.90 | 5.371.21 | 4.641.61 | |
GENS | 19.811.65 | 17.502.05 | 17.632.11 | 19.762.07 | 15.990.81 | 16.990.85 | 11.762.91 | 7.631.51 | 8.311.64 | |
Best | 19.81 | 13.32 | 14.87 | 19.76 | 13.17 | 12.78 | 11.76 | 3.35 | 4.09 | |
APPNP | Vanilla | 38.321.94 | 35.502.10 | 32.181.68 | 36.820.10 | 36.670.25 | 36.830.36 | 42.130.27 | 40.450.11 | 41.340.16 |
Reweight | 19.831.46 | 17.332.88 | 18.462.42 | 26.192.93 | 20.960.58 | 19.380.72 | 16.962.84 | 8.041.94 | 9.131.05 | |
ReNode | 18.092.52 | 16.872.95 | 19.472.00 | 25.953.66 | 22.091.43 | 20.421.57 | 14.493.81 | 10.251.93 | 3.951.02 | |
Resample | 27.282.13 | 18.372.34 | 18.722.32 | 32.711.23 | 15.871.02 | 23.724.14 | 25.864.38 | 13.601.68 | 9.491.65 | |
SMOTE | 27.861.78 | 18.612.35 | 19.421.81 | 33.261.10 | 14.910.93 | 22.904.49 | 26.374.47 | 13.371.97 | 8.702.29 | |
GSMOTE | 20.981.45 | 18.192.59 | 18.552.28 | 29.392.20 | 16.491.12 | 19.193.44 | 22.324.21 | 11.533.00 | 10.692.27 | |
Best | 18.09 | 16.87 | 18.46 | 25.95 | 14.91 | 19.19 | 14.49 | 8.04 | 3.95 | |
GPRGNN | Vanilla | 22.961.20 | 18.122.29 | 17.002.98 | 27.571.32 | 17.101.17 | 20.942.58 | 29.943.68 | 36.571.46 | 5.300.91 |
Reweight | 20.941.21 | 17.832.82 | 19.671.81 | 22.432.39 | 21.521.06 | 20.031.81 | 16.121.84 | 7.540.49 | 5.481.49 | |
ReNode | 18.842.19 | 16.782.53 | 17.892.96 | 24.141.47 | 19.841.79 | 22.831.40 | 14.403.18 | 9.752.20 | 6.611.47 | |
Resample | 25.621.80 | 19.232.26 | 17.612.77 | 33.080.66 | 17.040.78 | 15.980.93 | 22.592.75 | 7.622.50 | 7.760.85 | |
SMOTE | 25.441.88 | 16.973.19 | 17.382.78 | 32.850.95 | 15.091.23 | 16.852.51 | 21.352.76 | 9.412.67 | 6.090.62 | |
GSMOTE | 21.231.48 | 18.022.62 | 19.062.28 | 24.213.06 | 14.830.95 | 19.111.73 | 20.083.77 | 5.991.49 | 8.270.75 | |
Best | 18.84 | 16.78 | 17.00 | 22.43 | 14.83 | 15.98 | 14.40 | 5.99 | 5.30 |
Dataset | Cora | CiteSeer | PubMed | CS | Physics | ||||||
---|---|---|---|---|---|---|---|---|---|---|---|
Step IR | 10 | 20 | 10 | 20 | 10 | 20 | 10 | 20 | 10 | 20 | |
BAcc. | Base | 61.6 | 52.7 | 37.6 | 34.2 | 64.2 | 60.8 | 75.4 | 65.3 | 80.1 | 67.7 |
+ Bat | 69.8+13.4% | 71.3+35.2% | 55.4+47.2% | 51.3+49.9% | 68.6+6.8% | 63.3+4.1% | 82.6+9.6% | 79.9+22.2% | 87.6+9.4% | 88.0+29.9% | |
BestIGL | 70.1+13.9% | 66.5+26.2% | 56.0+48.9% | 47.2+38.0% | 74.0+15.2% | 71.1+17.0% | 84.1+11.6% | 81.3+24.4% | 89.4+11.6% | 85.7+26.6% | |
+ Bat | 74.2+20.6% | 71.6+35.9% | 62.7+66.6% | 62.5+82.6% | 76.9+19.7% | 75.7+24.5% | 86.3+14.5% | 85.6+31.0% | 91.2+13.9% | 90.9+34.2% | |
Macro-F1 | Base | 60.1 | 47.0 | 28.1 | 21.9 | 55.1 | 46.4 | 72.7 | 59.2 | 80.7 | 64.7 |
+ Bat | 68.7+14.3% | 69.6+48.1% | 54.9+95.8% | 48.9+123.5% | 67.2+21.9% | 60.7+30.8% | 78.6+8.1% | 74.7+26.1% | 88.8+10.0% | 87.8+35.8% | |
BestIGL | 70.0+16.4% | 66.2+40.9% | 54.5+94.1% | 45.0+105.6% | 71.3+29.4% | 68.9+48.3% | 83.9+15.3% | 80.9+36.7% | 89.5+10.9% | 86.2+33.2% | |
+ Bat | 72.8+21.2% | 70.2+49.4% | 62.5+122.7% | 62.1+183.6% | 76.9+39.5% | 74.9+61.2% | 85.4+17.5% | 84.6+43.0% | 90.7+12.4% | 90.0+39.2% | |
PerfStd | Base | 27.9 | 39.0 | 29.9 | 35.1 | 34.7 | 41.5 | 21.2 | 32.1 | 22.2 | 36.0 |
+ Bat | 21.3-23.7% | 24.4-37.5% | 13.9-53.5% | 16.7-52.5% | 21.8-37.2% | 29.1-29.9% | 17.4-18.2% | 22.9-28.8% | 11.5-48.3% | 25.6-29.0% | |
BestIGL | 20.0-28.1% | 21.9-43.8% | 16.9-43.4% | 18.0-48.6% | 11.9-65.6% | 14.2-65.7% | 8.9-58.3% | 12.3-61.8% | 6.3-71.7% | 12.4-65.5% | |
+ Bat | 15.2-45.3% | 17.5-55.2% | 13.9-53.5% | 16.7-52.5% | 5.1-85.2% | 4.6-89.0% | 7.9-62.7% | 10.1-68.5% | 6.6-70.2% | 6.9-80.8% | |
Natural IR | 50 | 100 | 50 | 100 | 50 | 100 | 50 | 100 | 50 | 100 | |
BAcc. | Base | 58.1 | 61.8 | 44.9 | 44.7 | 52.0 | 51.1 | 73.8 | 71.4 | 76.0 | 77.7 |
+ Bat | 69.1+18.9% | 68.3+10.6% | 58.4+29.9% | 57.4+28.5% | 55.6+7.0% | 56.5+10.4% | 82.1+11.3% | 81.9+14.8% | 86.9+14.3% | 84.1+8.3% | |
BestIGL | 71.0+22.3% | 73.8+19.5% | 56.3+25.3% | 56.3+26.0% | 72.7+39.8% | 72.8+42.5% | 81.2+10.0% | 81.4+14.0% | 85.8+12.9% | 87.2+12.2% | |
+ Bat | 73.1+25.8% | 76.9+24.5% | 62.1+38.2% | 61.3+37.3% | 75.8+45.7% | 75.9+48.5% | 85.0+15.1% | 84.5+18.5% | 88.6+16.5% | 89.7+15.4% | |
Macro-F1 | Base | 58.7 | 61.4 | 37.5 | 36.2 | 47.3 | 45.1 | 75.3 | 73.2 | 78.0 | 79.8 |
+ Bat | 68.7+17.1% | 67.5+10.0% | 57.1+52.6% | 55.8+54.3% | 52.8+11.6% | 52.0+15.4% | 82.6+9.7% | 82.6+12.8% | 87.6+12.3% | 85.2+6.8% | |
BestIGL | 71.1+21.2% | 73.4+19.5% | 54.3+44.8% | 53.8+48.8% | 72.9+53.9% | 73.7+63.6% | 82.5+9.5% | 82.4+12.6% | 87.7+12.4% | 88.3+10.6% | |
+ Bat | 72.7+23.9% | 76.0+23.9% | 60.2+60.8% | 59.4+64.3% | 75.3+59.2% | 76.1+68.8% | 85.7+13.7% | 85.1+16.2% | 88.8+13.8% | 89.4+12.0% | |
PerfStd | Base | 28.8 | 31.0 | 38.7 | 39.8 | 36.2 | 38.2 | 26.3 | 28.2 | 23.8 | 21.0 |
+ Bat | 18.3-36.4% | 25.4-18.1% | 24.9-35.6% | 33.1-17.0% | 33.3-8.1% | 35.9-6.2% | 19.0-27.9% | 19.5-30.9% | 17.0-28.7% | 19.6-6.7% | |
BestIGL | 18.9-34.4% | 17.3-44.4% | 28.7-25.9% | 29.7-25.3% | 6.0-83.4% | 9.6-75.0% | 14.4-45.4% | 15.4-45.5% | 11.2-53.1% | 9.7-53.8% | |
+ Bat | 15.9-44.8% | 14.7-52.8% | 21.9-43.4% | 19.8-50.3% | 4.2-88.3% | 5.6-85.3% | 12.2-53.5% | 12.8-54.7% | 7.4-68.9% | 7.2-65.7% |