中文

MutualNet:基于网络宽度与分辨率互学习的自适应卷积网络

计算机视觉与模式识别 2020-03-25 v3

摘要

我们提出宽度-分辨率互学习方法(MutualNet),用于训练一个可在动态资源约束下执行的网络,从而在运行时实现自适应的精度-效率权衡。我们的方法使用不同的输入分辨率训练一组具有不同宽度的子网络,使每个子网络相互学习多尺度表征。在不同的计算约束下,该方法在 ImageNet top-1 精度上始终优于最先进的自适应网络 US-Net,并以 1.5% 的优势超越 EfficientNet 中最佳的复合缩放 MobileNet。我们的方法的优越性也在 COCO 目标检测与实例分割以及迁移学习中得到验证。令人惊讶的是,MutualNet 的训练策略还能提升单一网络的性能,其在效率(GPU 搜索时长:15000 对比 0)和精度(ImageNet:77.6% 对比 78.6%)上均大幅优于强大的 AutoAugmentation。代码见 \url{https://github.com/taoyang1122/MutualNet}。

关键词

引用

@article{arxiv.1909.12978,
  title  = {MutualNet: Adaptive ConvNet via Mutual Learning from Network Width and Resolution},
  author = {Taojiannan Yang and Sijie Zhu and Chen Chen and Shen Yan and Mi Zhang and Andrew Willis},
  journal= {arXiv preprint arXiv:1909.12978},
  year   = {2020}
}