Amazon SageMaker HyperPod로 슈퍼브에이아이의 비전 파운데이션 모델 ‘ZERO’ 효율적으로 대규모 분산 학습하기

슈퍼브에이아이는 비전 AI 노하우를 바탕으로 개발한 비전 파운데이션 모델 'ZERO(Zero-shot Object Detector)'를 Amazon SageMaker HyperPod를 이용해 효율적으로 대규모 분산 학습시킨 경험을 공유합니다. 기존 클라우드 환경의 유연성 부족 문제를 해결하기 위해 SageMaker HyperPod를 선택했으며, 사용 시나리오에 따른 인스턴스 크기 유연성 조절과 고성능 네트워크(EFA) 탑재가 큰 장점이었습니다.

약 10억 장의 원시 데이터 중 400만 장의 고품질 데이터를 선별하여 학습시킨 결과, ZERO는 특정 도메인에 국한되지 않고 오픈 월드 환경에서 다양한 객체를 인식하는 범용성을 확보했습니다. ZERO는 텍스트 또는 이미지 프롬프트를 통해 객체 위치를 정밀하게 탐지하는 멀티모달 그라운딩 기능을 제공하여, 별도의 데이터 수집이나 파인튜닝 없이 제로샷 방식으로 산업 현장의 결함을 즉시 검출할 수 있습니다.

SageMaker HyperPod의 유연한 인스턴스 크기 조절 기능은 GPU 사용량이 일정하지 않은 개발 프로세스에서 비용 효율성을 높이는 데 기여했습니다.