개발공부

Transfer Learning 개념 본문

Python/Deep Learning

Transfer Learning 개념

mscha 2022. 6. 16. 17:32

Transfer Learning

 

전형적인 CNN 모델은 크게 Base Model과 Head Model로 나눌 수 있다.

Transfer Learning은 잘 만들어진 CNN model을 가져와서 사용하는 것이다.

 

보통 특징을 추출하는 잘만들어진 CNN 모델의 Base Model 부분은

이미 잘 학습된 상태이기 때문에 더는 학습하지 않도록 하는데

이를 Frozen이라고 하고, 이때 Base Model을 Frozen Layer라고 한다.

 

반대로 보통 가져온 CNN모델의 Head Model 부분은 사용하지 않고 

사용자가 생성한 ANN 모델을 Head Model로 사용하는데 이 Head Model은 

학습이 필요하기 때문에 Trainable Layers로 설정한다.