Hugging Face AutoTrain이 Image Classification 태스크를 추가해 코드 작성 없이 이미지 분류 모델을 학습할 수 있는 환경 제공

Context

머신러닝을 시작하려는 개발자들은 새로운 기술 스택 습득의 진입장벽에 직면해 있다. 특히 풀 스택 엔지니어가 사이드 프로젝트에 ML을 추가하려 할 때 시간과 기술적 복잡성이 장애물이 된다.

Technical Solution

• AutoTrain 플랫폼에 Computer Vision 태스크 추가: Image Classification 기능으로 이미지 카테고리화 모델 학습 지원
• 제로 구성 학습 방식 도입: 데이터 업로드 후 자동으로 모델 선택 및 학습 수행
• 드래그앤드롭 데이터 입력: 로컬 폴더 드래그 또는 Hugging Face Hub의 공개 데이터셋 연결 지원
• 자동 모델 후보 관리: 사용자가 시도할 모델 후보 개수 지정 시 AutoTrain이 자동으로 성능 저조 모델 중단 처리
• 무료 학습 옵션: 5개 후보 모델 + 500장 미만 이미지 조건에서 무료 학습 제공
• 통합 추론 위젯: 학습된 모델을 Hub에서 직접 테스트 가능한 인터페이스 제공

Impact

아티클의 예시 프로젝트(butterflies-classification)에서 5개 모델 후보 중 최고 성능 모델이 84% 정확도 달성