【发布时间】:2017-10-26 03:55:14
【问题描述】:
我正在为直接用 Tensorflow(不是 Keras 或 Tflearn)编写的代码搜索超参数调整包。你能提点建议吗?
【问题讨论】:
标签: optimization tensorflow machine-learning bayesian hyperparameters
我正在为直接用 Tensorflow(不是 Keras 或 Tflearn)编写的代码搜索超参数调整包。你能提点建议吗?
【问题讨论】:
标签: optimization tensorflow machine-learning bayesian hyperparameters
您可以试用 Ray Tune,这是一个用于缩放超参数搜索的简单库。我主要将它用于 Tensorflow 模型训练,但它与框架无关 - 可与 PyTorch、Keras 等无缝协作。这是文档页面 - ray.readthedocs.io/en/latest/tune.html
您可以使用它在大约 10 行代码中运行最先进算法的分布式版本,例如 HyperBand 或贝叶斯优化。
以一次运行 4 次并行评估为例:
import ray
import ray.tune as tune
from ray.tune.hyperband import HyperBandScheduler
def train_model(config, reporter): # add the reporter parameter
model = build_tf_model(config["alpha"], config["beta"])
loss = some_loss_function(model)
optimizer = tf.AdamOptimizer(loss)
for i in range(20):
optimizer.step()
stats = get_statistics()
reporter(timesteps_total=i,
mean_accuracy=stats["accuracy"])
ray.init(num_cpus=4)
tune.run(train_model,
name="my_experiment",
stop={"mean_accuracy": 100},
config={
"alpha": tune.grid_search([0.2, 0.4, 0.6]),
"beta": tune.grid_search([1, 2])
},
scheduler=HyperBandScheduler(reward_attr="mean_accuracy"))
如果您想在集群上运行此脚本,也无需更改代码。
免责声明:我从事这个项目 - 如果您有任何反馈,请告诉我!
【讨论】:
tune.run_experiments(...) 之后,我如何获得经过训练的 model 对象?
analysis = tune.run(...)。然后analysis.get_best_config.
您可以使用变分推理(贝叶斯)作为优化空间上的点云;超参数调整会好得多。张量流概率将是一种方法。
【讨论】:
我发现 sci-kit optimize 用于超参数的贝叶斯优化非常简单,它适用于任何 tensorflow API(估计器、自定义估计器、核心、keras 等)
【讨论】:
我不确定这是否也是您想要的参数,但您提到了 TensorFlow 超参数,所以我想我可以提出一些建议。
尝试克隆此存储库以获得所需的脚本;
git 克隆https://github.com/googlecodelabs/tensorflow-for-poets-2
在 Master 文件夹中,调用您的命令提示符并运行此行;
python -m scripts.retrain -h
获取可选参数列表。
来源:https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#6
【讨论】:
我想在@jdehesa 的列表中再添加一个库,我已将其应用到我的研究中,尤其是 tensorflow。它是hyper-engine,获得 Apache 2.0 许可。
它还实现了高斯过程贝叶斯优化和其他一些技术,如学习曲线预测,可以节省大量时间。
【讨论】:
通常您不需要将超参数优化逻辑与优化模型相结合(除非您的超参数优化逻辑特定于您正在训练的模型类型,在这种情况下,您需要告诉我们更多信息)。有几个工具和包可用于该任务。 Here 是一篇关于该主题的好论文,here 是一篇更实用的博文,并附有示例。
在这些中,我只有真正(即有一个真正的问题)将 hyperopt 与 TensorFlow 一起使用,并且没有花费太多精力。 API 在某些方面有点奇怪,文档也不是很详尽,但它确实有效并且似乎正在积极开发中,可能会出现更多优化算法和适应(例如专门针对神经网络)。但是,正如之前链接的博客文章中所建议的那样,Scikit-Optimize 可能也一样好,而且 SigOpt 看起来很容易使用,如果它适合您的话。
【讨论】: