ShanghaiTech University Knowledge Management System
SAS: A Simple, Accurate and Scalable Node Classification Algorithm | |
2021-04-19 | |
状态 | 已发表 |
摘要 | Graph neural networks have achieved state-of-the-art accuracy for graph node classification. However, GNNs are difficult to scale to large graphs, for example frequently encountering out-of-memory errors on even moderate size graphs. Recent works have sought to address this problem using a two-stage approach, which first aggregates data along graph edges, then trains a classifier without using additional graph information. These methods can run on much larger graphs and are orders of magnitude faster than GNNs, but achieve lower classification accuracy. We propose a novel two-stage algorithm based on a simple but effective observation: we should first train a classifier then aggregate, rather than the other way around. We show our algorithm is faster and can handle larger graphs than existing two-stage algorithms, while achieving comparable or higher accuracy than popular GNNs. We also present a theoretical basis to explain our algorithm's improved accuracy, by giving a synthetic nonlinear dataset in which performing aggregation before classification actually decreases accuracy compared to doing classification alone, while our classify then aggregate approach substantially improves accuracy compared to classification alone. |
关键词 | Graph Neural Networks Node Classification Scalable Graph Algorithms |
DOI | arXiv:2104.09120 |
相关网址 | 查看原文 |
出处 | Arxiv |
WOS记录号 | PPRN:11804149 |
WOS类目 | Computer Science, Artificial Intelligence |
文献类型 | 预印本 |
条目标识符 | https://kms.shanghaitech.edu.cn/handle/2MSLDSTB/348510 |
专题 | 信息科学与技术学院 信息科学与技术学院_PI研究组_范睿组 信息科学与技术学院_硕士生 |
作者单位 | ShanghaiTech Univ, Sch Informat Sci & Technol, Shanghai, Peoples R China |
推荐引用方式 GB/T 7714 | Wang, Ziyuan,Yang, Feiming,Fan, Rui. SAS: A Simple, Accurate and Scalable Node Classification Algorithm. 2021. |
条目包含的文件 | ||||||
文件名称/大小 | 文献类型 | 版本类型 | 开放类型 | 使用许可 |
修改评论
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。