MLtuner: System Support for Automatic Machine Learning Tuning
arXiv:1803.07445
Abstract
MLtuner automatically tunes settings for training tunables (such as the learning rate, the momentum, the mini-batch size, and the data staleness bound) that have a significant impact on large-scale machine learning (ML) performance. Traditionally, these tunables are set manually, which is unsurprisingly error-prone and difficult to do without extensive domain knowledge. MLtuner uses efficient snapshotting, branching, and optimization-guided online trial-and-error to find good initial settings as well as to re-tune settings during execution. Experiments show that MLtuner can robustly find and re-tune tunable settings for a variety of ML applications, including image classification (for 3 models and 2 datasets), video classification, and matrix factorization. Compared to state-of-the-art ML auto-tuning approaches, MLtuner is more robust for large problems and over an order of magnitude faster.
References in corpus (10)
- Adam: A Method for Stochastic Optimization
- Practical Bayesian Optimization of Machine Learning Algorithms
- ADADELTA: An Adaptive Learning Rate Method
- UCF101: A Dataset of 101 Human Actions Classes From Videos in The Wild
- Caffe: Convolutional Architecture for Fast Feature Embedding
- MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systems
- Going Deeper with Convolutions
- Gradient-based Hyperparameter Optimization through Reversible Learning
- Hyperparameter optimization with approximate gradient
- Poseidon: A System Architecture for Efficient GPU-based Deep Learning on Multiple Machines