多异构模型融合的分类模型训练方法、分类方法及装置

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
多异构模型融合的分类模型训练方法、分类方法及装置
申请号:CN202510011216
申请日期:2025-01-03
公开号:CN119992248B
公开日期:2025-11-18
类型:发明专利
摘要
本发明提供一种多异构模型融合的分类模型训练方法、分类方法及装置,该多异构模型融合的分类模型训练方法包括:基于DS证据理论对多个教师模型的输出表示进行融合,得到融合后的输出表示,并根据融合后的输出表示确定多个教师模型对应的类别集中各类别的概率分布向量;基于概率分布向量指导生成对抗网络GAN进行训练,得到训练后的GAN,并根据概率分布向量、训练后的GAN对学生模型进行蒸馏训练,得到目标学生模型,以执行图像分类任务。本发明所述方法实现利用DS证据理论以及生成模型对图像分类模型的融合和蒸馏,提高了图像分类模型的预测性能,进而提高了图像分类结果的准确率。
技术关键词
分类模型训练方法 分类图像数据 DS证据理论 联合损失函数 生成对抗网络 学生 异构 教师 图像分类模型 分类方法 非暂态计算机可读存储介质 蒸馏 图像获取模块 处理器 分类装置 存储器 样本 电子设备