- 专利标题: 一种图神经网络压缩方法、装置、电子设备及存储介质
-
申请号: CN202211299256.8申请日: 2022-10-24
-
公开(公告)号: CN115357554B公开(公告)日: 2023-02-24
- 发明人: 胡克坤 , 董刚 , 赵雅倩 , 李仁刚
- 申请人: 浪潮电子信息产业股份有限公司
- 申请人地址: 山东省济南市高新区浪潮路1036号
- 专利权人: 浪潮电子信息产业股份有限公司
- 当前专利权人: 浪潮电子信息产业股份有限公司
- 当前专利权人地址: 山东省济南市高新区浪潮路1036号
- 代理机构: 北京集佳知识产权代理有限公司
- 代理商 任洁芳
- 主分类号: G06F16/174
- IPC分类号: G06F16/174 ; G06F9/50 ; G06N3/0464 ; G06N3/048 ; G06N3/092
摘要:
本发明提供一种图神经网络压缩方法、装置、电子设备及存储介质,涉及神经网络领域,方法包括:获取已训练的图神经网络及其训练时所使用的图数据;确定图数据中所有图顶点对应的度数分布范围,并将度数分布范围划分为多个度数区间;在预设资源限制条件的约束下,利用强化学习及硬件加速器确定各度数区间对应的最优区间量化位宽以及图神经网络对应的最优网络量化位宽;利用最优区间量化位宽对图数据中对应度数的图顶点的顶点特征进行量化压缩,并利用最优网络量化位宽对图神经网络进行量化压缩,得到最优量化图数据和最优量化图神经网络;利用强化学习为图神经网络和图顶点特征确定最优量化位宽,确保量化图神经网络具有高精度及较低资源消耗率。
公开/授权文献
- CN115357554A 一种图神经网络压缩方法、装置、电子设备及存储介质 公开/授权日:2022-11-18