Amazon SageMaker AI MLflow Apps が MLflow 3.16 をサポート

Amazon SageMaker AI MLflow Apps において、MLflow 3.16 のサポートが開始されました。これにより、生成 AI アプリケーションやエージェントのデバッグに最適化された新しい Trace Explorer がデフォルトで利用可能になるほか、大規模なメトリクス履歴を持つ実験におけるチャート表示の高速化が実現します。新規作成される MLflow Apps はデフォルトで 3.16 が適用され、既存のアプリはメンテナンスウィンドウ中に自動でアップグレードされます。

🚀 アップデートのポイント

今回のアップデートは、特に Generative AI（生成 AI）の開発・デバッグ効率 を大きく向上させる内容になっています。

🛠 解決される課題と新機能

• 複雑なエージェントの可視化: 生成 AI エージェントの挙動を追跡する際、新しい「Trace Explorer」により、行の高さ調整や列の並べ替え、メタデータに基づくカスタム列の追加が可能になりました。これにより、「どのステップで意図しない挙動をしたか」の特定が格段に楽になります。
• 大量データの表示遅延: 実験データ（メトリクス）が蓄積されてくるとチャートの描画が重くなりがちですが、今回のリリースで表示速度が改善されています。

🇯🇵 日本国内リージョンの状況

最新のリージョン状況を確認したところ、以下の通りです。

• 東京リージョン (ap-northeast-1): 利用可能 です。
• 大阪リージョン (ap-northeast-3): 現在 計画中 であり、現時点では利用できません。

💡 運用上のアドバイス

• 移行作業は不要: 既存の MLflow Apps は AWS 側のメンテナンスウィンドウ中に自動でアップグレードされるため、ユーザー側での手動作業は発生しません。ただし、アップグレード後に UI の挙動や表示が変わるため、業務時間外のメンテナンス期間を念頭に置いておくと安心です。
• 新規プロジェクトでの活用: 新しく MLflow Apps を立ち上げる場合は、最初から MLflow 3.16 の恩恵を受けられます。

✅ 次のアクション

まずは東京リージョンにて、新しい Trace Explorer の操作感（カスタム列の追加など）を試してみることをおすすめします。

Reference: SageMaker AI MLflow developer guide