模型训练方法、装置、设备、存储介质及程序产品

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
模型训练方法、装置、设备、存储介质及程序产品
申请号:CN202411037429
申请日期:2024-07-31
公开号:CN118968219B
公开日期:2025-06-06
类型:发明专利
摘要
本发明实施例提供了模型训练方法、装置、设备、存储介质及程序产品。该方法包括:基于设定的图像数据集对设定的深度神经网络模型进行预训练,获得预训练后的深度神经网络模型,并作为全局模型以及每个子训练任务的模型;基于多种噪声强度确定目标噪声强度;基于所述目标噪声强度以及设定的子训练任务数量确定每个子训练任务的噪声分布;基于各个子训练任务的噪声分布、各个子训练任务的权重参数以及所述全局模型的权重参数对各个子训练任务以及所述全局模型进行多轮训练,获得训练后的全局模型。本公开实施例,可以使得训练后的全局模型在器件变化环境下具备更强的鲁棒性和性能,以及在不同硬件环境下,具备更好的适应能力和通用性。
技术关键词
噪声强度 深度神经网络模型 模型训练方法 参数 计算机可执行指令 精度 误差 模型训练装置 模型训练模块 处理器 计算机程序产品 模型更新 电子设备 存储装置 加速器 索引 图像