-
公开(公告)号:CN116796797A
公开(公告)日:2023-09-22
申请号:CN202310707540.2
申请日:2023-06-14
Applicant: 中国科学院自动化研究所
IPC: G06N3/045 , G06N3/086 , G06V10/764 , G06V10/774 , G06V10/82
Abstract: 本发明涉及人工智能领域,提供一种网络架构搜索方法、图像分类方法、装置和电子设备,其中搜索方法包括:获取初始网络结构,并基于所述初始网络结构的各结构参数,确定搜索空间;基于所述搜索空间搭建超网络,并基于第一样本图像对所述超网络进行预训练,得到预训练超网络的预训练权重;基于当前演化代数下候选突变操作的自适应概率分布,采用演化算法对当前演化代数下的网络结构进行搜索优化,得到下一演化代数下的网络结构,直至停止演化,得到目标网络结构。本发明提供的网络架构搜索方法、图像分类方法、装置和电子设备,提高了搜索效率,可以稳定且高效地实现对高性能的小规模ViT网络的优化。