• 博客园logo
  • 会员
  • 周边
  • 新闻
  • 博问
  • 闪存
  • 众包
  • 赞助商
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录
grapefruit-xh
博客园    首页    新随笔    联系   管理    订阅  订阅
MLP-Mixer: An all-MLP Architecture for Vision 纯MLP架构

典型的MLP结构包括三层:input、hidden、output。不同层之间都是全联接的。

MLP-Mixer完全利用基础的矩阵乘法运算和数据变换以及非线性层来完成复杂数据集的分类任务。

Step 1:将图像转成token作为后续模型的输入(该过程与ViT一致)

     

MLP-Mixer可以靠channel-mixing MLPs层结合不同channels的信息,也可以靠token-mixing MLPs层结合不同空间位置的信息。

 

 MLP1是token-mixing MLPs,在输入的列上操作,即对输入进行转置,所有列共享参数,得到的输出再次转置;MLP2是channel-mixing MLPs作用在输入的行上,共享参数MLP2

Step 2: 经过Mixer layer之后进入分类头

补充:MLP-Mixer没有使用位置编码信息,是因为Mixer layer的两种MLP对输入顺序比较敏感。

posted on 2022-04-13 17:26  爱吃柚子的小肖  阅读(256)  评论(0)    收藏  举报
刷新页面返回顶部
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3