Looking for breakthrough ideas for innovation challenges? Try Patsnap Eureka!

Unbalanced node classification method based on minority class weighted graph neural network

A neural network and node balancing technology, applied in the field of machine learning, can solve problems such as the decline of classification effect and the convergence of node characteristics, so as to achieve the effect of reducing bias, accelerating convergence speed and improving classification performance

Pending Publication Date: 2022-03-01
SHANGHAI APPLIED TECHNOLOGIES COLLEGE
View PDF0 Cites 0 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

However, the current method tends to ignore the influence of minority data on classification results in the stage of aggregation feature information and edge prediction. At the same time, there is a problem of convergence of node characteristics of different types in the process of information aggregation, which makes the classification effect decline.

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • Unbalanced node classification method based on minority class weighted graph neural network
  • Unbalanced node classification method based on minority class weighted graph neural network
  • Unbalanced node classification method based on minority class weighted graph neural network

Examples

Experimental program
Comparison scheme
Effect test

Embodiment Construction

[0043]The technical solutions of the present invention will be clearly and completely described below in conjunction with the accompanying drawings. Apparently, the described embodiments are some of the embodiments of the present invention, but not all of them. Based on the embodiments of the present invention, all other embodiments obtained by persons of ordinary skill in the art without making creative efforts belong to the protection scope of the present invention.

[0044] Such as figure 1 As shown, this embodiment discloses a method for classifying unbalanced nodes based on a minority class weighted graph neural network, comprising the following steps:

[0045] Step 1: For the input graph structure data, calculate the node membership value based on the adjacency information to obtain the weighted feature information of the node in the embedding space;

[0046] Specifically, in step 1, the node membership is calculated using the adjacency information of the nodes and the ...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to View More

PUM

No PUM Login to View More

Abstract

The invention discloses an unbalanced node classification method based on a minority class weighted graph neural network, and the method comprises the steps: 1, calculating a node membership value based on adjacency information for input graph structure data, so as to obtain the weighted feature information of a node in an embedded space; 2, executing a data oversampling operation in the embedding space to generate a new node; 3, cost sensitive learning is introduced in the training process of the edge predictor, and the trained edge predictor is used for obtaining adjacency information of a new node; and 4, constructing balanced graph structure data in combination with the features of the new nodes and the existing nodes and the adjacency information, performing node classification by using a graph neural network, and using a Gumbel distribution function as an activation function. According to the method, the neighbor aggregation of minority class nodes is enhanced, the bias to majority class nodes in the edge generation process is reduced, the convergence speed of the graph neural network model is improved, and the unbalanced node classification effect based on the graph neural network is remarkably improved.

Description

technical field [0001] The invention relates to the field of machine learning, in particular to an unbalanced node classification method based on minority class weighted graph neural network. Background technique [0002] Graph neural network plays a very important role in the analysis and mining of graph structure data, and has been effectively used in node classification tasks such as social, text, image and bioengineering. The node classification of graphs usually adopts the form of semi-supervised classification. In the graph structure data, only the label information of some nodes is known, and the label information of the remaining nodes is unknown. At present, most graph neural network algorithms are aimed at balanced data. However, in practice, due to the influence of data sampling bias and other objective reasons, the distribution of labeled data between categories is often unbalanced. For example, in the classification of webpage topics, some topics are easy to co...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to View More

Application Information

Patent Timeline
no application Login to View More
IPC IPC(8): G06K9/62G06N3/04G06N3/08
CPCG06N3/08G06N3/047G06F18/2414G06F18/2451G06F18/2415
Inventor 王克凡安静马超
Owner SHANGHAI APPLIED TECHNOLOGIES COLLEGE
Who we serve
  • R&D Engineer
  • R&D Manager
  • IP Professional
Why Patsnap Eureka
  • Industry Leading Data Capabilities
  • Powerful AI technology
  • Patent DNA Extraction
Social media
Patsnap Eureka Blog
Learn More
PatSnap group products