提出500万参数级轻量模型EMOv2,性能超越同类模型。
EMOv2: Pushing 5M Vision Model Frontier
- 设计统一的MMBlock结构,融合卷积与注意力机制。
- EMOv2-5M达79.4 Top-1准确率,检测mAP达41.5。
- 适合移动端部署,4G/5G下下载无感知延迟。
本文致力于构建参数高效、轻量化的密集预测模型,推动500万参数量级轻量模型的新边界。针对轻量级CNN中的倒置残差块(IRB)缺乏注意力架构对应物的问题,从统一视角重新思考高效IRB与Transformer实用组件,将基于CNN的IRB拓展至注意力模型,并抽象出一种单残差元移动块(MMBlock)。遵循简洁有效的设计准则,推导出改进型倒置残差移动块(i2RMB),并构建无需复杂结构的分层高效模型(EMOv2)。在4G/5G带宽下考虑移动端下载延迟可忽略的前提下,研究了500万参数量级模型的性能上限。大量实验表明,EMOv2在视觉识别、密集预测及图像生成任务上均优于现有方法:EMOv2-1M/2M/5M分别取得72.3、75.8、79.4的Top-1准确率,显著超越同规模的卷积与注意力模型;配备RetinaNet的EMOv2-5M在目标检测任务中达到41.5 mAP,较前代EMO-5M提升+2.6。采用更鲁棒训练策略后,EMOv2-5M最终实现82.9 Top-1准确率,将500万参数模型性能推至新高度。代码已开源。
原文摘要 · Abstract (English)
This work focuses on developing parameter-efficient and lightweight models for dense predictions while trading off parameters, FLOPs, and performance. Our goal is to set up the new frontier of the 5M magnitude lightweight model on various downstream tasks. Inverted Residual Block (IRB) serves as the infrastructure for lightweight CNNs, but no counterparts have been recognized by attention-based design. Our work rethinks the lightweight infrastructure of efficient IRB and practical components in Transformer from a unified perspective, extending CNN-based IRB to attention-based models and abstracting a one-residual Meta Mobile Block (MMBlock) for lightweight model design. Following neat but effective design criterion, we deduce a modern Improved Inverted Residual Mobile Block (i2RMB) and improve a hierarchical Efficient MOdel (EMOv2) with no elaborate complex structures. Considering the imperceptible latency for mobile users when downloading models under 4G/5G bandwidth and ensuring model performance, we investigate the performance upper limit of lightweight models with a magnitude of 5M. Extensive experiments on various vision recognition, dense prediction, and image generation tasks demonstrate the superiority of our EMOv2 over state-of-the-art methods, e.g., EMOv2-1M/2M/5M achieve 72.3, 75.8, and 79.4 Top-1 that surpass equal-order CNN-/Attention-based models significantly. At the same time, EMOv2-5M equipped RetinaNet achieves 41.5 mAP for object detection tasks that surpasses the previous EMO-5M by +2.6. When employing the more robust training recipe, our EMOv2-5M eventually achieves 82.9 Top-1 accuracy, which elevates the performance of 5M magnitude models to a new level. Code is available at https://github.com/zhangzjn/EMOv2.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。