学习率、迭代次数和初始化方式对模型准确率的影响
想必学过机器学习的人都知道,学习率、训练迭代次数和模型参数的初始化方式都对模型最后的准确率有一定的影响,那么影响到底有多大呢?
我初步做了个实验,在 TensorFlow
框架下使用 Logistics Regression 对经典的 MNIST
数据集进行分类。
本文所说的 准确率 均指 测试准确率。
代码
1 | from tensorflow.examples.tutorials.mnist import input_data |
通过修改 learning_rate
和 training_epochs
来修改学习率和迭代次数,修改
1 | # 所有变量初始化为0 |
来修改变量的初始化方式。程序最终会输出损失和准确率随着迭代次数的变化趋势图。
结果
以下结果的背景是:TensorFlow,Logistics Regression,MNIST数据集,很可能换一个数据集下面的结论中的某一条就不成立啦,所以要具体情况具体分析,找到最优的超参数组合。
多次的更改会输出多个不同的图,我们先来看下最终的准确率比较,然后再看下每种情况的详细的损失和准确率变化。
符号说明
lr
:Learning Rate,学习率te
:Training Epochs,训练迭代次数z
:tf.zeros()
,变量初始化为0t
:tf.truncated_normal()
,变量初始化为标准截断正态分布的随机数
最终准确率比较
可以看到
- 学习率为0.1,迭代次数为50次,并且采用随机初始化方式时准确率远远低于其他方式,甚至不足90%。而学习率为0.1,迭代次数为50次,并且采用随机初始化的方式时准确率最高。
- 对于采用随机初始化的方式,在其他参数相同的情况下增大迭代次数会明显的提高准确率。而对于初始化为0的情况则无明显变化。
- 其他参数相同的情况下,过度增大学习率的确是会导致准确率下降的,查看详细变化过程时可以看到准确率变化波动比较大。
- 在学习率适中,迭代次数较大时变量初始化方式对最终准确率的影响不大。
每种情况损失和准确率的详细变化趋势
与上图的顺序保持一致,从上至下。
每张图的标题在图的下面,斜体字。
学习率为1,迭代次数为50,随机初始化
学习率为1,迭代次数为50,初始化为0
学习率为0.1,迭代次数为50,随机初始化
学习率为0.1,迭代次数为50,初始化为0
学习率为0.1,迭代次数为25,随机初始化
学习率为0.1,迭代次数为25,初始化为0
学习率为0.01,迭代次数为50,随机初始化
学习率为0.01,迭代次数为50,初始化为0
目录
大部分情况下准确率和损失的变化时单调的,但是当学习率过大(=1
)时准确率开始不稳定。
END
暂且就是这么多,我说的难免有不合适的地方,有错误的地方欢迎指出。