MLflowダイエット大作戦

2025年12月9日に開催された「AIエージェント/MLモデル開発を加速するMLflow:実践ノウハウ共有」での発表資料です。

Transcript

1. © LY Corporation AIPFユニット 湛溢洋 MLflowダイエット大作戦
2. © LY Corporation • 統一的な実験管理 • 様々なライブラリとの連携 • AIエージェント開発の強力なパートナー 2 MLflowは素晴らしい
3. © LY Corporation 3 一方で、MLflowのちょっと気にな るところ
4. © LY Corporation • pip install mlflow すると79個ライブラリが入る（2025年12月時点） • MLflow Tracking Serverにちょっと実験結果を送りたいだけの場合でも依 存ライブラリが全部ついてきてしまう 4 MLflowは依存ライブラリが多い
5. © LY Corporation • コンテナイメージが重くなる • 脆弱性スキャナーに指摘されがち 5 依存ライブラリが多いと困ること
6. © LY Corporation • mlflowパッケージの軽量版 • SQLストレージ、サーバー、UI、データサイエンス系ライブラリが依存か ら除かれている 6 mlflow-skinny
7. © LY Corporation 7 依存ライブラリ数: 79 → 39
8. © LY Corporation • まだfastapiとかuvicornとか入っている 8 まだいける？ 本当に必要？
9. © LY Corporation • まだfastapiとかuvicornとか入っている • pip install --no-deps -r requirements.txt のようにすれば依存ライブラリ を無視できるので要らなそうなライブラリを無理やり削ってみる • 動作は保証されません 9 まだいける？
10. © LY Corporation # パラメーターとメトリクスを送るだけ # https://github.com/zhanpon/mlflow-skinnier mlflow.set_tracking_uri("http://localhost:5000") with mlflow.start(): mlflow.log_params({"magical_params" : 3.1415}) mlflow.log_metric("cool_metric", 0.8) 10 今回のテストコード
11. © LY Corporation 11 依存ライブラリ数: 39 → 22
12. © LY Corporation 12
13. © LY Corporation • 削るとImportErrorが出るが実行時には使わなさそうなライブラリがある: e.g. PyYaml 13 まだいける？？
14. © LY Corporation 14 本当に必要？
15. © LY Corporation • 削るとImportErrorが出るが実行時には使わなさそうなライブラリがある • import文をpassに置換してみる（動作は保証されません） MLFLOW_DIR=.tox/test/lib/python3.14/site-packages/mlflow find ${MLFLOW_DIR} -type f -exec sed -i 's/import yaml.*/pass/g' {} + find ${MLFLOW_DIR} -type f -exec sed -i 's/from yaml.*/pass/g' {} + 15 まだいける？？
16. © LY Corporation 16 依存ライブラリ数: 22 → 21 ※今回はPyYAMLだけ検証しましたがもっといけるはず
17. © LY Corporation • 2026年10月リリース予定のPython 3.15からはlazy importが使えるように なる予定: PEP 810 • lazy import yaml みたいに書けば本当に必要になるまでimportされない 17 ちなみに
18. © LY Corporation • Step 1: mlflow-skinnyを使う（推奨） • Step 2: 指定を無視して依存ライブラリを削る（非推奨） • Step 3: import文を削除してさらに削る（非推奨） 18 まとめ
19. © LY Corporation • 湛 溢洋（ざん いいやん） • 福岡在住 • 「Python Meetup Fukuoka」や「ゆるっと数学勉強会」をやってます 19 自己紹介