위키득계정 ↗
기술

ProtoTree

ProtoTree기술
첫 공개
2020년
발표
CVPR 2021
영문 이름
Neural Prototype Tree
결합 구성
원형 학습과 결정 트리
ProtoTree는 학습 가능한 대표 부분을 결정 트리의 노드에 넣은 이미지 분류 방법이다. 이미지에서 해당 부분의 존재 여부에 따라 분기하며 하나의 예측을 트리 경로로 설명한다.

원형을 검사하는 노드

이진 트리의 각 노드는 학습 가능한 원형 부분을 포함한다. 입력 영상에서 그 부분이 있는지 없는지에 따라 다음 경로를 정해 분류한다.

전체 구조와 개별 예측

트리 구조는 모델의 전체 판단 방식을 살펴보는 데 사용된다. 특정 이미지의 예측은 통과한 결정 경로를 보여주어 국소적으로 설명할 수 있다.

정확도와 간결성

가지치기와 이진화, 앙상블로 정확도와 설명의 간결성 사이의 절충을 조절했다. 원형 기반의 근거를 유지하면서 불필요한 분기를 줄이려는 접근이다.

세밀한 분류의 평가

CUB-200-2011과 Stanford Cars에서 평가했다. 단일 트리와 여러 트리를 결합한 구성을 통해 원형 부분을 사용한 결정 경로와 분류 정확도를 조사했다.

관련 문서

레이블 없는 개념 병목 모델 / ProtoPNet / 개념 병목 모델 / ACE 개념 설명 / TCAV / ROAR 설명 평가