ICLR 2020 | 超越传统,基于图神经网络的归纳矩阵补全
这是一篇ICLR2020入选论文《INDUCTIVE MATRIX COMPLETION BASED ON GRAPH NEURAL NETWORKS》(基于图神经网络的归纳矩阵补全)。文章来自华盛顿大学圣路易斯分校博士、Facebook AI 研究院研究科学家张牧涵。该论文提出一种新的Inductive Graph-based Matrix Completion (IGMC) 模型,在保持归纳推理(inductive reasoning)的同时,完全不借助任何内容信息。能不借助内容信息达成归纳推理的秘诀就在于子图结构。IGMC为每一个(user, item) pair提取一个包含子图(enclosing subgraph),并用图神经网络(graph neural network)训练一个由子图结构映射到用户对商品评分(rating)的回归模型。IGMC在多个数据集上取得了最先进的性能;它不仅能够适用于没在训练集中出现的用户和商品,更可以迁移(transfer)到新数据上。研究人员使用一个在MovieLens上训练的IGMC模型去预测豆瓣电影评分,取得了非常好的性能,甚至好于许多专门在豆瓣数据上训练的模型。