테마
10회 · 숨은 변수와 EM · 노트
3 EM 알고리즘
3.1 알고리즘과 단조성
알고리즘 3.1 (EM (Dempster–Laird–Rubin, 1977)) 시작값 에서 에 대해 다음을 되풀이한다.
- E
지금의 추정값으로 숨은 자료의 조건부분포 를 구하고 를 만든다.
- M
.
이 충분히 작아지면 멈춘다.
완전 자료 모형이 지수족이면 는 충분통계량에 대해 선형이므로, E 단계는 “충분통계량의 조건부 기댓값을 구하기”이고 M 단계는 “그 기댓값을 완전 자료의 MLE 공식에 넣기”다. 유전 연관 자료에서는 E 단계가 , M 단계가 이다(연습 2.2). 한 바퀴의 그림이 그림 3.1이다.
정리 3.2 (EM은 가능도를 줄이지 않는다) EM의 반복은 를 만족한다. 등호는 이고 일 때뿐이다.
증명 전략 보조정리 2.4로 가능도의 증가를 의 증가와 의 변화로 나눈다. 는 M 단계가 늘리고, 는 Gibbs 부등식으로 제자리에서 가장 크다.
보조정리 2.4를 로 두고 과 에 쓴 두 식을 빼면
보조정리 2.4첫 괄호는 0 이상이다. 은 의 최대점이다.M 단계
둘째 괄호를 기댓값 하나로 쓴다.
와 의 정의이므로 둘째 괄호는 0 이하이다.Gibbs 부등식, 2회 정리 8.3
따라서 이다.1 2 4
등호는 두 괄호가 모두 0일 때, 곧 이고 일 때뿐이다.Gibbs의 등호 조건
조건 확인 보조정리 2.4의 첫 등식은 인 곳에서 성립한다. 인 곳에서 이어야 1의 평균을 취할 수 있다. 숨은 자료의 받침이 모수에 의존하지 않으면 성립한다. 와 는 유한하다고 가정한다. 3의 계산은 9회 보조정리 1.4와 같다.
증명에서 M 단계가 한 일은 를 늘린 것뿐이다. 그래서 를 완전히 최대화하지 않고 늘리기만 해도(예: Newton 한 걸음) 단조성은 그대로다. 이것을 일반화 EM(GEM)이라 한다.
3.2 하한을 오르는 알고리즘
명제 3.3 (증거 하한 (ELBO)) 위의 아무 확률밀도 에 대해
따라서 이고, 등호는 일 때다.
증명 전략 완전 밀도를 로 쓰고 아무 분포 로 평균한 뒤, 양변에서 를 뺀다. 남는 항이 Kullback–Leibler 발산이다. 단계별 증명은 연습 3.5(유도형)에서 직접 쓴다. 완성된 증명은 그 연습의 풀이에 있다.
이 명제는 EM을 한 함수 를 번갈아 최대화하는 좌표 상승으로 보여 준다(Neal–Hinton, 1998). E 단계는 를 고정하고 에 대해 최대화한다. 답은 이고, 하한이 그 점에서 가능도에 닿는다. M 단계는 를 고정하고 에 대해 최대화한다. 는 에 의존하지 않으므로 이것은 의 최대화다. 가능도 아래에 그 점에서 닿는 하한을 깔고, 하한의 꼭대기로 옮기고, 그 점에서 다시 하한을 까는 것이다(그림 3.2). 8절에서 를 신경망으로 근사하면 변분 오토인코더가 된다.
따름정리 3.4 (EM의 고정점은 정류점이다) (내점)이고 가 두 인수에 대해 연속미분가능하면 이다.
증명 전략 M 단계의 1계 조건에서 로 극한을 취하고, 명제 2.5로 그 극한을 관측 점수로 읽는다. 단계별 증명은 연습 3.6(유도형)에서 직접 쓴다. 완성된 증명은 그 연습의 풀이에 있다.
가정 주의 정류점은 극대점이 아닐 수도 있다. EM은 가능도를 늘리기만 하므로 극소점에 머무는 일은 드물지만(9회 4절의 Newton과 다르다), 국소 최대점이나 안장점에 멈출 수 있다. 원논문의 수렴 증명에는 틀린 단계가 있었고, Wu(1983)가 정확한 조건을 정리했다. 실전에서는 여러 시작점에서 돌려 가능도가 가장 높은 것을 고른다.
3.3 잃어버린 정보와 수렴 속도
숨은 자료를 보지 못해서 정보를 얼마나 잃었는가? 완전 자료의 정보에서 관측 자료의 정보를 뺀 것이다. 이 직관을 정확히 쓴 것이 다음 정리다.
정리 3.5 (잃어버린 정보의 원리 (Orchard–Woodbury 1972, Louis 1982)) 관측 정보 는
로 쓰인다. 여기서 는 완전 자료 정보의 조건부 기댓값이고,
는 잃어버린 정보다(둘째 등식이 Louis 공식).
증명 전략 보조정리 2.4의 첫 등식 를 두 번 미분하고 숨은 자료의 조건부분포로 평균한다. 둘째 등식은 가 의 밀도라는 것과 정리 1.2의 두 표현이다.
모든 에서 이다.보조정리 2.4
두 번 미분하면 이다.1
왼쪽은 에 의존하지 않으므로 로 평균해도 그대로다. 따라서 이다., 의 정의
이고 둘째 항은 에 대해 상수다.1을 한 번 미분
상수를 빼도 공분산은 같으므로 이다.4 5
조건 확인 4에서 을 적분 기호 아래에서 두 번 미분한다 [T6]. 정리 1.2의 정칙조건이 숨은 자료의 조건부 모형에서도 성립한다고 가정한다.
정리 3.6 (EM의 수렴 속도 (DLR 1977)) (내점, )이고 가 두 번 연속미분가능하며 이면
의 고유값은 모두 에 있다. 모수가 하나이면 는 잃어버린 정보의 비율이다.
증명 전략 M 단계의 1계 조건 을 고정점 주위에서 일차 전개하고, 두 미분 행렬을 와 으로 읽는다. , 는 의 첫 인수와 둘째 인수에 대한 미분이고, 아래 미분은 모두 에서 잰다.
M 단계의 조건을 일차 전개한다. 로 쓰면
, 1, [T8]이다.의 정의, [T6]
보조정리 2.4로 이므로
보조정리 2.44를 로 미분하고 로 두면, 이므로 이다.정리 3.5의 4
2에 3 5를 넣으면 이다. 가 가역이므로 이고, 나머지항은 이다. 곧 이다.
이다. 은 대칭행렬 와 닮음이므로 고유값이 실수이고, 이므로 0 이상이다. 은 과 닮음이므로 고유값이 1보다 작다.정리 3.5,
조건 확인 가 근처에서 두 번 연속미분가능해야 2의 전개가 성립한다. 3과 5에서는 미분과 조건부 기댓값의 순서를 바꾼다 [T6].
정리는 EM의 성격을 한 문장으로 말한다. 숨은 자료가 정보를 많이 가져갈수록 EM은 느리다. Newton은 이차 수렴하지만(9회 연습 4.2), EM은 선형 수렴이고 그 비율이 잃어버린 정보의 비율이다. 대신 EM은 헤세 행렬을 구하지 않고, 걸음마다 가능도가 늘며, 모수의 제약(확률의 합이 1, 분산이 양수)을 M 단계가 저절로 지킨다.
예제 3.7 (유전 연관 자료의 EM) 에서 EM은 로 간다. 오차의 비 는 로 모인다. 정리 3.5로 계산하면 에서 관측 정보 , 완전 자료 정보 , 잃어버린 정보 이고, 이 정확히 그 비다(DLR의 표 1). 은 Louis 공식으로 이다(연습 3.2). 의 표준오차는 이다.
연습문제
연습 3.1 (는 제자리에서 최대 ★) 를 Jensen 부등식으로 직접 보여라.
풀이
(Jensen, 는 오목).
연습 3.2 (유전 연관의 수렴 속도 ★★) 유전 연관 자료에서 EM 사상 의 에서의 미분계수가 임을 확인하라. Louis 공식 도 확인하라.
풀이
EM 사상 , 에서 , . 에서 , 이므로 . Louis: , , . 이고 .
연습 3.3 (중도절단의 EM ★★★) 연습 2.3에서 EM 반복이 임을 보여라. 고정점이 MLE 이고, 수렴 비율이 절단된 비율 임을 보여라. 이것이 와 같음을 확인하라.
풀이
E 단계: 이므로 완전 자료 합의 기댓값은 . M 단계는 완전 자료 MLE(합)이다. 고정점 를 풀면 . 사상의 미분이 이다. 에서 , 이므로 .
연습 3.4 (일반화 EM ★) M 단계 대신 인 아무 을 고르면 정리 3.2의 결론이 그대로임을 보여라.
풀이
정리 3.2의 증명에서 M 단계가 쓰인 곳은 첫 괄호가 0 이상이라는 것뿐이다.
연습 3.5 (증거 하한의 증명 ★ 유도형) 명제 3.3를 다음 순서로 증명하라.
- (a)
모든 에서 임을 보이고, 양변을 로 평균하라.
- (b)
양변에서 를 빼서 를 얻어라.
- (c)
Gibbs 부등식(2회 정리 8.3)으로 부등식과 등호 조건을 얻어라.
- (d)
(c)에서 를 고정하고 에 대해 를 최대화하면 답이 무엇인지 적어라. 이것이 E 단계다.
풀이
(a) 에 로그를 취한다. 평균하면 이다. 은 에 의존하지 않으므로 평균해도 그대로다. (b) 양변에서 를 빼면 . (c) Gibbs 부등식으로 이고 등호는 일 때뿐이다. 따라서 이고 등호 조건도 같다. (d) 를 고정하면 는 상수이므로 의 최대화는 의 최소화이고, 답은 다. 숨은 자료의 조건부분포를 구하는 것이 E 단계다.
연습 3.6 (고정점은 정류점: 증명 ★★ 유도형) 따름정리 3.4를 다음 순서로 증명하라.