【技术实现步骤摘要】
搜索方法、装置、设备和计算机可读存储介质
[0001]本申请涉及计算机
,尤其涉及一种搜索方法、装置、设备和计算机可读存储介质。
技术介绍
[0002]神经网络架构搜索技术(Neural Network Architecture Search,NAS)是指对于神经网络模型,通过算法自动化完成最优架构的设计,以替代算法工程师的手工设计。硬件感知的神经架构搜索技术(Hardware
‑
aware Neural Network Architecture Search,HW NAS)是指在NAS的基础上,加入了硬件感知的功能。在实际应用中,神经网络(Neural Network,NN)通常需要部署在特定的目标硬件上。例如,芯片需要在部署特定场景NN时,同时达到精确度、延时、功耗、存储等最优的状态。因此,在目标硬件上部署NN网络时,不仅需要考虑模型的性能表现,同时也要权衡模型在硬件上的延时、功耗等硬件指标,从而使产品达到最佳效果。
[0003]相关技术中,通过人工设置固定的硬件指标约束,利用进化算法(Evolution Algorithm,EA)在架构搜索中求取帕累托最优解(Pareto Optimal Solution)。
[0004]然而,在给定硬件指标约束c(例如,延迟)的架构搜索中,有可能出现某个搜索出来的模型x的性能Px,与另外一个模型y的性能Py一致:Px=Py,或者Px仅略大于Py,但是,硬件约束明显有c>Cx>Cy,使得搜索出来的模型x并非用户 ...
【技术保护点】
【技术特征摘要】
1.一种搜索方法,其特征在于,所述方法包括:在架构搜索空间对应的超网络中,选择满足固定约束的染色体,确定初始种群;所述超网络包括至少两个子网络,每个染色体对应一个子网络;基于所述初始种群和初始目标进化方向,确定父代种群,并对所述父代种群所包括的染色体进行变异和交叉,以确定目标种群;继续基于所述目标种群和下一目标进化方向,确定下一父代种群,并对所述下一父代种群所包括的下一染色体进行变异和交叉,以确定下一目标种群,直至达到迭代次数,实现多目标进化,得到最后一个目标种群;从所述最后一个目标种群中确定目标染色体,以实现多目标的架构搜索,得到所述目标染色体对应的目标子网络。2.根据权利要求1所述的方法,其特征在于,所述基于所述初始种群和初始目标进化方向,确定父代种群,包括:对所述初始种群中各个初始染色体的各个指标进行预测,得到所述各个初始染色体的至少两个指标信息;指标包括子网络的模型性能和硬件需求;根据所述各个初始染色体的至少两个指标信息中与所述初始目标进化方向对应的指标信息,对所述初始种群所包括的初始染色体进行排序,确定排序结果;所述初始目标进化方向为任意一个指标,或者,为至少两个具有相关性或冲突性的指标;根据所述排序结果,按照预设顺序选择第一预设数量的染色体,构建所述父代种群。3.根据权利要求1所述的方法,其特征在于,所述父代种群包括第一预设数量的染色体;所述对所述父代种群所包括的染色体进行变异和交叉,以确定目标种群,包括:基于所述第一预设数量的染色体,按照初始变异比例系数进行变异,确定第一数量的变异染色体;所述第一数量为所述初始变异比例系数与第二预设数量之积;基于所述第一预设数量的染色体,按照初始交叉比例系数进行交叉,确定第二数量的交叉染色体;所述初始交叉比例系数为1与所述初始变异比例系数之差;根据所述第一数量的变异染色体和所述第二数量的交叉染色体,确定所述目标种群;所述目标种群包括所述第二预设数量的进化染色体。4.根据权利要求1
‑
3任一项所述的方法,其特征在于,所述方法还包括:根据当前父代种群中各个当前染色体在当前目标进化方向上的指标信息,确定均值和方差;在所述均值小于预设均值、且所述方差小于预设方差的情况下,将初始变异比例系数调整为第一系数;所述第一系数大于所述初始变异比例系数;在所述均值大于所述预设均值、且所述方差小于所述预设方差的情况下,将初始交叉比例系数调整为第二系数;所述第二系数大于所述初始交叉比例系数;在所述均值小于所述预设均值、且所述方差大于所述预设方差的情况下,保留所述初始交叉比例系数和所述初始变异比例系数;在所述均值大于所述预设均值、且所述方差大于所述预设方差的情况下,保留所述初始交叉比例系数和所述初始变异比例系数。5.根据权利要求4所述的方法,其特征在于,所述方法还包括:
根据上一父代种群中各个上一染色体在所述当前目标进化方向上的指标信息,确定所述预设均值和所述预设方差。6.根据权利要求4所述的方法,其特征在于,所述方法还包括:根据所述超网络中最大网络或最小网络在所述当前目标进化方向上的指标信息,确定所述预设均值。7.根据权利要求1
‑
3任一项所述的方法,其特征在于,所述继续基于所述目标种群和下一目标进化方向,确定下一父代种群,包括:继续对所述目标种群中各个进化染色体的各个指标进行预测,得到所述各个进化染色体的至少两个指标信息;根据所述各个进化染色体的至少两个指标信息中与下一目标进化方向对应的指标信息,对所述目标种群所包括...
【专利技术属性】
技术研发人员:冯乾泰,赵娟萍,
申请(专利权)人:哲库科技上海有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。