[发明专利]数据分类方法、装置、电子设备和存储介质

权利要求书

1.一种数据分类方法，包括：

将待处理数据输入数据分类模型，得到所述待处理数据针对预定类别的初始概率数据；

根据目标损失函数和所述初始概率数据，确定待处理数据针对所述预定类别的调整后概率数据；其中，所述目标损失函数以所述待处理数据针对所述预定类别的调整后概率数据为自变量；以及

基于所述调整后概率数据，确定所述待处理数据的类别，

其中，所述目标损失函数包括以下约束条件：所述调整后概率数据的数据分布满足预定分布。

2.根据权利要求1所述的方法，其中，所述目标损失函数还包括以下约束条件：

所述调整后概率数据的数据之和为预定值。

3.根据权利要求1或2所述的方法，其中，所述目标损失函数采用以下公式表示：

所述目标损失函数的约束条件包括：

{Y}^T{1}_N＝μ；

其中，表示Y与之间的距离，Y为所述调整后概率数据，为所述初始概率数据，Y与均为行数为N的矩阵数据；α为超参；{1}_N表示包括N个元素、且该N个元素的取值均为1的列向量，μ为所述预定分布。

4.根据权利要求3所述的方法，其中，根据目标损失函数和所述初始概率数据，确定待处理数据针对所述预定类别的调整后概率数据包括：

采用Sinkhorn算法最小化所述目标损失函数，得到所述调整后概率数据。

5.根据权利要求1或2所述的方法，其中，所述待处理数据包括多个数据；所述方法还包括：

针对所述多个数据中的每个数据，确定所述多个数据中针对所述每个数据的近邻数据；

基于针对所述每个数据的近邻数据，确定针对所述多个数据的数据关联信息；以及

基于所述数据关联信息，确定与所述调整后概率数据关联的正则项，

其中，所述正则项指示所述数据关联信息，且所述目标损失函数包括所述正则项。

6.根据权利要求5所述的方法，其中，所述目标损失函数采用以下公式表示：

所述目标损失函数的约束条件包括：

{Y}^T{1}_N＝μ；

其中，表示Y与之间的距离，Y为所述调整后概率数据，为所述初始概率数据，Y与均为行数为N的矩阵数据，α、β为超参，为所述正则项；为指示所述数据关联信息的拉普拉斯矩阵；{1}_N表示包括N个元素、且该N个元素的取值均为1的列向量，μ为所述预定分布。

7.根据权利要求5所述的方法，其中，根据目标损失函数和所述初始概率数据，确定待处理数据针对所述预定类别的调整后概率数据包括：

采用广义条件梯度算法最小化所述目标损失函数，得到所述调整后概率数据。

8.一种数据分类装置，包括：

初始概率获得模块，用于将待处理数据输入数据分类模型，得到所述待处理数据针对预定类别的初始概率数据；

调整后概率确定模块，用于根据目标损失函数和所述初始概率数据，确定待处理数据针对所述预定类别的调整后概率数据；其中，所述目标损失函数以所述待处理数据针对所述预定类别的调整后概率数据为自变量；以及

类别确定模块，用于基于所述调整后概率数据，确定所述待处理数据的类别，

其中，所述目标损失函数包括以下约束条件：所述调整后概率数据的数据分布满足预定分布。

9.根据权利要求8所述的装置，其中，所述目标损失函数还包括以下约束条件：

所述调整后概率数据的数据之和为预定值。