有效

启发式的神经网络架构搜索方法

王磊、张建、公璇、刘玉晓、王薇、张宝昌、王瑞生、白洋、张依漪
北京航空航天大学
王磊机构 暂无
技术领域 暂无
张建机构 暂无
技术领域 暂无
公璇机构 暂无
技术领域 暂无
刘玉晓机构 暂无
技术领域 暂无
王薇机构 暂无
技术领域 暂无
张宝昌机构 暂无
技术领域 暂无
王瑞生机构 暂无
技术领域 暂无
白洋机构 暂无
技术领域 暂无
张依漪机构 暂无
技术领域 暂无

摘要

本发明公开了启发式神经网络架构搜索方法及信息处理设备。所提供的启发式神经网络架构搜索方法,包括:获取用层次化表示技术描述的待搜索的神经网络模型的单元堆叠模型;对所述多个单元的每个单元的各边进行采样,得到待搜索的神经网络模型的实例;对所述神经网络模型的实例进行训练和验证,得到验证集正确率;更新各个单元的各边的所采样操作的操作性能;更新各个单元的各边所采样操作的置信上限;更新各个单元的各边的各操作被采样的概率;从各个单元的各边的操作空间中去除操作以缩小操作空间;以及将各个单元的各边的操作空间剩余的最后操作作为所选择的操作,并得到要搜索的神经网络架构。