跳到正文
原文
arXiv cs.AI· Bohan Yoon, Hyejin Jang·· 4 小时前AI 评分34

GDPool:层次图神经网络中的图依赖池化与自适应粒度

Graph-dependent pooling: adaptive granularity in hierarchical graph neural networks

AI 导读

GDPool 为每个图动态确定层次池化粒度,并用自注意力选择图特定 anchors、通过软节点分配构建粗图。在金属有机框架和分子图基准上,GDPool 均取得最佳预测性能。其 modularity 持续高于现有方法,活动簇数随深度单调下降,敏感性分析显示跨超参数与架构选择表现稳定。

来源:arXiv cs.AI · arxiv.org