优化图像分类模型 - Amazon SageMaker

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

优化图像分类模型

自动模型优化(也称作超参数优化)通过运行很多在数据集上测试一系列超参数的作业来查找模型的最佳版本。您可以选择可优化超参数、每个超参数的值范围和一个目标指标。您可以从算法计算的指标中选择目标指标。自动模型优化将搜索所选超参数以找到导致优化目标指标的模型的值组合。

有关模型优化的更多信息,请参阅自动调整模型 SageMaker

由图像分类算法计算的指标

图像分类算法是一种指导式算法。它报告在训练期间计算的准确率指标。优化模型时,请选择此指标作为目标指标。

指标名称 描述 优化方向
validation:accuracy

正确预测数量与预测总数之比。

最大化

可优化图像分类超参数

使用以下超参数优化图像分类模型。对图像分类目标指标影响最大的超参数包括:mini_batch_sizelearning_rateoptimizer。根据选定 optimizer 优化与优化程序相关的超参数,例如 momentumweight_decaybeta_1beta_2epsgamma。例如,仅当 adamoptimizer 时,使用 beta_1beta_2

有关每个优化程序中使用哪些超参数的更多信息,请参阅图像分类超参数

参数名称 参数类型 建议的范围
beta_1

ContinuousParameterRanges

MinValue: 1e-6, MaxValue: 0.999

beta_2

ContinuousParameterRanges

MinValue: 1e-6, MaxValue: 0.999

eps

ContinuousParameterRanges

MinValue: 1e-8, MaxValue: 1.0

gamma

ContinuousParameterRanges

MinValue: 1e-8, MaxValue: 0.999

learning_rate

ContinuousParameterRanges

MinValue: 1e-6, MaxValue: 0.5

mini_batch_size

IntegerParameterRanges

MinValue: 8, MaxValue: 512

momentum

ContinuousParameterRanges

MinValue: 0.0, MaxValue: 0.999

optimizer

CategoricalParameterRanges

['sgd'、‘adam’、‘rmsprop’、'nag']

weight_decay

ContinuousParameterRanges

MinValue: 0.0, MaxValue: 0.999