- 专利标题: 基于神经架构搜索的轻量级多退出点模型的建立方法
-
申请号: CN202210423171.X申请日: 2022-04-21
-
公开(公告)号: CN114861906A公开(公告)日: 2022-08-05
- 发明人: 王晓飞 , 张子桐 , 王晓辉 , 王勇 , 郭鹏天 , 李琪 , 刘志成 , 赵云凤 , 仇超
- 申请人: 天津大学 , 国网山东省电力公司 , 中国电力科学研究院有限公司
- 申请人地址: 天津市南开区卫津路92号; ;
- 专利权人: 天津大学,国网山东省电力公司,中国电力科学研究院有限公司
- 当前专利权人: 天津大学,国网山东省电力公司,中国电力科学研究院有限公司
- 当前专利权人地址: 天津市南开区卫津路92号; ;
- 代理机构: 深圳众邦专利代理有限公司
- 代理商 丁曹凯
- 主分类号: G06N3/08
- IPC分类号: G06N3/08 ; G06N3/04 ; G06K9/62 ; G06V10/764 ; G06V10/82
摘要:
本发明公开了一种基于神经架构搜索的轻量级多退出点模型的建立方法,包括如下步骤:搜集电网图像数据构建样本数据集,将样本数据集划分为训练集和测试集;以树形结构为基础,设计包括主干网络和分支分类器的多退出点模型架构;通过将若干种卷积神经网络进行剪裁并重新组合的方法构建搜索空间,基于图像分类任务以权衡模型分类精度和模型大小为目标建立目标函数;将训练集输入搜索空间中,利用策略梯度的强化学习算法对目标函数进行优化,以多退出点模型架构为基础进行训练获取多退出点模型。本发明能很好地权衡了搜索得到的神经网络的精度和参数大小,实现对电力边缘侧的图像分类模型的压缩。
公开/授权文献
- CN114861906B 基于神经架构搜索的轻量级多退出点模型的建立方法 公开/授权日:2024-09-06