Aug, 2023

轻量级卷积神经网络的数据端效率

TL;DR我们研究了在图像分类和目标检测的两个重要视觉任务中,选择数据方面属性如何帮助选择或设计轻量级卷积神经网络。通过实验,我们展示了四个数据属性 - 类别数目、目标颜色、图像分辨率和目标尺度如何影响神经网络模型的大小和效率。通过基于度量学习的内部和外部类别相似度度量,来指导这些属性的评估从而实现轻量级模型。使用这些度量进行评估,其计算量比完全进行推理测试少 30 倍。我们提供了一个应用这些评估和方法来选择一个轻量级模型用于机器人路径规划应用的示例,其计算量减少了 66%,精度提高了 3.5%。