Amazon Web Services 한국 블로그

Amazon SageMaker JumpStart – 기업 내에서 ML 모델 및 노트북 공유 기능 출시

Amazon SageMaker JumpStart는 ML 여정을 가속화하는 데 도움이 되는 기계 학습(ML) 허브입니다. SageMaker JumpStart를 사용하면 인기 모델 허브의 사전 학습된 모델, 기사 요약 및 이미지 생성과 같은 작업을 수행하는 데 도움이 되는 사전 학습된 기초 모델, 일반적인 사용 사례를 해결하는 엔드 투 엔드 솔루션을 포함한 내장 알고리즘에 액세스할 수 있습니다.

이제 SageMaker JumpStart를 사용하여 AWS 계정을 공유하는 다른 사용자와 모델 및 노트북과 같은 ML 아티팩트를 더 쉽게 공유할 수 있게 되었다는 소식을 전하게 되어 기쁩니다.

SageMaker JumpStart를 사용하여 ML 아티팩트 공유
기계 학습은 팀 스포츠입니다. 팀 내 다른 데이터 과학자와 모델 및 노트북을 공유하여 협업하고 생산성을 높일 수 있습니다. 또는 운영 팀과 모델을 공유하여 모델을 생산에 투입할 수도 있습니다. SageMaker JumpStart를 사용하여 ML 아티팩트를 공유하는 방법을 보여 드리겠습니다.

SageMaker Studio의 왼쪽 탐색 메뉴에서 Models(모델)을 선택합니다. 그런 다음 Shared models(공유된 모델) 및 Shared by my organization(내 조직에서 공유됨)을 선택합니다. 이제 AWS 계정에서 다른 사용자가 공유한 ML 아티팩트를 검색하고 검색할 수 있습니다. SageMaker로 개발한 ML 아티팩트는 물론 SageMaker 외부에서 개발된 아티팩트를 추가하고 공유할 수 있습니다.

모델이나 노트북을 공유하려면 Add(추가)를 선택합니다. 모델의 경우 제목, 설명, 데이터 유형, ML 작업, 프레임워크 및 기타 추가 메타데이터와 같은 기본 정보를 제공합니다. 이 정보는 다른 사용자가 사용 사례에 적합한 모델을 찾는 데 도움이 됩니다. 모델에 대한 학습 및 배포를 활성화할 수도 있습니다. 이를 통해 사용자는 SageMaker JumpStart를 통해 몇 번의 클릭만으로 공유 모델을 미세 조정하고 모델을 배포할 수 있습니다.

Amazon SageMaker JumpStart - 프라이빗 ML 허브에 모델 추가

모델 훈련을 활성화하려면 모든 관련 정보를 자동으로 채우는 기존 SageMaker 훈련 작업을 선택할 수 있습니다. 이 정보에는 컨테이너 프레임워크, 훈련 스크립트 위치, 모델 아티팩트 위치, 인스턴스 유형, 기본 학습 및 검증 데이터 세트, 대상 열이 포함됩니다. 또한 사전 구축된 SageMaker Deep Learning Container를 선택하거나 Amazon ECR에서 custom Docker container(사용자 지정 Docker 컨테이너)를 선택하여 사용자 지정 모델 훈련 정보를 제공할 수 있습니다. 모델 학습을 위한 기본 하이퍼파라미터 및 지표를 지정할 수도 있습니다.

모델 배포를 활성화하려면 사용할 컨테이너 이미지, 추론 스크립트 및 모델 아티팩트 위치, 기본 인스턴스 유형도 정의해야 합니다. 모델 훈련 및 모델 배포 옵션에 대해 자세히 알아보려면 SageMaker Developer Guide(SageMaker 개발자 안내서)를 참조하세요.

노트북 공유도 유사하게 작동합니다. 노트북에 대한 기본 정보와 노트북 파일의 Amazon S3 위치를 제공해야 합니다.

Amazon SageMaker JumpStart - 프라이빗 ML 허브에 노트북 추가

AWS 계정을 공유하는 사용자는 이제 공유 모델을 찾아 선택하여 SageMaker JumpStart에서 직접 세부 조정, 엔드포인트 배포 또는 노트북을 실행할 수 있습니다.

SageMaker Studio의 왼쪽 탐색 메뉴에서Quick start solutions(빠른 시작 솔루션)을 선택한 다음 Solutions, models, example notebooks(솔루션, 모델, 예시 노트북)을 선택하여 인기 모델 허브 및 엔드 투 엔드 솔루션의 사전 학습된 모델과 함께 모든 공유 ML 아티팩트에 액세스할 수 있습니다.

Amazon SageMaker JumpStart

정식 출시
Amazon SageMaker JumpStart의 새로운 ML 아티팩트 공유 기능은 현재 Amazon SageMaker JumpStart를 사용할 수 있는 모든 AWS 리전에서 사용할 수 있습니다. 자세히 알아보려면 Amazon SageMaker JumpStartSageMaker JumpStart 설명서를 참조하세요.

지금 Amazon SageMaker JumpStart로 모델 및 노트북 공유를 시작하세요!

– Antje