위키득계정 ↗
기술

BootMAE

BootMAE기술
연구 공개
2022년
발표
ECCV 2022
목표 제공
모멘텀 인코더
복원 구성
목표 인지 디코더
BootMAE는 마스크 오토인코더에 모멘텀 인코더와 목표 인지 디코더를 더하는 자기지도 사전 학습 방법이다. 학습 중 얻은 특징을 예측 목표로 활용하고 인코더가 의미 표현에 집중하도록 구성한다.

자체 특징으로 목표 만들기

원래 MAE 인코더와 나란히 모멘텀 인코더를 둔다. 학습하는 모형 자신의 표현을 마스크 토큰의 추가 예측 목표로 사용해 사전 학습을 보완한다.

목표 특화 정보의 전달

가리지 않은 패치의 픽셀 값처럼 예측 목표에 필요한 정보를 디코더에 직접 전달한다. 인코더가 그 세부 정보를 모두 기억해야 하는 부담을 낮춰 의미 모델링에 역량을 쓰도록 한다.

여러 시각 과제의 평가

원 연구는 같은 사전 학습 반복 수의 MAE와 이미지 분류를 비교했다. 의미 분할과 물체 탐지 및 인스턴스 분할에서도 학습한 인코더의 표현을 평가했다.

관련 문서

마스크드 오토인코더 / CAE 문맥 오토인코더 / SimMIM / BEiT / I-JEPA와 V-JEPA / MSN 마스크 샴 신경망