Ray tune:一个可扩展的超参数优化Python库
Tune是一个超参数优化库,可以用于PyTorch、TensorFlow, MXnet,keras等深度学习框架
超参数优化
在深度学习中, 除了可以学习参数外,还存在很多超参数,这些超参数对于网络的性能影响也十分巨大,不同的机器学习任务往往需要不 同的超参数,常见的超参数有: - 网络结构,包括神经元的连接关系,层数,每层的神经元的数量、激活函数的类型 - 优化参数,包括优化方法,学习率,小批量的样本数量等 - 正则化系数
对于超参数的搜索,一般使用的方法:人工搜索,网格搜索,随机搜索,贝叶斯优化
动态资源分配
由于目前深度学习的优化方法一般都采取随机梯度下降,因此我们可以通 过一组超参数的学习曲线来预估这组超参数配置是否有希望得到比较好的结 果。如果一组超参数配置的学习曲线不收敛或者收敛比较差,我们可以应用早 期停止(early-stopping)策略来中止当前的训练。
Tune的核心特征
- 多计算节点的分布式超参数的查找
- 支持多种深度学习框架,例如:pytorch,TensorFlow
- 结果直接可以用tensorboard可视化
- 可拓展的SOTA算法,例如:PBT,HyperBand/ASHA
- 整合了很多超参数优化库, 例如:Ax, HyperOpt,Bayesian Optimization
快速开始
安装下列工具:
pip install ray torch torchvison filelock
实例是使用网格搜索来训练一个CNN模型
import
使用Tensorboard可视化实验结果:
tensorboard --logdir ray_results
![v2-c2513a1fa2ffc60259ed547533715a78_b.jpg](http://img-03.proxy.5ce.com/view/image?&type=2&guid=8bf21e38-1830-eb11-8da9-e4434bdf6706&url=https://pic1.zhimg.com/v2-c2513a1fa2ffc60259ed547533715a78_b.jpg)
参考:Tune: A Scalable Hyperparameter Tuning Libraryray.readthedocs.io
![v2-05d9ddf18e9fbfcc5e16a36e5ef49e34_180x120.jpg](http://img-03.proxy.5ce.com/view/image?&type=2&guid=8bf21e38-1830-eb11-8da9-e4434bdf6706&url=https://pic1.zhimg.com/v2-05d9ddf18e9fbfcc5e16a36e5ef49e34_180x120.jpg)
![v2-a9add337ee39d79878975f2c2d0a1c43_ipico.jpg](http://img-03.proxy.5ce.com/view/image?&type=2&guid=8bf21e38-1830-eb11-8da9-e4434bdf6706&url=https://pic4.zhimg.com/v2-a9add337ee39d79878975f2c2d0a1c43_ipico.jpg)