Enregistrer et déployer des modèles avec Model Registry - Amazon SageMaker

Enregistrer et déployer des modèles avec Model Registry

Voici ce que vous pouvez faire avec le registre de modèles SageMaker :

  • Cataloguer des modèles pour la production.

  • Gérer les versions de modèles.

  • Associer des métadonnées, telles que des métriques d'entraînement, à un modèle.

  • Gérer le statut d'approbation d'un modèle.

  • Déployer des modèles en production.

  • Automatiser le déploiement de modèles avec CI/CD.

Cataloguez les modèles en créant des groupes de packages de modèles qui contiennent différentes versions d'un modèle. Vous pouvez créer un groupe de modèles qui suit tous les modèles que vous entraînez pour résoudre un problème particulier. Vous pouvez ensuite enregistrer chaque modèle entraîné et le registre de modèles l'ajoute au groupe de modèles en tant que nouvelle version de modèle. Un flux type peut ressembler à ce qui suit :

  • Créez un groupe de modèles.

  • Créez un pipeline ML qui entraîne un modèle. Pour obtenir des informations sur les pipelines SageMaker, veuillez consulter Créer et gérer SageMaker Pipelines.

  • Pour chaque exécution du pipeline ML, créez une version de modèle que vous enregistrez dans le groupe de modèles créé lors de la première étape.

Structure de Model Registry

SageMaker Model Registry est structuré en plusieurs groupes de modèles, chaque groupe comportant des packages de modèles. Chaque package de modèles d'un groupe de modèles correspond à un modèle entraîné. La version de chaque package de modèles est une valeur numérique qui commence à 1 et qui est incrémentée chaque fois qu'un nouveau package de modèles est ajouté à un groupe de modèles. Par exemple, si 5 packages de modèles sont ajoutés à un groupe de modèles, les versions du package de modèles seront 1, 2, 3, 4 et 5. L'exemple Model Registry illustré dans l'image suivante contient 3 groupes de modèles, chacun comportant les packages de modèles associés à un problème de ML particulier.

Il existe deux types de packages de modèles dans SageMaker. Un type est utilisé dans AWS Marketplace, et l'autre est utilisé dans Model Registry. Les packages de modèles utilisés dans AWS Marketplace ne sont pas des entités pouvant être versionnées et ne sont pas associés à des groupes de modèles dans Model Registry. Pour plus d'informations sur les packages de modèles utilisés dans AWS Marketplace, consultez Acheter et vendre des algorithmes et des modèles Amazon SageMaker sur AWS Marketplace.

Les packages de modèles utilisés dans Model Registry sont versionnés et doivent être associés à un groupe de modèles. L'ARN de ce type de package de modèles possède la structure suivante : 'arn:aws:sagemaker:region:account:model-group/version'

Les rubriques suivantes vous expliquent comment utiliser Model Registry.