bayesflow-org/bayesflow

何を解決するか

BayesFlowは、ディープラーニングを用いたベイズ推論を簡素化するツールです。複雑なシミュレータや従来のパラメトリックモデルから生成されたデータからパラメータを推定するプロセスを容易にし、従来のベイズ手法の遅い反復プロセスを回避する「アモルタイズド（ amortized ）」ワークフローを活用できます。

動作方法

このライブラリはニューラルネットワークを用いて観測値（データ）とパラメータの関係を学習します。ポストリオリ、尤度、尤度比、ポイント推定器など、さまざまなニューラル推定器をサポートしており、拡散モデルや一貫性モデルといった現代的な生成AIを活用して、ポストリオリ分布を近似します。Keras 3に基づいているため、JAX、PyTorch、TensorFlowの複数のバックエンドをサポートしています。

対象ユーザー

感染症モデリング（SIR）、認知モデリング、空間データや画像データなど、複雑なシミュレーションベースのデータに対してベイズ推論を実行する必要がある研究者やデータサイエンティスト向けです。

特徴

• 複数バックエンド対応：Keras 3を介してJAX、PyTorch、TensorFlowとシームレスに連携可能。
• アモルタイズドワークフロー：ネットワークの学習が完了すれば、高速な推論が可能。同じモデルを複数のデータセットに再利用可能。
• 生成AIの統合：拡散モデルや一貫性モデルを含む豊富な生成モデルを搭載。
• クロスオーバー機能：PyMCと連携し、尤度推定やMCMCサンプリングを実現。
• 組み込み診断ツール：推論プロセスの結果を検証・理解するためのツールセットを提供。