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개의 댓글

관련 채용 정보