分类模型的训练方法、分类方法、装置、设备和存储介质
申请号:CN202410741973
申请日期:2024-06-07
公开号:CN118691882A
公开日期:2024-09-24
类型:发明专利
摘要
本发明实施例公开分类模型的训练方法、分类方法、装置、设备和存储介质,涉及计算机技术领域,该方法包括:获取有标签数据和无标签数据;确定与无标签数据对应的增强无标签数据;将有标签数据、有标签数据的真实类型、无标签数据以及增强无标签数据,作为训练数据输入初始分类模型,对初始分类模型进行模型训练,并计算损失函数;基于反向传播算法进行模型优化直至满足预设优化条件,得到训练完成的目标分类模型。上述技术方案,损失函数包括有监督损失函数和对称性一致度量损失函数构成的无监督损失函数,使得目标分类模型学习到较为准确的有标签数据和无标签数据的信息,根据目标分类模型可以确定待分类数据较为准确的概率分布,提升分类精度。
技术关键词
无标签数据
分类图像数据
无监督
分类方法
计算机可执行指令
传播算法
度量
计算机设备
处理器
分类装置
训练装置
输入模块
存储装置
策略
程序
强度