分类模型的训练方法及装置、电子设备、存储介质
基本信息
申请号 | CN202210021820.3 | 申请日 | - |
公开(公告)号 | CN114358197A | 公开(公告)日 | 2022-04-15 |
申请公布号 | CN114358197A | 申请公布日 | 2022-04-15 |
分类号 | G06K9/62(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I;G06V10/764(2022.01)I;G06V10/774(2022.01)I | 分类 | 计算;推算;计数; |
发明人 | 单齐齐;周涛;史治国 | 申请(专利权)人 | 上海亿保健康管理有限公司 |
代理机构 | 北京超凡志成知识产权代理事务所(普通合伙) | 代理人 | 李丙林 |
地址 | 201800上海市嘉定区沪宜公路1185号1幢17层1718室、1720室 | ||
法律状态 | - |
摘要
摘要 | 本申请提供一种分类模型的训练方法及装置、电子设备、计算机可读存储介质,方法包括:构建与已训练的标准分类模型对应的轻量模型;将样本数据集中的样本发票图像分别输入标准分类模型和轻量模型,获得标准分类模型在预测过程的第一损失参数和轻量模型在预测过程中的第二损失参数;对第一损失参数和第二损失参数加权求和,得到目标损失参数;依据目标损失参数对轻量模型的网络参数进行调整,并迭代至轻量模型收敛,得到轻量分类模型。本申请方案,训练过程中参考了标准分类模型的第一凭据参数,可将标准分类模型从样本发票图像中学习到的知识传递至轻量模型,从而训练出预测效果更佳、模型复杂度较低的轻量分类模型。 |
