计算机科学 > 机器学习
[提交于 2019年5月11日
]
标题: 图U-Net
标题: Graph U-Nets
摘要: 我们研究了图数据的表征学习问题。卷积神经网络可以自然地处理图像,但在处理图数据时面临重大挑战。由于图像可以看作节点位于二维网格上的特殊图,图嵌入任务与像素级预测任务(如分割)有自然的对应关系。虽然像U-Net这样的编码器-解码器架构已在许多像素级预测任务中成功应用,但对于图数据来说,类似的方法还很缺乏。这是因为池化和上采样操作在图数据上并不自然。为了解决这些挑战,本研究提出了新颖的图池化(gPool)和图解池化(gUnpool)操作。gPool层根据节点在可训练投影向量上的标量投影值自适应选择一些节点以形成更小的图。我们进一步提出了gUnpool层作为gPool层的逆操作。gUnpool层利用相应gPool层中选择节点的位置信息恢复图的原始结构。基于提出的gPool和gUnpool层,我们在图上开发了一种编码器-解码器模型,称为图U-Net。我们的节点分类和图分类任务的实验结果表明,我们的方法比以前的模型表现得更好。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.