一种损失函数优化方法、装置、计算机设备及存储介质制造方法及图纸

本申请属于图像分类技术领域，公开了一种损失函数优化方法、装置、计算机设备及存储介质，该方法包括：获得各图像的标签概率列表；将真实标签相同的标签概率列表进行拼接，并对其执行标签平滑操作，得到n个待筛选矩阵；采用主成分分析法对待筛选矩阵进行降维，得到待叠加矩阵；将待叠加矩阵中同列数值相加，得到待约束列表；对待约束列表进行归一化后计算概率方差值，并获取真实标签对应的预测概率值；对n个待筛选矩阵遍历上述步骤，得到n个概率方差值和n个预测概率值；基于上述概率方差值、预测概率值和真实标签得到优化损失函数。本申请能够优化损失函数，提升损失函数的召回率和收敛速度，使得优化后的损失函数输出的分类结果更加准确。准确。准确。

【技术实现步骤摘要】
一种损失函数优化方法、装置、计算机设备及存储介质


[0001]本申请涉及图像分类
，尤其涉及一种损失函数优化方法、装置、计算机设备及存储介质。

技术介绍

[0002]在车款图像的分类任务中，通常先采用卷积网络分类模型对车款图像进行分类，得到每个图像样本的多个分类标签和各分类标签对应的概率，再采用损失函数根据标签和概率分布图进行预测，得到最终的分类结果。Distribution Focal Loss(DFL，狄拉克)损失函数是计算模型预测值与现实真实值之间差异的一种公式，损失函数到达最优则表明预测值与真实值差异最小，预测的分类结果最准确，而若要求该损失函数每次分类的结果百分之百准确，则要求在标签与概率的分布图中，真实标签上的概率是最高值应该为1，其余标签概率值之和为0。使得DFL损失函数计算真实标签左边以及右边与对应概率曲线所形成的面积最小，此时损失函数的预测值为最准确的分类结果。但是在车款图像分类问题之中，不同车款的图像之间的差异较小，例如同一品牌下的燃油车外观与新能源车外观区别仅存在于车头视角之中，使得卷积网络分类模型输出的标签和概率分本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种损失函数优化方法，其特征在于，所述方法包括：步骤S1，将训练图像集输入卷积网络分类模型，得到所述训练图像集中各图像的标签概率列表，所述训练图像集中的各所述图像包括真实标签；步骤S2，将所述真实标签相同的多个所述图像对应的所述标签概率列表进行拼接，得到n个标签概率矩阵，对n个所述标签概率矩阵执行标签平滑操作，得到n个待筛选矩阵；步骤S3，采用主成分分析法对所述待筛选矩阵进行降维，得到待叠加矩阵；步骤S4，将所述待叠加矩阵中的同列的概率值进行相加，得到待约束列表；对所述待约束列表进行归一化操作，得到归一化列表；计算所述归一化列表的概率方差值，并获取所述归一化列表中所述真实标签对应的预测概率值；步骤S5，分别对n个所述待筛选矩阵依次执行步骤S3和步骤S4，得到n个概率方差值和n个预测概率值；基于n个所述概率方差值、n个所述预测概率值和各所述预测概率值对应的所述真实标签得到优化损失函数。2.根据权利要求1所述的方法，其特征在于，所述对n个所述标签概率矩阵执行标签平滑操作，得到n个待筛选矩阵，包括：将n个所述标签概率矩阵中的所述概率值均加上预设平滑值，得到n个所述待筛选矩阵。3.根据权利要求2所述的方法，其特征在于，所述预设平滑值为0.1。4.根据权利要求1所述的方法，其特征在于，所述对所述待约束列表进行归一化操作，得到归一化列表，包括：采用归一化指数函数对所述待约束列表进行归一化操作，以使所述待约束列表中的各所述概率值均大于等于0且小于等于1，得到所述归一化列表。5.根据权利要求1所述的方法，其特征在于，所述基于n个所述概率方差值、n个所述预测概率值和各所述预测概率值对应的所述真实标签得到优化损失函数，包括：根据n个所述概率方差值得到方差平均值；基于n个所述预测概率值和各所述预测概率值对应的所述真实标签得到狄拉克损失函数；基于所述方差平均值和n个所述预测概率值得到焦点损失函数；根据所...

【专利技术属性】
技术研发人员：王子镜，甘霖，
申请(专利权)人：国汽大有时空科技安庆有限公司，
类型：发明
国别省市：