MobileNetV4 -- Universal Models for the Mobile Ecosystem

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Qin, Danfeng, Leichner, Chas, Delakis, Manolis, Fornoni, Marco, Luo, Shixin, Yang, Fan, Wang, Weijun, Banbury, Colby, Ye, Chengxi, Akin, Berkin, Aggarwal, Vaibhav, Zhu, Tenghui, Moro, Daniele, Howard, Andrew
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910623940476928
author Qin, Danfeng
Leichner, Chas
Delakis, Manolis
Fornoni, Marco
Luo, Shixin
Yang, Fan
Wang, Weijun
Banbury, Colby
Ye, Chengxi
Akin, Berkin
Aggarwal, Vaibhav
Zhu, Tenghui
Moro, Daniele
Howard, Andrew
author_facet Qin, Danfeng
Leichner, Chas
Delakis, Manolis
Fornoni, Marco
Luo, Shixin
Yang, Fan
Wang, Weijun
Banbury, Colby
Ye, Chengxi
Akin, Berkin
Aggarwal, Vaibhav
Zhu, Tenghui
Moro, Daniele
Howard, Andrew
contents We present the latest generation of MobileNets, known as MobileNetV4 (MNv4), featuring universally efficient architecture designs for mobile devices. At its core, we introduce the Universal Inverted Bottleneck (UIB) search block, a unified and flexible structure that merges Inverted Bottleneck (IB), ConvNext, Feed Forward Network (FFN), and a novel Extra Depthwise (ExtraDW) variant. Alongside UIB, we present Mobile MQA, an attention block tailored for mobile accelerators, delivering a significant 39% speedup. An optimized neural architecture search (NAS) recipe is also introduced which improves MNv4 search effectiveness. The integration of UIB, Mobile MQA and the refined NAS recipe results in a new suite of MNv4 models that are mostly Pareto optimal across mobile CPUs, DSPs, GPUs, as well as specialized accelerators like Apple Neural Engine and Google Pixel EdgeTPU - a characteristic not found in any other models tested. Finally, to further boost accuracy, we introduce a novel distillation technique. Enhanced by this technique, our MNv4-Hybrid-Large model delivers 87% ImageNet-1K accuracy, with a Pixel 8 EdgeTPU runtime of just 3.8ms.
format Preprint
id arxiv_https___arxiv_org_abs_2404_10518
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MobileNetV4 -- Universal Models for the Mobile Ecosystem
Qin, Danfeng
Leichner, Chas
Delakis, Manolis
Fornoni, Marco
Luo, Shixin
Yang, Fan
Wang, Weijun
Banbury, Colby
Ye, Chengxi
Akin, Berkin
Aggarwal, Vaibhav
Zhu, Tenghui
Moro, Daniele
Howard, Andrew
Computer Vision and Pattern Recognition
We present the latest generation of MobileNets, known as MobileNetV4 (MNv4), featuring universally efficient architecture designs for mobile devices. At its core, we introduce the Universal Inverted Bottleneck (UIB) search block, a unified and flexible structure that merges Inverted Bottleneck (IB), ConvNext, Feed Forward Network (FFN), and a novel Extra Depthwise (ExtraDW) variant. Alongside UIB, we present Mobile MQA, an attention block tailored for mobile accelerators, delivering a significant 39% speedup. An optimized neural architecture search (NAS) recipe is also introduced which improves MNv4 search effectiveness. The integration of UIB, Mobile MQA and the refined NAS recipe results in a new suite of MNv4 models that are mostly Pareto optimal across mobile CPUs, DSPs, GPUs, as well as specialized accelerators like Apple Neural Engine and Google Pixel EdgeTPU - a characteristic not found in any other models tested. Finally, to further boost accuracy, we introduce a novel distillation technique. Enhanced by this technique, our MNv4-Hybrid-Large model delivers 87% ImageNet-1K accuracy, with a Pixel 8 EdgeTPU runtime of just 3.8ms.
title MobileNetV4 -- Universal Models for the Mobile Ecosystem
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2404.10518