一种模型训练方法
实质审查的生效
摘要

本申请提供一种模型训练方法,包括定义一个新的量化线性层;将量化线性层的多维输入张量中全部元素量化为PINT格式,对量化线性层的待计算张量中全部元素量化为PINT数据格式,将量化后的多维输入张量和待计算张量进行矩阵乘计算得到定点结果;将定点结果反量化为浮点数并传播到后续网络层;将模型中原有的线性层替换为量化线性层,以及基于浮点数和PINT数据格式对模型进行训练。本申请开发了一种基于PINT数据格式的量化线性层,将低比特、高表示能力的PINT数据格式应用到模型训练,用量化线性层替换模型中使用到的线性层,使得在训练出的模型准确度变化很小的情况下有效的降低了数据计算和存储等方面的需求。

基本信息
专利标题 :
一种模型训练方法
专利标题(英):
暂无
公开(公告)号 :
CN114418088A
申请号 :
CN202111628710.5
公开(公告)日 :
2022-04-29
申请日 :
2021-12-28
授权号 :
暂无
授权日 :
暂无
发明人 :
王中风邵海阔鲁金铭魏敬和
申请人 :
南京大学
申请人地址 :
江苏省南京市栖霞区仙林大道163号
代理机构 :
北京弘权知识产权代理有限公司
代理人 :
逯长明
优先权 :
CN202111628710.5
主分类号 :
G06N3/08
IPC分类号 :
G06N3/08  G06N3/04  G06K9/62  G06F16/332  G06F40/295  G06F16/35  
IPC结构图谱
G
G部——物理
G06
计算;推算或计数
G06N
基于特定计算模型的计算机系统
G06N3/00
基于生物学模型的计算机系统
G06N3/02
采用神经网络模型
G06N3/08
学习方法
法律状态
2022-05-20 :
实质审查的生效
IPC(主分类) : G06N 3/08
申请日 : 20211228
2022-04-29 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载
  • 联系电话
    电话:023-6033-8768
    QQ:1493236332
  • 联系 Q Q
    电话:023-6033-8768
    QQ:1493236332
  • 关注微信
    电话:023-6033-8768
    QQ:1493236332
  • 收藏
    电话:023-6033-8768
    QQ:1493236332