中文

MnasFPN:面向移动设备目标检测的延迟感知金字塔架构学习

计算机视觉与模式识别 2020-08-03 v2

摘要

尽管在资源受限环境下视觉任务的架构搜索取得了蓬勃发展,但设备上目标检测架构的设计大多仍是手工的。少数自动化搜索工作要么围绕非移动友好的搜索空间展开,要么未受设备端延迟引导。我们提出 MnasFPN,一种用于检测头的移动友好搜索空间,并将其与延迟感知架构搜索结合以生成高效的目标检测模型。所学习的 MnasFPN 检测头与 MobileNetV2 主体配对时,在 Pixel 上以相似延迟优于 MobileNetV3+SSDLite 达 1.8 mAP。它也比 NAS-FPNLite 准确率高 1.0 mAP 且快 10%。消融研究表明,性能提升的大部分来自搜索空间中的创新。进一步探索揭示了搜索空间设计与搜索算法之间有趣的耦合关系,并且 MnasFPN 搜索空间的复杂度可能处于一个局部最优。

关键词

引用

@article{arxiv.1912.01106,
  title  = {MnasFPN: Learning Latency-aware Pyramid Architecture for Object Detection on Mobile Devices},
  author = {Bo Chen and Golnaz Ghiasi and Hanxiao Liu and Tsung-Yi Lin and Dmitry Kalenichenko and Hartwig Adams and Quoc V. Le},
  journal= {arXiv preprint arXiv:1912.01106},
  year   = {2020}
}

备注

10 pages, 7 figures