[发明专利]深度学习自动化模型训练方法及设备

摘要

本发明的目的是提供一种深度学习自动化模型训练方法及设备，本发明通过根据所述任务确定对应的算法模型的类型，可以在多种已知的适合算法模型中直接筛选，并通过每隔第三预设轮次后，根据每个并行训练的算法模型的每个轮次后在所述验证集数据上测试模型性能，筛选掉一个或多个不合格的所述并行训练的算法模型；获取每个未被筛选掉的并行的训练的算法模型的历史最佳性能对应的轮次中的批次区间；训练每个批次区间对应的未被筛选掉的并行的训练的算法模型，选择一个最佳的最佳批次的未被筛选掉的并行的训练的算法模型，实现在训练过程中继续阶梯性的选择算法，减少了计算资源的要求和计算的时间，降低了成本。

主权项

1.一种深度学习自动化模型训练方法，其中，该方法包括：获取将要进行的任务，根据所述任务确定对应的验证数据集、算法模型的类型和所述算法模型的参数限制与性能要求；根据数据集、算法模型的类型和所述算法模型的参数限制与性能要求，确认并行的训练多个算法模型后，进行多个所述算法模型的并行训练，其中，每个并行训练的算法模型进行第一预设轮次的并行训练，每个并行训练的算法模型在其前一轮次的训练结果上进行后一轮次的训练，每一轮次的训练包括多个批次的训练，每一轮次的训练中，每个并行训练的算法模型在其同一轮次的前一批次的训练结果上进行该轮次的后一批次的训练，所述并行训练中，在每个并行训练的算法模型的每个轮次后，在所述验证集数据上测试模型性能；每隔第二预设轮次后，存储每个并行训练的算法模型对应的训练得到算法模型，其中，所述第二预设轮次小于所述第一预设轮次；每隔第三预设轮次后，根据在每个并行训练的算法模型的每个轮次后，在所述验证集数据上测试模型性能，筛选掉不合格的并行训练的算法模型，其中，所述第三预设轮次小于所述第一预设轮次；得到尚未被筛选掉的一个或者若干个并行的训练的算法模型，根据每个未被筛选掉的并行的训练的算法模型的在每个轮次后在所述验证集数据上测试的算法模型性能，得到每个未被筛选掉的并行的训练的算法模型的历史最佳性能，获取每个未被筛选掉的并行的训练的算法模型的历史最佳性能对应的轮次中的批次区间，所述批次区间包括初始点和结束点；以每个批次区间的初始点为开始点，在所述存储的每个并行训练的算法模型对应的训练得到算法模型的基础，继续训练每个批次区间对应的未被筛选掉的并行的训练的算法模型，并用所述验证集数据测试所述批次区间内每个批次的算法模型性能，得到多个最佳批次的训练得到的算法模型，比较每个最佳批次的训练得到的算法模型的性能，选择一个最佳的最佳批次的训练得到的算法模型，作为最终算法模型。