AIDC
← 返回全部动态
arXiv 机器学习AI 评分 68/10009月24日 12:00

基于电阻曲率的可扩展图神经网络子图采样方法

针对大规模图神经网络(GNN)训练中传统采样标准易忽略边几何角色的问题,本研究提出了一种由电阻曲率引导的高效子图采样框架。该方法依托面向大图的曲率近似算法 ERC-LG,结合 Johnson-Lindenstrauss 投影与多 GPU 批处理共轭梯度求解器,避免了昂贵的显式拉普拉斯伪逆计算和全局嵌入存储,能有效利用几何曲率指导节点与边的采样概率。注:原始摘要文本在实验部分存在截断。

推荐理由将图几何曲率与 GPU 快速求解相结合,为大规模图神经网络训练开辟了兼顾结构特征与效率的采样新路径。

原标题:Scalable Subgraph Sampling via Resistance Curvature

阅读 arXiv 机器学习 原文 ↗