Focal Loss

HJW·2022년 8월 19일
0


reshaped cross entropy loss로서 class imbalance 문제를 해결하기 위하여 도입되었다.

detect 하기 어려운 class에 더 큰 가중치를 할당하는 방식이다


binary calssification을 위한 cross entropy

pt를 다음과 같이 정의하면,

Binary Cross Entroy 정의

Focal Loss 정의

RetinaNet의 loss function으로 사용됨

profile
be passionate

0개의 댓글