arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.10995cs.CVcs.AI

RapidNet:基于多级膨胀卷积的移动端骨干网络

RapidNet: Multi-Level Dilated Convolution Based Mobile Backbone

The University of Texas at Austin(德克萨斯大学奥斯汀分校)

Mustafa Munir, Md Mostafijur Rahman, Radu Marculescu

AI总结:

提出基于多级膨胀卷积的纯CNN移动端骨干网络RapidNet,通过扩大感受野和长短程特征交互,在图像分类、检测、分割等任务上以更高速度和精度超越SOTA混合模型及ViT。

摘要:

近年来,视觉变换器(ViTs)主导了计算机视觉领域。然而,ViTs计算开销大,不适合移动设备;这导致了卷积神经网络(CNN)和基于ViT的混合模型在移动视觉应用中的普及。最近,视觉图神经网络(ViG)和CNN混合模型也被提出用于移动视觉任务。然而,所有这些方法在速度上仍比纯CNN模型慢。在这项工作中,我们提出多级膨胀卷积来设计一个纯CNN的移动端骨干网络。使用多级膨胀卷积可以获得比标准卷积更大的理论感受野。不同级别的膨胀还允许图像中短程和长程特征之间的交互。实验表明,我们提出的模型在图像分类、目标检测、实例分割和语义分割上,在准确性和/或速度方面优于最先进的(SOTA)移动CNN、ViT、ViG和混合架构。我们最快的模型RapidNet-Ti在ImageNet-1K上达到76.3%的top-1准确率,在iPhone 13 mini NPU上推理延迟为0.9毫秒,比MobileNetV2x1.4(74.7% top-1准确率,1.0毫秒延迟)更快且更准确。我们的工作表明,纯CNN架构在正确设计时,可以在准确性和速度上击败SOTA混合模型和ViT模型。

英文摘要:

Vision transformers (ViTs) have dominated computer vision in recent years. However, ViTs are computationally expensive and not well suited for mobile devices; this led to the prevalence of convolutional neural network (CNN) and ViT-based hybrid models for mobile vision applications. Recently, Vision GNN (ViG) and CNN hybrid models have also been proposed for mobile vision tasks. However, all of these methods remain slower compared to pure CNN-based models. In this work, we propose Multi-Level Dilated Convolutions to devise a purely CNN-based mobile backbone. Using Multi-Level Dilated Convolutions allows for a larger theoretical receptive field than standard convolutions. Different levels of dilation also allow for interactions between the short-range and long-range features in an image. Experiments show that our proposed model outperforms state-of-the-art (SOTA) mobile CNN, ViT, ViG, and hybrid architectures in terms of accuracy and/or speed on image classification, object detection, instance segmentation, and semantic segmentation. Our fastest model, RapidNet-Ti, achieves 76.3\% top-1 accuracy on ImageNet-1K with 0.9 ms inference latency on an iPhone 13 mini NPU, which is faster and more accurate than MobileNetV2x1.4 (74.7\% top-1 with 1.0 ms latency). Our work shows that pure CNN architectures can beat SOTA hybrid and ViT models in terms of accuracy and speed when designed properly.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑