An IMage Is Worth 16 X 16 Words (VIT)

1 篇文章 0 订阅

An IMage Is Worth 16 X 16 Words (VIT):

1. 网络

1.1 emb层

  1. 每张图N个patch,将patch flatten成一维

  2. 将每个patch的维度用线性层升到D维,这层叫patch embeddings

  3. 开始加cls, 每个emb要加pos_emb
    在这里插入图片描述

E是linear pro

1.2 剩下部分

在这里插入图片描述

  1. MSA是multiheaded self- attention
  2. LN - layer-norm
  3. MLP激活函数是GELU
  4. 每层都加了残差的

1.3 work的原因

  1. CNN过于拘泥于局部邻居,而VIT的self-att层是全局的
  2. 除了pos emb, VIT没有任何关于位置的提示,这意味着训练出的可以获得更好的泛化性

1.4 CNN+VIT

先把原始图像过一遍CNN,然后将CNN的feature map当作原始图像输入到VIT中。

1.5 在更高维度微调

patch数量会更多,所以pos emb会失效,所以作者对pos emb进行了一个二维的插值(interpolation)

2.实验

mask patch对VIT有用

策略(patch维度):

50%替换:

  • 80%ask
  • 10%随机patch
    略(patch维度):

50%替换:

  • 80%ask
  • 10%随机patch
  • 10%不变
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值