基于强化学习的缺陷识别模型轻量化方法及系统

本发明公开了一种基于强化学习的缺陷识别模型轻量化方法及系统，所述方法包括根据缺陷识别模型构建样本数据集；利用第i个样本对策略网络和价值网络进行第i次更新，得到第i次更新后的策略网络和价值网络；根据第i次更新后的策略网络和总目标裁剪比例对缺陷识别模型进行裁剪，并计算裁剪后的缺陷识别模型在验证集上的识别精度；在完成策略网络和价值网络的训练后，根据训练好的策略网络和总目标裁剪比例对缺陷识别模型进行裁剪，得到轻量化的缺陷识别模型。本发明在确保缺陷识别精度的基础上，降低了缺陷识别模型的参数量，实现了缺陷识别模型的识别性能与轻量化平衡。