[重读经典论文] MobileViT

参考博客:MobileViT模型简介

参考视频:15.1 MobileViT网络讲解

MobileViT结构上是一个混合模型,CNN+Transformer,比主流的轻量化ViT网络和轻量化CNN网络参数量少,精度高,缺点是推理速度慢,下图可以看到,推理时间是MobileNetV2的8倍。

推理速度

网络结构:

网络结构2

其中MV2为MobileNetV2的逆残差结构:

a0a78fe10e3843768b8121c3b50503e8

MobileViT结构是主要的创新,Unfold过程,其实就是对局部特征进行分组形成序列,再输入Transformer模块进行self-attention操作:

fold

selfattention

如上图所示,就是把feature map分成若干n*n的patch,然后将每个patch中相同位置(图中相同颜色)的特征形成一个个序列,再进行self-attention操作。

最后再将输出的序列进行unfold,也就是将每个序列相同位置的特征恢复到相应的位置。

posted @   大师兄啊哈  阅读(442)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示