中文

任意宽度网络

计算机视觉与模式识别 2020-12-08 v1 机器学习

摘要

尽管卷积神经网络(CNN)在速度与精度上取得了显著改进,其在推理时仍通常作为整体实体运行。这对资源受限的实际应用构成了挑战,因为计算预算与性能需求会随情境变化。为应对这些约束,我们提出了任意宽度网络(AWN),一种可调宽度的CNN架构及相应的训练流程,可在推理时对速度与精度进行细粒度控制。我们的关键创新是使用下三角权重矩阵,其显式地处理了宽度变化的批统计量,并且天然适用于多宽度操作。我们还展示了该设计有助于基于随机宽度采样的高效训练流程。我们通过实验证明,所提出的AWN在提供最大粒度推理控制的同时,优于现有方法。

关键词

引用

@article{arxiv.2012.03153,
  title  = {Any-Width Networks},
  author = {Thanh Vu and Marc Eder and True Price and Jan-Michael Frahm},
  journal= {arXiv preprint arXiv:2012.03153},
  year   = {2020}
}

备注

8 pages. Published at CVPR 2020 Workshop on Efficient Deep Learning in Computer Vision. Code at https://github.com/thanhmvu/awn