meta-pytorch/botorch
Bayesian optimization in PyTorch
何を解決するか
BoTorch はベイズ最適化のためのモジュール型フレームワークを提供し、高コストな関数の最適パラメータを効率的に見つけることを可能にします。確率的モデル、獲得関数、最適化アルゴリズムの組み合わせを簡素化し、最適化ループを実行できます。
動作方法
PyTorch に基づいて構築されており、自動微分と GPU 加速を活用して獲得関数を最適化します。再パラメータ化トリックを用いたモンテカルロベースの獲得関数により、制限のない仮定なしに幅広い確率的モデルに対応できます。GPyTorch とシームレスに統合され、ガウス過程(GPs)、ディープおよび畳み込みアーキテクチャなどの最先端の確率的モデルをサポートします。
対象ユーザー
ベイズ最適化および AI 分野の研究者や高度な実務家で、新しいアルゴリズムを実装するための低レベル API を必要とする方々。研究活動を行わないエンドユーザー向けには、Ax の使用を推奨しています。
特徴
- ベイズ最適化の基本要素を組み合わせるためのモジュール型インターフェース。
- GPU 加速と動的計算グラフをサポートするネイティブ PyTorch 対応。
- モンテカルロベースの獲得関数のサポート。
- 多タスク GPs、ディープカーネル学習、ディープ GPs に対応する GPyTorch との第一級の統合。
関連
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト