CutMix 쓰는 이유

OpenJR·2023년 8월 28일

CutMix 이름 부터가 Regularization Strategy to Train Strong Classifiers with Localizable Features이다.
즉, CutMix는 모델이 Localizable Feature에 대해서 학습할 수 있도록 Regularization해주는 기법이다.

Localizable Feature

정확한 단어는 아니지만, 이렇게 이해하였다.
고양이 그림에서 고양이라고 예측하기 위해선 고양이의 얼굴부터, 몸, 발 등 다양한 부분을 볼 수 있다. 하지만 이러한 부분 부분 중에서 고양이를 고양이라고 분류하기 위해 가장 중요한 정보는 아마 얼굴일 것이다.
모델을 학습과정에서 위와 같은 정보를 학습해 버려 고양이 그림에서 얼굴을 가려버리면 멍청해져서 분류를 못한다.
Cutmix는 고양이의 얼굴뿐만 다른 부위들도 같이 고려해서 고양이라고 예측할 수 있도록 규제화하는 기법이다.

profile
Jacob

0개의 댓글