-
公开(公告)号:CN117150300A
公开(公告)日:2023-12-01
申请号:CN202311139754.0
申请日:2023-09-05
Applicant: 安徽工业大学
IPC: G06F18/214 , G06F18/24 , G06N3/0464
Abstract: 本申请公开了一种损失重平衡方法、装置和电子设备,所述方法包括:利用搭建的机器学习模型对带有标签的训练样本进行分类预测,获得分类预测结果;利用所述分类预测结果与对应的所述标签计算每个所述训练样本的梯度,根据所述梯度的正负区分所述训练样本获得正梯度样本和负梯度样本;采用困难样本挖掘方法对所述正梯度样本进行处理获得对应的正样本损失权重,联合采用困难样本挖掘方法和累积正负梯度比方法对所述负梯度样本进行处理获得对应的负样本损失权重;根据所述正样本损失权重和所述负样本损失权重调整交叉熵损失并获得对应的目标损失值;对所述目标损失值进行迭代直至所述机器学习模型收敛。