【エムエルフロー】

MLflow とは?

最終更新:
💡 実験の記録とモデルの版を、追いかけやすくする

機械学習の実験やモデルを管理し、生成AIの追跡や評価も支えるオープンソースツール。記録する内容、モデルのバージョン管理、運用費用の考え方を説明します。

📌 このページのポイント
MLflow:実験の記録とモデルの版管理TrackingModel Registryrunの記録設定・評価値モデル等の成果物モデルの版管理バージョン・別名タグ必要なモデルを選んで登録自動記録の範囲は連携・設定で確認保存・サーバー運用・AI呼び出し費用は別
実験を記録することと、モデルを登録することは別です。生成AIの追跡や評価にも対応しますが、記録や別名だけで品質が保証されるわけではありません。
ひよこ ひよこ
何のために、実験を記録する?
ペンギン先生 ペンギン先生
学習の条件を変えたあと、どの設定でどんな結果が出たかを比べるためだよ。MLflow Trackingでは、一回の実行をrunとして扱い、パラメータ、評価値、モデルなどの成果物を記録するんだ。
ひよこ ひよこ
何も書かなくても、全部記録される?
ペンギン先生 ペンギン先生
記録用のAPIで必要な情報を残す方法と、対応するライブラリの自動記録を使う方法があるよ。autologを有効にしても、記録できる内容は連携先や設定による。独自の評価や必要なファイルは、どう残すか確認するんだ。
ひよこ ひよこ
モデルも、版を管理できる?
ペンギン先生 ペンギン先生
Model Registryで、登録したモデルのバージョンや別名、タグを管理できる。例えばchampionという別名を、利用したい版へ結び付ける方法があるよ。別名を付けただけで、モデルの品質が保証されるわけではないんだ。
ひよこ ひよこ
Productionへステージを変える説明を見たよ?
ペンギン先生 ペンギン先生
古い手順にはモデルのステージを使う例があるけれど、その仕組みはMLflow 2.9から非推奨だよ。現在の公式資料は別名やタグなどを使う移行方法を案内している。利用するバージョンの手順を確認しよう。
ひよこ ひよこ
生成AIにも、無料で使える?
ペンギン先生 ペンギン先生
オープンソースのツールとして、生成AIの処理を追跡するtracingや、評価、プロンプトの版管理にも対応しているよ。ただし、保存先やサーバーの運用、呼び出すAIサービスなどの費用は別。評価の仕組みを使えば回答の間違いが必ずなくなる、というものではないんだ。
もっと詳しく知りたい人へ

実験の記録とモデルの登録は、同じ?

実験の記録は、どの実行で何をしたかを残します。モデルの登録は、モデルの版や別名などを管理します。実験でモデルを保存したら、必ずすべてが自動で登録されるという意味ではありません。

別名とバージョン番号は、どう違う?

バージョン番号は登録したモデルの版を識別します。別名は選んだ版へ結び付ける名前で、後から別の版へ付け替えられます。設定した意味や変更の履歴を、チームで共有して使います。

ペンギン
まとめ:ざっくりこれだけ覚えればOK!
「MLflow」って出てきたら「機械学習の実験やモデルを、記録・管理するツール」と思えばだいたいOK!
📖 おまけ:英語の意味
「MLflow」 = 機械学習の実験やモデルを扱うツールの名前
💬 MLはMachine Learning(機械学習)、flowは流れを表す語です。現在は生成AIの処理の追跡や評価にも機能が広がっています。

参考資料

← 用語集にもどる