跳到正文
OpenReview · Representation learning· Yunhe Li; Yaochen Hu; Yingxue Zhang·· 2021-01-01精选AI 评分72

基于对抗式变分贝叶斯的图表示学习

Graph Representation Learning via Adversarial Variational Bayes

AI 导读

基于摘要分析。本文研究图节点表示中的不确定性建模,提出概率生成模型 GraphAVB,在保留图连接模式的同时,以神经网络参数化的隐式分布表示节点,而非为每个节点学习单一向量或限定为高斯分布。 核心机制是将 Adversarial Variational Bayes(AVB)引入图表示学习。由于隐式分布通过神经采样器定义,作者引入辅助判别网络,估计目标函数中的对数概率比项,从而将目标最大化转化为双参与者博弈。相对于摘要提及的 Graph2Gauss,主要区别在于放宽节点表示的分布形式;更灵活的分布能否带来稳定收益,仍需结合正文中的对照与消融核对。 作者报告,在多个真实图数据集的链路预测任务上,GraphAVB 优于多种有竞争力的基线,并据此认为下游任务可以受益于所捕获的不确定性。摘要未提供数据集名称、划分方式、指标、具体数值及基线配置;实验协议、消融及统计信息尚未验证,代码与复现条件也尚未验证。 平台推测(待验证):若将 EEG 通道及其连接关系构建为图,隐式节点分布可能用于表征连接估计的不确定性;这是假设,不是本文已验证的 EEG 结果。可复用部分是节点分布建模与概率比估计机制,需改造部分包括 EEG 图构建及下游任务接口。低信噪比、跨被试通道差异和小样本可能使图结构不稳定,并增加对抗训练与不确定性估计的难度。可先在固定的被试内 EEG 数据划分与相同图构建条件下,对比单向量、高斯分布及隐式分布表示,检验其在受控通道噪声扰动下的链路预测表现及不确定性与错误的关联;该实验仅验证机制可迁移性,不代表 BCI 解码获益。已有 EEG 相关工作尚未检索确认。

阅读价值

GraphAVB 通过隐式分布与对抗式变分推断建模节点表示的不确定性,值得核对其相对高斯分布表示的收益及训练稳定性,但摘要中的链路预测结果尚不能证明其适用于 EEG/BCI。

来源:OpenReview · Representation learning · openreview.net