AmazonがSageMaker AIを強化、MLflow対応を拡充
- •Amazon SageMaker AIがMLflowバージョン3.10をサポート
- •生成AIワークフロー向けの高度な可観測性とトレーシング機能を導入
- •LLMの品質評価を自動化する新しいAPIを提供
生成AIの急速な発展に伴い、実験段階から本番環境への導入に至るまでのアプリケーション管理は極めて困難な課題となっている。データサイエンティストにとって、これは千もの要素を同時に追いかける作業に等しい。AWSはこの課題を解決すべく、機械学習モデルの実験追跡・管理ツールであるMLflowのバージョン3.10をAmazon SageMaker AIに統合した。
MLflowは、AI開発におけるバージョン管理システムのような役割を担う。今回のアップデートは、単なる入力と出力のループにとどまらず、複雑な対話を繰り返す生成AIワークフローに特化している。SageMakerに直接統合することで、チーム全体でAI開発の進捗を標準化し、監視することが容易になった。
今回の目玉の一つが、可観測性の強化である。これはシステムの出力から内部状態を把握する能力を指す。生成AIは確率論的に動作するため、同じ入力でも結果が異なることがあり、デバッグは困難を極める。最新版では、詳細なトレースのフィルタリングや事前構築されたダッシュボードを提供し、遅延やトークン消費量、品質スコアをリアルタイムで可視化できるようになった。
エージェンティックAI(自律型AI)のワークフローにおいても大幅な改善が図られた。自律型AIは複数のツールやデータベースを介して意思決定を行うため、監視が複雑化しやすい。今回追加されたトレース機能により、AIエージェントの推論プロセスを追跡し、論理の連鎖がどこで破綻したかを特定できるようになった。
さらに、新たに導入された「mlflow.genai.evaluation」APIは、品質管理のプログラム的アプローチを可能にする。従来の手動によるスポットチェックとは異なり、モデルの忠実度や正確性といった指標を開発パイプライン内で自動測定できる。自動化された評価手法の導入は、プロジェクトを実験室から実用的なエンタープライズ規模へと引き上げるために不可欠である。