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

摘要：

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

关键词：目标检测、神经结构搜索、硬件开销

所属期刊栏目：45

分类号：TP18(自动化基础理论)

资助基金：国家重点研发计划;国家自然科学基金;国家自然科学基金;国家自然科学基金;国家自然科学基金;国家自然科学基金;中国科学院战略性先导科技专项;中国科学院稳定支持基础研究领域青年团队计划;科学探索奖资助

在线出版日期：2022-12-09（万方平台首次上网日期，不代表论文的发表时间）

页数：共13页

页码：2420-2432