Cosine similarity distance pruning algorithm Based on graph attention mechanism

In recent years, graph neural network has been widely used. Attention mechanism is introduced into the graph neural network to make it more applicable. Both GAT and AGNN prove that attention mechanism plays an important role in graph neural network. Attention mechanism algorithms such as gat and AGNN directly use a self-learning variable to do the point product after calculating the connection (or similarity calculation) of node and neighbor features (without further processing of the calculation results). Finally, we get an aggregation of neighbor information. A cosine similarity distance pruning algorithm based on graph attention mechanism (CDP-GA) is proposed to optimize the attention matrix of nodes and their adjacent nodes. By calculating the cosine similarity between node features and neighbor features (the feature here is obtained by linear transformation), the similarity of nodes is regarded as the distance between nodes (or the weight of edges). And we think that the aggregation degree of node information is inversely proportional to the distance between nodes (similar to the heat conduction formula). In the method, we prune the neighborhood of the node according to the cosine similarity to get the final attention coefficient matrix. In this way, the attention mechanism in the graph neural network is further refined, and the loss of aggregation neighbor information is reduced. In the experiments of three datasets, our model is compared with the experimental classification of GAT and AGNN and the experiment of correlation graph neural network algorithm. Finally, it is proved that the algorithm is better than three known datasets.