论文标题
通过预测编码来实现强大的图表学习
Robust Graph Representation Learning via Predictive Coding
论文作者
论文摘要
预测编码是一个最初开发的消息框架,该框架最初是为了建模大脑中的信息处理,现在由于某些有趣的属性而导致的机器学习研究主题。这样的属性之一是生成模型由于其特殊的信用分配规则而自然能够学习稳健表示的能力,该规则使神经活动在更新突触权重之前可以融合到解决方案上。图形神经网络也是通信模型,最近在机器学习中的任务各种任务中显示出了出色的结果,从而在结构化数据上提供了跨学科的最先进的性能。但是,它们容易受到不可察觉的对抗性攻击,并且不适合分布概括。在这项工作中,我们通过建立具有与流行图神经网络体系结构相同结构但依赖预测编码的消息规则的模型来解决这一问题。通过一系列广泛的实验,我们表明所提出的模型(i)在电感和跨性任务中的性能方面与标准模型相媲美,(ii)更好地校准,以及(iii)对多种对抗性攻击的鲁棒性。
Predictive coding is a message-passing framework initially developed to model information processing in the brain, and now also topic of research in machine learning due to some interesting properties. One of such properties is the natural ability of generative models to learn robust representations thanks to their peculiar credit assignment rule, that allows neural activities to converge to a solution before updating the synaptic weights. Graph neural networks are also message-passing models, which have recently shown outstanding results in diverse types of tasks in machine learning, providing interdisciplinary state-of-the-art performance on structured data. However, they are vulnerable to imperceptible adversarial attacks, and unfit for out-of-distribution generalization. In this work, we address this by building models that have the same structure of popular graph neural network architectures, but rely on the message-passing rule of predictive coding. Through an extensive set of experiments, we show that the proposed models are (i) comparable to standard ones in terms of performance in both inductive and transductive tasks, (ii) better calibrated, and (iii) robust against multiple kinds of adversarial attacks.