Object Classification with Angular Margin Loss Function

각도 마진 손실 함수를 적용한 객체 분류

  • Park, Seonji (Department of ECE, INMC, Seoul National University) ;
  • Cho, Namik (Department of ECE, INMC, Seoul National University)
  • 박선지 (서울대학교 전기정보공학부, 뉴미디어통신공동연구소) ;
  • 조남익 (서울대학교 전기정보공학부, 뉴미디어통신공동연구소)
  • Published : 2022.06.20

Abstract

객체 분류는 입력으로 주어진 이미지에 포함된 객체의 종류를 판단하는 기술이다. 대표적인 딥러닝 기반의 객체 분류 방법으로서 Faster R-CNN[2], YOLO[3] 등의 모델이 개발되었으나, 여전히 성능 향상의 여지가 있다. 본 연구에서는 각도 마진 손실 함수를 기존의 몇 가지 객채 분류 모델에 적용하여 성능 향상을 유도한다. 각도 마진 손실 함수는 얼굴 인식 모델인 SphereFace [4]에서 제안한 방법으로, 얼굴 인식과 같이 단일 도메인의 데이터셋을 분류하는 문제를 풀기 위해 제안되었다. 이는 기존 소프트맥스 함수에서 클래스 결정 경계선에 마진을 주는 방식으로 클래스 간의 구분 능력을 향상시킨다. 본 논문은 각도 마진 손실 함수를 CIFAR10, CIFAR100 데이터셋의 분류 문제에 적용하였으며 ResNet, EfficientNet, MobileNet 등의 백본 네트워크로 실험하여 평균적으로 mAP 성능이 향상되는 것을 확인하였다.

Keywords

Acknowledgement

본 연구는 과학기술정보통신부 및 정보통신기획평가원의 정보통신·방송 연구개발사업의 일환으로 수행하였으며 2022년도 BK21 FOUR 정보기술 미래인재 교육연구단에 의하여 지원되었음. [2021-0-01062-001, 자율주행용 수집/활용 데이터에 대한 개인정보 처리 기술개발]