소규모 문서 타입 분류 딥러닝 모델 추천

소규모 문서 타입 분류(Classification)를 할 때 어떤 백본 모델(ResNet, EfficientNet 등)이 성능이 좋은가요?

1개의 답변이 있어요!

  • 안녕하세요. 조원우 정보처리기사입니다.

     일반적으로 ResNet은 구조가 단순하고 학습이 안정적이어서 데이터가 적은 환경에서도 과적합 위험이 비교적 낮아 기본 모델로 적합합니다. 

     반면 EfficientNet은 파라미터 효율이 높고 성능이 우수하지만, 데이터가 부족한 경우에는 과적합이 발생할 수 있어 주의가 필요하며, 사전학습된 모델을 활용한 전이학습 환경에서 특히 강점을 보입니다.

     데이터 규모가 작다면 ResNet 계열을 우선적으로 고려하는 것이 적절하며, 전이학습을 활용할 경우에는 EfficientNet도 좋은 선택이 될 수 있습니다. 

     또한 문서 분류가 텍스트 기반이라면 CNN 모델보다 BERT와 같은 자연어 처리 모델을 사용하는 것이 더 높은 성능을 기대할 수 있습니다