DOI QR코드

DOI QR Code

자기지도 학습에서 와서스타인 (Wasserstein) 거리의 손실함수로의 이용가능성 연구

A Research on Using Wasserstein Distance as a Loss Function in Self-Supervised Learning

  • 구인화 (한양대학교 인공지능학과) ;
  • 채동규 (한양대학교 인공지능학과)
  • Koo, Inhwa (Dept. of Artificial Intelligence, Hanyang University) ;
  • Chae, Dong-Kyu (Dept. of Artificial Intelligence, Hanyang University)
  • 발행 : 2022.11.21

초록

딥러닝의 높은 예측 정확도를 위해서는 많은 양의 학습 데이터가 필요하다. 그러나 실세계에서 많은 양의 레이블이 붙은 데이터를 구하는 것은 어렵고 많은 비용이 든다. 때문에 레이블이 없이도 양질의 표현 학습이 가능한 자기지도학습이 각광을 받고 있다. 와서스타인 거리는 생성모델에도 쓰이지만 의사 레이블 (pseudo label) 을 만들어 레이블이 없는 데이터들을 분류 하는데도 좋은 성능을 보이고 있다. 따라서. 본 연구는 와서스타인 거리를 자기지도학습에 접목시키는 방법을 제안한다. 실험을 통해 연구의 가능성을 보인다.

키워드

과제정보

이 논문은 2022 년도 정부(과학기술정보통신부)의 재원으로 (1) 한국연구재단 바이오. 의료기술개발사업의 지원 (No. NRF-2021M3E5D2A01021156)과 (2) 정보통신기획평가원의 지원을 받아 수행된 연구임 (No.2020-0-01373, 인공지능대학원지원(한양대학교))