基于多粒度互信息最大化的图分类方法及装置

摘要

本发明公开了一种基于多粒度互信息最大化的图分类方法及装置，通过对训练集中的图数据进行节点表征；将经过节点表征后的图数据输入基于多粒度互信息最大化训练层次化无监督图池化模型；将训练过程中损失函数l oss值最低的一轮epoch下的层次化无监督图池化模型作为训练好的层次化无监督图池化模型；待测试的数据集中的图数据输入训练好的层次化无监督图池化模型得到数据集中各图数据的图级表征；将图级表征输入到用于分类的网络模型中进行图分类。由于多粒度互信息最大化使得本发明可以有效提取图中子结构，并充分保留其结构及特征信息，从而使获得的图级表征充分保留了图结构及特征信息，使得图分类精度更高。