Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Relational Representation Distillation

Created by
  • Haebom

저자

Nikolaos Giakoumoglou, Tania Stathaki

개요

본 논문은 지식 증류(Knowledge Distillation)에서 기존의 Kullback-Leibler (KL) divergence 최소화 방식의 한계를 극복하기 위해 새로운 목적 함수를 제안합니다. KL divergence 방식은 teacher 모델의 내부 표현 구조를 제대로 반영하지 못하는 반면, 최근의 contrastive learning 방식은 유사한 샘플들까지 과도하게 분리시키는 문제가 있습니다. 본 논문에서는 teacher와 student 분포에 각각 다른 온도 매개변수를 사용하여, student 출력을 더 날카롭게 만들고 주요 관계는 정확하게 학습하면서 부차적인 유사성은 유지하는 방법을 제시합니다. 이 방법은 InfoNCE loss와 KL divergence와의 이론적 연관성을 가지며, 다양한 지식 전달 작업에서 기존 방법보다 우수한 성능을 보이며, teacher 모델과의 정렬을 개선하고 심지어 teacher 네트워크의 성능을 능가하는 경우도 있습니다.

시사점, 한계점

시사점:
기존 지식 증류 방법의 한계를 극복하는 새로운 목적 함수 제시
Teacher 모델의 내부 표현 구조를 더 잘 반영하는 지식 전달 가능
Teacher 모델 성능을 능가하는 Student 모델 학습 가능성 제시
InfoNCE loss와 KL divergence와의 이론적 연관성 규명
한계점:
제안된 방법의 온도 매개변수 최적화에 대한 자세한 설명 부족 (추정)
다양한 teacher-student 모델 구조에 대한 일반화 성능 검증 필요 (추정)
실험 결과의 일반화 가능성에 대한 추가 검토 필요 (추정)
👍