1. 2D CNN 연산 복잡도 공식
총 연산량 (Multiply–Accumulate operations):
O(Hout⋅Wout⋅Cin⋅KH⋅KW⋅Cout)
여기서:
기호 | 의미 |
---|
Hout,Wout | 출력 feature map의 height, width |
Cin | 입력 채널 수 (예: RGB 이미지면 3) |
KH,KW | 커널의 높이와 너비 (예: 3x3이면 3, 3) |
Cout | 출력 채널 수 (필터 수) |
2. 예시
예를 들어:
- 입력 이미지: 128x128, RGB(3채널)
- 커널: 3x3
- 출력 채널 수: 64
- stride=1, padding=1 (출력 크기 유지됨 → H_out = H, W_out = W)
그럼 연산량은:
128×128×3×3×3
=O(1282×3×32×64)
=O(1282×1728)
→ 즉, 약 2억 8천만 개의 곱셈-덧셈 연산이 일어납니다.
3. 간단한 형태로 요약
대개는 입력/출력 크기를 일정하게 유지한다고 가정하고 다음과 같이 단순화합니다:
O(H×W×Cin×K2×Cout)
즉,
공간 크기 × 채널 수 × 커널 크기 × 출력 채널 수
로 보시면 됩니다.