解决跨尺度图分类难题,提升小样本图的识别准确率
GSpect: Spectral Filtering for Cross-Scale Graph Classification
- 用图小波网络聚合多尺度信息,生成更鲁棒的图表示
- 设计频谱池化层将不同规模图统一为相同尺寸,提升泛化能力
- 在真实数据集上最高提升3.33%准确率,适合生物网络分析
识别常见结构是网络系统设计与优化的基础。然而,现实中由图表示的真实结构往往具有不同规模,导致传统图分类方法精度较低,这类图称为跨尺度图。为克服此局限,本研究提出GSpect,一种用于跨尺度图分类任务的先进谱图滤波模型。相较于其他方法,我们采用图小波神经网络作为模型卷积层,以聚合多尺度信息生成图表示;设计频谱池化层,将节点聚合为单个节点,从而将跨尺度图缩减至相同尺寸。我们构建了跨尺度基准数据集MSG(Multi Scale Graphs)。实验表明,在公开数据集上,GSpect平均提升分类准确率1.62%,在PROTEINS数据集上最高提升3.33%;在MSG数据集上,平均提升15.55%。该方法填补了跨尺度图分类研究的空白,有望应用于脑疾病诊断中预测脑网络标签,以及通过学习其他系统中的分子结构开发新药。
原文摘要 · Abstract (English)
Identifying structures in common forms the basis for networked systems design and optimization. However, real structures represented by graphs are often of varying sizes, leading to the low accuracy of traditional graph classification methods. These graphs are called cross-scale graphs. To overcome this limitation, in this study, we propose GSpect, an advanced spectral graph filtering model for cross-scale graph classification tasks. Compared with other methods, we use graph wavelet neural networks for the convolution layer of the model, which aggregates multi-scale messages to generate graph representations. We design a spectral-pooling layer which aggregates nodes to one node to reduce the cross-scale graphs to the same size. We collect and construct the cross-scale benchmark data set, MSG (Multi Scale Graphs). Experiments reveal that, on open data sets, GSpect improves the performance of classification accuracy by 1.62% on average, and for a maximum of 3.33% on PROTEINS. On MSG, GSpect improves the performance of classification accuracy by 15.55% on average. GSpect fills the gap in cross-scale graph classification studies and has potential to provide assistance in application research like diagnosis of brain disease by predicting the brain network's label and developing new drugs with molecular structures learned from their counterparts in other systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。