针对类别不均衡数据集的网络训练方法
实质审查的生效
摘要
本发明公开了一种针对类别不均衡数据集的网络训练方法,包括:获取目标图像数据集,确定类别数目和每类样本量;利用每类样本量计算相应类别的权值,并结合设置的超参数构建限制误差损失函数;利用所述目标图像数据集对神经网络模型进行训练,并将样本的预测结果与真实标签带入所述限制误差损失函数进行误差计算,使用反向传播不断更新神经网络模型的参数,直至网络收敛达到预期目标。构建的限制误差损失函数按类别数量进行加权,并且通过引入超参数对尾部泛化进行正则化的LDAM,可以将训练的关注度更多的偏向于数量较少的尾部类别,防止网络训练对尾部类欠拟合,可以应用于不均衡的图像数据集中,并可以显著提高网络对于不均衡数据集的识别准确度。
基本信息
专利标题 :
针对类别不均衡数据集的网络训练方法
专利标题(英):
暂无
公开(公告)号 :
CN114332539A
申请号 :
CN202111671005.3
公开(公告)日 :
2022-04-12
申请日 :
2021-12-31
授权号 :
暂无
授权日 :
暂无
发明人 :
任维姚鹏徐亮陈明潇
申请人 :
深圳友一生物科技有限公司;中国科学技术大学
申请人地址 :
广东省深圳市龙岗区园山街道安良社区油田路28号B1栋805
代理机构 :
北京凯特来知识产权代理有限公司
代理人 :
郑立明
优先权 :
CN202111671005.3
主分类号 :
G06V10/764
IPC分类号 :
G06V10/764 G06V10/82 G06K9/62 G06N3/08
法律状态
2022-04-29 :
实质审查的生效
IPC(主分类) : G06V 10/764
申请日 : 20211231
申请日 : 20211231
2022-04-12 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载