[TIL] Test Time Augmentation (TTA)

박상우·2023년 2월 2일
0

TIL

목록 보기
9/12
post-thumbnail

Test Time Augmentation

일반적으로 데이터 증강은 model의 robust함을 상승시키기 위해 training data에 진행되는 작업

이떄 TTA는 test 단계에서 augmentation을 진행 해 이미지의 예측값을 평균내어 최종 예측을 하는 작업

Why?

  • 일종의 ensemble 기법에 해당
  • image task에서 편향된 데이터가 존재할 때 여러 augmentation 기법을 적용해 평균을 내게 되면 일반적으로 높은 성능을 보임
  • Kaggle 등 competition에서 사용하면 좋은 기법
import ttach as tta

다음 라이브러리에서 사용 가능

다만 Time Cost가 더 소모됨으로, 적절히 사용하는 것이 중요

profile
세상아 덤벼라

0개의 댓글