FLAML: モデル選択とハイパーパラメータ最適化のための高速かつ経済的なAutoMLエンジン

FLAML: モデル選択とハイパーパラメータ最適化のための高速かつ経済的なAutoMLエンジン

何を解決するか

FLAMLは、機械学習とAIオペレーションをより経済的かつ効率的にするために設計されています。過剰な計算リソースを消費したり、深い手動チューニングの専門知識を必要としたりすることなく、最高のパフォーマンスを発揮するモデルとハイパーパラメータを見つけるという問題に対処します。

仕組み

FLAMLは、機械学習モデルの選択とハイパーパラメータの最適化を自動化する軽量なPythonライブラリを提供します。分類や回帰などの一般的なタスクをサポートし、scikit-learn スタイルのエスティメータとして使用できます。また、カスタム関数、基盤モデルの推論ハイパーパラメータ、およびMLOps/LMOpsワークフローのための汎用的なハイパーパラメータチューニングツールも提供します。

対象ユーザー

低い計算コストでデータに対して高品質なモデルを迅速に見つけたいデータサイエンティストやAI実務家、または、幅広いAIワークフローのために高速でリソース制約のあるハイパーパラメータチューニングツールを必要とする人々を対象としています。

ハイライト

  • scikit-learn スタイルのAPIを所有しており、容易に統合可能
  • 学習データに基づいてハイパーパラメータを自動的に設定するZero-shot AutoMLをサポート
  • 複雑な制約と早期終了(early stopping)を伴う大きな探索空間を扱うことが可能
  • MLflow および Microsoft Fabric Data Science との統合
  • XGBoost, LightGBM, および Random Forest を含む幅広いエスティメータをサポート

Sources