이미지 분류 모델 개발, 아직도 데이터만 기다려?

카카오맵 리뷰 이미지 분류 모델 개발 과정을 공유하는 글입니다. 딥러닝 기술이 성숙함에 따라 모델 자체보다는 학습 데이터의 양과 품질이 중요해졌습니다.

본 프로젝트에서는 '실내', '장소 외관', '야외', '음식', '메뉴판', '스팸' 6가지 카테고리를 분류하기 위해 CVPR 2022에서 공개된 ConvNeXt 모델을 선정했습니다. 특히, Zero-shot 분류 기법을 활용하여 가공된 선분류 데이터셋으로 모델 개발 기간을 단축하고, OOD(Out-of-Distribution) 필터링을 통해 정의되지 않은 카테고리까지 분류하는 방법을 중점적으로 다룹니다.

이를 통해 효율적인 이미지 분류 모델 개발 방법을 제시합니다.