公开(公告)号：CN113283426A

公开(公告)日：2021-08-20

申请号：CN202110478018.2

申请日：2021-04-30

Applicant: 南京大学

Inventor： 俞扬 ,  詹德川 ,  周志华 ,  张书瑜 ,  胡毅奇 ,  秦熔均 ,  袁雷

IPC: G06K9/32 ,  G06N3/04

Abstract: 本发明公开一种基于多目标神经网络搜索的嵌入式目标检测模型生成方法，选择嵌入式目标检测模型的基础检测结构并设定搜索空间；构建包含搜索空间中所有子网络的超网络；将训练好的超网络联合设定好的检测头在目标检测数据集上进行微调；在搜索前通过批量枚举搜索空间中的子网络，自动构建可推理的模型，并将模型部署到嵌入式设备上进行运行时间的测试，建立搜索空间中子网络结构编码与运行时延之间的速查表；使用强化学习的方法展开搜索；搜索过程使得网络结构朝着多个目标的方向优化；维护一个帕累托最优解集来获得符合多目标要求的目标检测模型；选择帕累托最优解集中的元素，将其对应的检测网络在嵌入式平台上进行编译部署。

公开(公告)号：CN113283426B

公开(公告)日：2024-07-26

申请号：CN202110478018.2

申请日：2021-04-30

Applicant: 南京大学

Inventor： 俞扬 ,  詹德川 ,  周志华 ,  张书瑜 ,  胡毅奇 ,  秦熔均 ,  袁雷

IPC: G06V10/25 ,  G06V10/82 ,  G06N3/0464

Abstract: 本发明公开一种基于多目标神经网络搜索的嵌入式目标检测模型生成方法，选择嵌入式目标检测模型的基础检测结构并设定搜索空间；构建包含搜索空间中所有子网络的超网络；将训练好的超网络联合设定好的检测头在目标检测数据集上进行微调；在搜索前通过批量枚举搜索空间中的子网络，自动构建可推理的模型，并将模型部署到嵌入式设备上进行运行时间的测试，建立搜索空间中子网络结构编码与运行时延之间的速查表；使用强化学习的方法展开搜索；搜索过程使得网络结构朝着多个目标的方向优化；维护一个帕累托最优解集来获得符合多目标要求的目标检测模型；选择帕累托最优解集中的元素，将其对应的检测网络在嵌入式平台上进行编译部署。