다중 레이블 분류 작업에서의 Coarse-to-Fine Curriculum Learning 메카니즘 적용 방안

Applying Coarse-to-Fine Curriculum Learning Mechanism to the multi-label classification task

  • 공희산 (성균관대학교 소프트웨어학과) ;
  • 박재훈 (성균관대학교 소프트웨어학과) ;
  • 김광수 (성균관대학교 소프트웨어학과)
  • Kong, Heesan (Dept. of Computer Science and Engineering, SungKyunKwan University) ;
  • Park, Jaehun (Dept. of Computer Science and Engineering, SungKyunKwan University) ;
  • Kim, Kwangsu (Dept. of Computer Science and Engineering, SungKyunKwan University)
  • 발행 : 2022.07.13

초록

Curriculum learning은 딥러닝의 성능을 향상시키기 위해 사람의 학습 과정과 유사하게 일종의 'curriculum'을 도입해 모델을 학습시키는 방법이다. 대부분의 연구는 학습 데이터 중 개별 샘플의 난이도를 기반으로 점진적으로 모델을 학습시키는 방안에 중점을 두고 있다. 그러나, coarse-to-fine 메카니즘은 데이터의 난이도보다 학습에 사용되는 class의 유사도가 더욱 중요하다고 주장하며, 여러 난이도의 auxiliary task를 차례로 학습하는 방법을 제안했다. 그러나, 이 방법은 혼동행렬 기반으로 class의 유사성을 판단해 auxiliary task를 생성함으로 다중 레이블 분류에는 적용하기 어렵다는 한계점이 있다. 따라서, 본 논문에서는 multi-label 환경에서 multi-class와 binary task를 생성하는 방법을 제안해 coarse-to-fine 메카니즘 적용을 위한 방안을 제시하고, 그 결과를 분석한다.

키워드

과제정보

2022년도 정부(과학기술정보통신부)의 재원으로 정보통신기획평가원의 지원을 받아 수행된 연구임 (No.2021-0-02068, 인공지능 혁신 허브 연구 개발)과 과학기술정보통신부 및 정보통신기획평가원의 지역지능화혁신인재양성(Grand ICT연구센터) 사업의 연구결과로 수행되었음 (IITP-2022-2015-0-00742)