珠海博韬申请一种工业缺陷检测预训练模型的微调方法专利,提高模型泛化性能和收敛速度|算法
发布日期:2025-01-22 13:29 点击次数:197
金融界2024年12月10日消息,国家知识产权局信息显示,珠海博韬科技有限公司申请一项名为“一种工业缺陷检测预训练模型的微调方法”的专利,公开号 CN 119091250 A,申请日期为2024年8月。专利摘要显示,本发明公开了一种工业缺陷检测预训练模型的微调方法,旨在提供一种实际应用只需较少的缺陷标注数据,使用增量学习方法,抽样学习旧的数据,使用对数形式的类别权重平衡函数,提高模型泛化性能和收敛速度的工业缺陷检测预训练模型的微调方法。本发明包括以下具体步骤:预训练模型制作阶段S1、在工业训练模型进行权重迁移、通过类别权重交叉熵损失函数进行缺陷样本的训练、计算每个缺陷类别的权重、利用余弦退火算法对学习率进行动态调整;预训练模型微调阶段S2、确定与训练数据集、预训练模型中初始化参数并在特定任务的数据集上进行微调、使用Adam优化算法更新模型参数。本发明应用于工业视觉检测的技术领域。本文源自:金融界作者:情报员