일부를 알게 되면 나머지에 대한 예측이 바뀝니다. 키를 알면 몸무게의 분포가 좁아지고, 검사 결과를 알면 병에 걸렸을 확률이 바뀝니다. 이 장에서는 “알게 된 정보로 조건을 붙인” 분포와 기댓값을 다룹니다. 조건부 기댓값 E[Y∣X]는 “X를 보고 Y를 가장 잘 예측하는 함수”이고, 이것이 머신러닝의 회귀가 학습하려는 대상입니다.
5.1 조건부 분포
정의 5.1 (조건부 분포). P(X=x)>0이면 p(y∣x)=pX(x)p(x,y). 연속이면 f(y∣x)=fX(x)f(x,y) (fX(x)>0).
결합분포 표에서 X=x인 줄만 잘라 그 합이 1이 되게 나눈 것입니다. 연속에서 f(y∣x)는 결합밀도를 직선 X=x로 자른 단면을 정규화한 것입니다.
예제 5.2. 4장 문제 4.1의 표에서 P(Y=1∣X=1)=0.50.4=0.8이고 P(Y=1∣X=0)=0.50.2=0.4입니다. X를 알면 Y의 예측이 달라지므로 독립이 아닙니다.
5.2 연속 베이즈 정리
정리 5.3 (베이즈 정리).
f(θ∣x)=∫f(x∣θ′)f(θ′)dθ′f(x∣θ)f(θ).
이산수학 13강의 베이즈 정리에서 합을 적분으로 바꾼 것입니다. 분모는 θ와 상관없는 정규화 상수이므로 흔히 f(θ∣x)∝f(x∣θ)f(θ)로 씁니다. 10장 베이즈 추론의 기본 식입니다.
5.3 조건부 기댓값
정의 5.4. E[Y∣X=x]=∑yyp(y∣x) 또는 ∫yf(y∣x)dy. 이것을 x의 함수 g(x)로 보고 X를 넣은 확률변수 g(X)를 E[Y∣X]로 씁니다.
정리 5.5 (반복 기댓값의 법칙). E[E[Y∣X]]=E[Y].
증명 (연속). ∫(∫yf(y∣x)dy)fX(x)dx=∬yf(x,y)dydx=E[Y]. □
“경우를 나눠 평균을 내고, 그 평균들을 경우의 확률로 다시 평균 내면 전체 평균”입니다.
예제 5.6. 하루 손님 수 N∼Pois(20)이고 손님마다 평균 15000원을 쓰면, 하루 매출의 기댓값은 E[E[매출∣N]]=E[15000N]=300000원입니다.
정리 5.7 (전분산 공식). Var(Y)=E[Var(Y∣X)]+Var(E[Y∣X]).
전체 퍼짐 = 그룹 안의 퍼짐의 평균 + 그룹 평균들 사이의 퍼짐. 분산분석(ANOVA)과 R2(“설명된 분산”)가 이 식입니다.
5.4 최선의 예측
정리 5.8. 평균제곱오차 E[(Y−g(X))2]를 가장 작게 하는 함수는 g(X)=E[Y∣X]입니다.
증명 개요. E[(Y−g)2]=E[(Y−E[Y∣X])2]+E[(E[Y∣X]−g)2]로 나뉩니다(교차항의 기댓값이 0). 둘째 항은 g=E[Y∣X]일 때 0입니다. □
제곱오차로 학습한 회귀 모델은 조건부 평균을, 절댓값 오차로 학습한 모델은 조건부 중앙값을 근사하려 합니다. 손실 함수의 선택이 “무엇을 예측하는가”를 정한다는 뜻입니다.
예제 5.9 (정규분포). (X,Y)가 이변량 정규분포이면 E[Y∣X=x]=μY+ρσXσY(x−μX)로 x의 일차식이고, Var(Y∣X)=σY2(1−ρ2)로 x와 상관없습니다. 정규분포에서는 최선의 예측이 바로 선형회귀입니다. 또 ∣ρ∣<1이면 기울기의 절댓값이 σXσY보다 작아, 극단적인 x에서 예측이 평균 쪽으로 끌려옵니다(평균으로의 회귀).
5.5 이 장의 요약
- 조건부 분포 = 결합분포의 단면을 정규화. 베이즈: 사후 ∝ 가능도 × 사전.
- E[E[Y∣X]]=E[Y], 전분산 공식.
- 제곱오차를 최소로 하는 예측은 E[Y∣X].
쓰이는 곳. 회귀 모델은 E[Y∣X]를, 분류 모델은 P(Y∣X)를 근사합니다. 강화학습의 가치 함수 V(s)=E[보상의 합∣s]와 벨만 방정식은 반복 기댓값의 법칙이고, EM 알고리즘과 칼만 필터는 조건부 기댓값을 되풀이해 계산합니다.
문제
5.1 [계산] · 확인 문제
결합 확률이 p(0,0)=0.1, p(0,1)=0.3, p(1,0)=0.2, p(1,1)=0.4일 때 P(X=1∣Y=1)은? (분수로 답할 수 있습니다)
정답
풀이
P(Y=1)=0.7이므로 0.70.4=74입니다.
5.2 [계산] · 확인 문제
동전을 던져 앞면이면 주사위 하나, 뒷면이면 주사위 두 개를 던져 눈의 합을 Y라 할 때 E[Y]는?
정답
풀이
E[Y∣앞]=3.5, E[Y∣뒤]=7이므로 E[Y]=21(3.5+7)=5.25입니다.
5.3 [계산] · 확인 문제
이변량 정규분포에서 μX=170, σX=6, μY=65, σY=9, ρ=0.6일 때 E[Y∣X=182]는?
정답
풀이
65+0.6⋅69⋅(182−170)=65+0.9⋅12=75.8입니다.
5.4 [계산]
문제 5.3에서 Var(Y∣X)는?
정답
풀이
81(1−0.36)=51.84입니다. X를 알면 분산이 81에서 51.84로 줄어듭니다.
5.5 [증명]
전분산 공식 Var(Y)=E[Var(Y∣X)]+Var(E[Y∣X])를 유도합니다.
풀이
Var(Y∣X)=E[Y2∣X]−(E[Y∣X])2이므로 E[Var(Y∣X)]=E[Y2]−E[(E[Y∣X])2](반복 기댓값의 법칙). 또 Var(E[Y∣X])=E[(E[Y∣X])2]−(E[E[Y∣X]])2=E[(E[Y∣X])2]−(E[Y])2. 두 식을 더하면 E[Y2]−(E[Y])2=Var(Y)입니다.