Amazon SageMaker Model Registry 是一個用於管理機器學習模型全生命週期的服務。主要功能包括:
• 為生產環境的模型建立目錄,集中管理所有模型。
• 管理模型的不同版本,追蹤每次訓練產生的模型。• 關聯模型的中繼資料(如訓練指標)與模型本身。
• 管理與檢視模型的核准狀態,支援模型審核流程。
• 支援模型的可追蹤性與可重現性,方便合規與審計。
• 將模型部署到生產環境,並可整合 CI/CD 流程自動化部署。
• 與其他用戶共享模型。
典型工作流程如下:
1. 建立模型群組(Model Group),用於管理同一類問題下的所有模型版本。
2. 每次訓練後,將新模型註冊為模型群組中的新版本。
3. 可進一步將多個模型群組整理到集合(Collection)中,方便分類管理。
開發者可透過 AWS SDK(如 Boto3)或 SageMaker Studio 主控台進行這些操作。
總結:SageMaker Model Registry 幫助企業標準化、版本化、審核與部署機器學習模型,是 MLOps 重要組件。