대화 데이터셋의 클래스 불균형 문제 보정을 위한 적대적 학습 기법

Adversarial Training Method for Handling Class Imbalance Problems in Dialog Datasets

  • 조수필 (한양대학교 컴퓨터.소프트웨어학과) ;
  • 최용석 (한양대학교 공과대학 컴퓨터공학부)
  • Cho, Su-Phil (Department of Computer Science, Hanyang University) ;
  • Choi, Yong Suk (Division of Computer Science and Engineering, Hanyang University)
  • 발행 : 2019.10.10

초록

딥러닝 기반 분류 모델에 있어 데이터의 클래스 불균형 문제는 소수 클래스의 분류 성능을 크게 저하시킨다. 본 논문에서는 앞서 언급한 클래스 불균형 문제를 보완하기 위한 방안으로 적대적 학습 기법을 제안한다. 적대적 학습 기법의 성능 향상 여부를 확인하기 위해 총 4종의 딥러닝 기반 분류 모델을 정의하였으며, 해당 모델 간 분류 성능을 비교하였다. 실험 결과, 대화 데이터셋을 이용한 모델 학습 시 적대적 학습 기법을 적용할 경우 다수 클래스의 분류 성능은 유지하면서 동시에 소수 클래스의 분류 성능을 크게 향상시킬 수 있음을 확인하였다.

키워드

과제정보

본 연구는 2019년도 산업통상자원부 및 산업기술평가관리원(KEIT) 연구비 지원에 의한 연구이며(과제번호:10077553), 산업통상자원부의 재원으로 기술혁신사업의 지원을 받아 수행한 연구 과제 (No. 10060086, 개인 서비스용 로봇을 위한 지능-지식 집약·개방·진화형 로봇지능 소프트웨어 프레임워크 기술 개발)입니다