연구성과

학술발표
논문명 Few-shot 이미지 분류를 위한 프로토타입 노드 기반 그래프 신경망
개최일 2021.06.25
학술회의명 한국정보과학회 2021 한국소프트웨어종합학술대회 (KCC2021)
책임교수
구분 구두발표
제1저자 장성은
교신저자 김준태
공동저자 김준태
국내/국외 국내
개최국가 대한민국
주관기관

딥러닝 모델의 학습에는 일반적으로 많은 양의 데이터를 필요로 하나, 실제로는 극히 일부의 도메인을 제외하고는 다수의 도메인에서 양질의 데이터를 획득하기 어렵다. 최근에는 그래프 신경망 기반의 few-shot 학습을 통해 적은 수의 데이터만으로도 학습이 가능하도록 하는 연구가 활발하게 이루어지고 있다. 본 연구에서는 few-shot 태스크 그래프에 프로토타입 노드와 엣지 레이블을 적용하여, 보다 명시적으로 클래스 및 데이터 간 유사도를 모델링하는 방법론을 제안한다. 제안하는 모델은 프로토타입-쿼리 엣지 레이블과 프로토타입 노드와 쿼리 노드 사이의 거리를 동시에 고려하여, 입력되는 태스크 그래프에 따라 다르게 생성되는 일반화된 프로토타입 노드를 제공하고 few-shot 이미지 분류를 수행한다. 모델의 5-way 5-shot 성능은 엣지 레이블 기반의 few-shot 이미지 분류 모델인 EGNN 모델과 비교한 결과, 제안하는 모델에서 의미 있는 성능 향상을 확인하였다 

04620 서울특별시 중구 필동로1길 30 동국대학교 Knowledge Science 연구센터(KSRC) Tel.02-2290-1441
Copyright© 2021 DONGGUK UNIVERSITY. ALL RIGHTS RESERVED.

×