arXiv 人工智能· Bohan Yoon, Hyejin Jang·· 1 天前AI 评分27
图依赖池化 GDPool:分层图神经网络中的自适应粒度
Graph-dependent pooling: adaptive granularity in hierarchical graph neural networks
AI 导读
GDPool 提出一种图依赖池化方法,能为每个输入图动态确定池化粒度,解决分层图神经网络依赖固定池化比例的限制。该方法利用自注意力机制挑选结构关键节点作为锚点,并通过软节点分配构建粗化图。在金属有机框架与分子图基准测试中,GDPool 均取得最佳预测性能,且具备更高的模块度与架构稳定性。
来源:arXiv 人工智能 · arxiv.org