위키득계정 ↗
기술

AdaHessian

AdaHessian기술
영문
AdaHessian
주요 정보
손실의 곡률
추정 방식
Hutchinson 기반 대각 근사
안정화
이동 평균과 블록 평균
AdaHessian은 손실 함수의 헤시안 대각을 근사해 갱신 크기에 곡률 정보를 반영하는 적응적 최적화 방법이다. 무작위 추정과 이동 평균 등을 이용해 큰 곡률 행렬을 직접 저장하는 부담을 줄인다.

곡률을 활용하기

기울기는 현재 위치에서 손실이 변하는 방향을 알려 준다. 곡률은 그 기울기가 주변에서 얼마나 달라지는지에 관한 정보다. AdaHessian은 이를 근사해 갱신을 조절하며 손실의 전체 헤시안 행렬을 매번 명시적으로 구성하지 않는다.

대각 추정의 독립 예제

H=[[4, 1], [1, 2]]라고 하자. 부호 벡터 (1, 1)에 대해 성분별 곱 z⊙Hz는 (5, 3)이고 (1, -1)에 대해서는 (3, 1)이다. 두 결과의 평균은 (4, 2)로 H의 대각과 같다. 이 작은 예에서의 일치는 한 번의 무작위 표본이 항상 정확하다는 뜻이 아니다.

변동을 줄이는 장치

무작위 곡률 추정은 표본과 학습 배치에 따라 흔들릴 수 있다. 대각 추정의 이동 평균과 블록 단위 평균은 이런 변동을 줄이는 데 쓰인다. 분산을 줄이면서 세부 정보가 완화될 수 있으므로 평균 범위도 학습 동작에 영향을 준다.

비용과 한계

헤시안과 벡터의 곱을 얻는 추가 미분 계산이 필요하다. 전체 행렬을 저장하지 않는다고 일차 방법과 비용이 항상 같아지는 것은 아니다. 같은 품질 수준에 도달하는 실제 시간과 메모리, 학습 안정성을 함께 비교한다.

관련 문서

Shampoo 옵티마이저 / SOAP 옵티마이저 / 경사 하강법 / 행렬 / 고유값 / Lion 옵티마이저