硬件感知的高效特征融合网络搜索

作者：郭家明; 张蕊; 支天; 何得园; 黄迪; 常明; 张曦珊; 郭崎

来源：计算机学报, 2022, 45(11): 2420-2432.

DOI：10.11897/SP.J.1016.2022.02420

• 目标检测
• 神经结构搜索
• 硬件开销

特征融合网络通过融合多尺度特征来提高目标检测精度,是深度学习目标检测框架中的关键部分.已有的研究工作通过优化融合网络的拓扑结构来提高结果精度,忽略了所需的硬件资源开销以及特征选择和特征融合操作对结果的影响.本文提出了支持多尺度特征融合的注意力感知融合网络(Attention-aware Fusion Network,AFN),通过软硬件协同可实现硬件开销(参数存储、计算时间等)敏感的神经网络自动搜索,从融合网络的特征、路径和操作三方面实现一体化的优化部署.实验结果表明,当主干网络为ResNet50时,在实现相似检测精度时,相比现有最先进的搜索网络NAS-FPN,本文方法的参数量和计算量分别减少29.6%和22.3%,相比现有人工设计网络FPN,本文方法的AP可以提高2.1%.当主干网络为VGG时,相比现有最先进的搜索网络Auto-FPN,本文方法的AP提高了1.7%.

• 单位
  中国科学院计算技术研究所; 中国科学院大学