损失函数动态加权的方法、装置、设备和存储介质

基本信息

申请号 CN202111242467.3 申请日 -
公开(公告)号 CN114090952A 公开(公告)日 2022-02-25
申请公布号 CN114090952A 申请公布日 2022-02-25
分类号 G06F17/15(2006.01)I;G06F17/18(2006.01)I;G06N20/00(2019.01)I 分类 计算;推算;计数;
发明人 周开龙;陈颖辉;王范萍;张玥 申请(专利权)人 上海小零网络科技有限公司
代理机构 北京万思博知识产权代理有限公司 代理人 秦贺余
地址 201306上海市浦东新区中国(上海)自由贸易试验区临港新片区海基六路218弄12号2楼
法律状态 -

摘要

摘要 本发明公开了一种损失函数动态加权的方法、装置、设备和存储介质,包括:获取样本集内每个类别的先验概率;根据所述先验概率和当次模型训练中所输出的每个类别的预测概率,确定每个类别的难易权值;基于所述每个类别的难易权值对预设的损失函数中的权值变量进行更新,得到难易权值更新后的损失函数,实现了动态调整损失函数的权值变量,避免人为划分的难易权值在所选取的损失函数中效果不佳的现象,这样,在利用更新后的损失函数,确定当次模型训练的损失值后,对模型的超参数进行更新,使得越难的类别损失占比越大,越简单的类别损失占比越小,从而提高了模型训练效果。