AI로 옷을 찾아주는 무신사, 6주간의 여정
by DD
11개월 전
조회수 20
무신사 스냅의 방대한 이미지 데이터에서 객체 탐지, 임베딩, 벡터 검색을 활용하여 유사 상품 검색 시스템을 구축함
YOLOS, FashionCLIP 2.0, AWS OpenSearch를 사용하여 이미지 기반 검색 파이프라인을 구현하고, Precision@5 81%, MRR 0.76의 성능을 달성함
Amazon SageMaker, Bedrock, OpenSearch 등 AWS 서비스를 활용하여, AI/ML 모델 서빙 및 LLM API 호출 비용이 높다는 것을 확인하고, 개선 방안을 모색함
이미지 검색 아키텍처: End-to-End 파이프라인
스냅 이미지 검색은 객체 탐지 모델(YOLOS)로 상품 영역을 찾고, FashionCLIP 2.0으로 임베딩하여 벡터화한다. AWS OpenSearch에서 HNSW 알고리즘을 활용해 유사 상품을 검색한다. 따라서, 모델 서빙을 위한 SageMaker Endpoint와 API Gateway, Lambda를 구축하여, 유연한 확장이 가능한 구조를 설계했다. 🚀
