网络架构搜索方法、图像分类方法、装置和电子设备

    公开(公告)号:CN116796797A

    公开(公告)日:2023-09-22

    申请号:CN202310707540.2

    申请日:2023-06-14

    Abstract: 本发明涉及人工智能领域,提供一种网络架构搜索方法、图像分类方法、装置和电子设备,其中搜索方法包括:获取初始网络结构,并基于所述初始网络结构的各结构参数,确定搜索空间;基于所述搜索空间搭建超网络,并基于第一样本图像对所述超网络进行预训练,得到预训练超网络的预训练权重;基于当前演化代数下候选突变操作的自适应概率分布,采用演化算法对当前演化代数下的网络结构进行搜索优化,得到下一演化代数下的网络结构,直至停止演化,得到目标网络结构。本发明提供的网络架构搜索方法、图像分类方法、装置和电子设备,提高了搜索效率,可以稳定且高效地实现对高性能的小规模ViT网络的优化。

Patent Agency Ranking