DOI QR코드

DOI QR Code

Active Learning with Pseudo Labeling for Robust Object Detection

강건한 객체탐지 구축을 위해 Pseudo Labeling 을 활용한 Active Learning

  • ChaeYoon Kim (Dept. of Artificial Intelligence Applications, Kwangwoon University) ;
  • Sangmin Lee (Dept. of Artificial Intelligence Applications, Kwangwoon University)
  • 김채윤 (광운대학교 인공지능응용학과) ;
  • 이상민 (광운대학교 인공지능응용학과)
  • Published : 2023.11.02

Abstract

딥러닝 기술의 발전은 고품질의 대규모 데이터에 크게 의존한다. 그러나, 데이터의 품질과 일관성을 유지하는 것은 상당한 비용과 시간이 소요된다. 이러한 문제를 해결하기 위해 최근 연구에서 최소한의 비용으로 최대의 성능을 추구하는 액티브 러닝(active learning) 기법이 주목받고 있는데, 액티브 러닝은 모델 관점에서 불확실성(uncertainty)이 높은 데이터들을 샘플링 하는데 중점을 둔다. 하지만, 레이블 생성에 있어서 여전히 많은 시간적, 자원적 비용이 불가피한 점을 고려할 때 보완이 불가피 하다. 본 논문에서는 의사-라벨링(pseudo labeling)을 활용한 준지도학습(semi-supervised learning) 방식과 학습 손실을 동시에 사용하여 모델의 불확실성(uncertainty)을 측정하는 방법론을 제안한다. 제안 방식은 레이블의 신뢰도(confidence)와 학습 손실의 최적화를 통해 비용 효율적인 데이터 레이블 생성 방식을 제안한다. 특히, 레이블 데이터의 품질(quality) 및 일관성(consistency) 측면에서 딥러닝 모델의 정확도 성능을 높임과 동시에 적은 데이터만으로도 효과적인 학습이 가능할 수 있는 메커니즘을 제안한다.

Keywords