The invention discloses an unbalanced node classification method based on graph contrast learning, and belongs to the technical field of
artificial intelligence. According to the method, a graph comparison learning framework of adaptive balance data is provided, minority classes can be automatically identified, the
minority class performance is improved, and then the overall performance of the model is improved. Firstly, an
Encoder-
Decoder architecture is used for pre-training, and compared with a traditional pseudo tag generation method, an unbalance rate self-adaptive sampling strategy is designed, the unbalance rate of data is calculated according to pseudo tags, and the sampling strategy is selected in a self-adaptive mode. For a
data set with a low unbalance rate, a simple downsampling method is adopted, and the proportion of
minority class information is increased; for a
data set with a relatively high unbalance rate, a mixed sampling strategy is adopted, and over-sampling and down-sampling are combined, so that the
information loss of majority of nodes is reduced while the information proportion of minority of nodes is increased. In addition, the pre-training model used in the invention can provide more accurate
label information, thereby improving the distinguishing ability of the model in subsequent GCL training. Then, a new data augmentation technology is designed, in the node masking process, pseudo
label information is utilized, information of
minority class nodes is reserved preferentially, and meanwhile
majority class nodes are masked; the method is helpful for the model to better capture minority class features in an
unbalanced data set. And finally, a
linear classifier is used for classification. According to the method, the unbalanced node classification performance under the self-supervision condition can be effectively improved.