Hugging Face Transformers と Ray Tune の統合
Hugging Face Transformers 3.1 は、Ray Tune と統合することで、NLP モデルの高度なハイパーパラメータチューニングを効率的に行う方法を提供します。この統合により、ユーザーは Transformers フレームワークのカスタマイズ性を損なうことなく、最先端のチューニングアルゴリズムにアクセスできるようになります。
高度なチューニングによるパフォーマンスの向上
高度なハイパーパラメータ最適化技術は、単純なグリッドサーチと比較して、精度とリソース効率の両面で大幅に上回ります。RTE データセットで BERT モデルを使用した実験では、Population-based Training (PBT) が他の手法と比較して、最高のテスト精度と最小の総 GPU 時間を達成しました:
| Algorithm | Best Val Acc. | Best Test Acc. | Total GPU min | Total $ cost |
|---|---|---|---|---|
| Grid Search | 74% | 65.4% | 45 min | $2.30 |
| Bayesian Optimization + Early Stop | 77% | 66.9% | 104 min | $5.30 |
| Population-based Training | 78% | 70.5% | 48 min | $2.45 |
統合の詳細と実装
この統合は、Transformers ライブラリの Trainer クラスを介して利用可能です。hyperparameter_search メソッド内で backend 引数を `