train loss와 validation loss 그래프로 모델 학습 상태 확인하기

train loss와 validation loss 그래프로 모델 학습 상태 확인하기

파란 선이 train loss, 빨간 선이 validation loss이다.

(출처: 모두의연구소)

A

train loss와 validation loss가 둘다 감소하지 않음

학습이 잘 안되고 있는 상태

 

B

train loss는 계속해서 감소하지만 validation loss는 약간 감소하다가 증가하고 있다

overfitting 상태이므로 모델을 수정해서 overfitting을 처리해야 함

 

C(good)

train loss와 validation loss가 둘다 감소하는 상태

train loss와 validation loss가 유사하게 감소하여 validation data가 유출됐다고 생각할 수도 있으나 X

학습이 잘 진행되는 중

 

D(good)

train loss와 validation loss가 약간의 차이를 두고 둘다 감소하는 상태

학습이 잘 진행되는 중