神经网络架构搜索的方法及图像分类的方法和装置

    公开(公告)号:CN113344186A

    公开(公告)日:2021-09-03

    申请号:CN202110641506.0

    申请日:2021-06-09

    Abstract: 本说明书公开了一种神经网络架构搜索的方法及图像分类的方法和装置,搜索模型中包含第一权重和第二权重,对于待生成神经网络架构中的一个架构单元而言,第一权重表示该架构单元与待生成神经网络架构中的每个其他架构单元之间的连接的重要性,第二权重表示采用搜索空间中的每个备选单元来实现该架构单元的重要性,在训练搜索模型时,根据第一权重选择与该架构单元相连的其他架构单元,根据第二权重选择用于实现该架构单元的备选单元,根据选择的结果生成神经网络架构,预测生成的神经网络架构的质量,根据预测结果对搜索模型进行训练,以调整第一权重和第二权重,通过训练完成的搜索模型生成神经网络架构,可有效提高生成的神经网络架构的质量。

Patent Agency Ranking