【阅读笔记】Fuzzy Graph Neural Network for Few-Shot Learning

核心思想

以往的基于图神经网络的小样本学习方法在构建邻接矩阵时,要么通过神经网络计算、要么通过欧式距离计算。然而,这种做法有一个弊端,就是节点之间都有边相连,在图传播的过程中会造成噪声的传播。因此,本文通过对邻接矩阵进行阈值处理,降低噪声的传播。

方法

如图 1 所示,模型由三部分组成:预训练的特征提取器、关系编码器和 GNN 分类器。关系编码器和分类器是 GNN。首先,在大规模数据上用一个临时分类器训练特征提取器。在元学习阶段,设计了一个隶属度函数来生成隶属度(对邻接矩阵进行阈值处理),它为 GNN 分类器的图形提供了强大的归纳偏差

1、特征提取

采用预训练的 ResNet-12 来增强特征提取器的能力,使用优化特征提取器θ和分类器𝜃′如下:

在学习了特征提取器 Θ 之后,它仍然会在元学习阶段以较小的学习率进行优化。

2、关系编码器

在边缘预测中,从特征提取器中提取的特征是节点的初始特征。在第 l 层关系编码器中,边缘特征首先由边缘更新函数更新,输入是边缘两端节点的特征:

然后节点特征由节点更新函数更新,

关系编码器 φ 将在任务 T 结束时通过损失 𝐿′ 进行优化,

3、隶属度函数

4、GNN分类器

通过隶属函数对 E 进行变换后,将成为 GNN 分类器 θ 的图的邻接矩阵 A。与之前的工作不同,这个相邻矩阵是稀疏的,并且为图提供了关系归纳偏差

在使用以下经验损失优化整个模型之前,为每个任务 T 优化 GNN 分类器 θ:

将 Tq 定义为从 T 中采样的查询集 Q,并将 Yq 定义为查询集 Q 的标签。然后,考虑在节点 * 处为所有参数评估的交叉熵损失:

5、算法流程

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值