모델 학습 파이프라인 & 순전파·역전파 시각화
모델 학습 전체 파이프라인
순전파·역전파에 앞서 반드시 이해해야 할 7단계 학습 절차
| # |
단계 |
내용 |
역할 |
PyTorch |
| 1 |
EDA |
데이터 파악 & 전처리 |
데이터 품질 확보 및 분포 파악 |
pandas
matplotlib
|
| 2 |
벡터 변환 |
데이터를 숫자(벡터)로 임베딩 |
모델 입력 형식으로 변환 |
torch.tensor()
DataLoader
|
| 3 |
방정식 도출 |
가중치(W)로 예측값 계산 구조 설계 |
모델 아키텍처 정의 |
nn.Module
|
| 4 |
순전파 |
입력 x → 예측값 ŷ 계산 |
현재 가중치로 예측 수행 |
pred = model(x)
|
| 5 |
LOSS 구성 |
예측값 vs 정답 오차 계산 |
오차 정량화 |
nn.MSELoss()
loss = criterion(pred, y)
|
| 6 |
미분(역전파) |
LOSS를 각 W로 편미분 |
방향 + 기울기 크기 파악 |
optimizer.zero_grad()
loss.backward()
|
| 7 |
경사하강 |
W = W − lr × 기울기 |
lr로 보폭 결정 후 W 업데이트 |
optimizer = optim.SGD(lr=0.01)
optimizer.step()
|
⟳
4~7단계를 반복 — 충분히 수렴할 때까지 순전파 → LOSS → 역전파 → 가중치 업데이트 사이클을 반복 수행합니다.
| 주의: optimizer.zero_grad()는 step 6 시작 시 반드시 호출해야 기울기 누적을 방지할 수 있습니다.
INTERACTIVE VISUALIZATION
버튼을 눌러 시뮬레이션을 시작하세요 ✨