[CNN] 가중치 초기화의 이해

Updated:

가중치 초기화

좋은 가중치 초기화 조건

  1. 값이 동일 해서는 안된다. (선형적으로 증가 혹은 감소하기만 한다.)
  2. 충분히 작아야 한다. (Weight sum하기 때문에 충분히 작아야 한다.)
  3. 적당한 분산(또는 표준편차)를 가져야 한다.

Xavier Glorot Initialization

입력 노드와 출력 노드의 갯수를 감안하여 동적으로 Weight 초기화 수행.

glorot_uniform

한도값 = $\sqrt \frac{6}{fan \ in + fan \ out}$

glorot_normal

표준편차 = $\sqrt \frac{2}{fan \ in + fan \ out}$

He Initialization

Relu에 보다 최적화된 가중치 초기화

he_uniform

한도값 = $\sqrt \frac {6}{fan \ in}$

he_normal

표준편차= $\sqrt \frac {2}{fan \ in}$

참고 자료

딥러닝 CNN 완벽가이드

Categories:

Updated:

Leave a comment