신경망의 가중치 행렬 제약을 통한 최적화 방법
Modular Manifolds
신경망의 텐서 크기를 관리하기 위해 정규화가 필요하며, 이는 가중치 행렬에 덜 적용된다. EDM2 확산 모델은 가중치 제약을 사용하여 이점을 보고하였고, BiT는 가중치 표준화를 사용한다. 가중치 행렬을 정규화하면 최적화 업데이트의 상대 크기를 이해하기 쉬워지고, 하이퍼파라미터 조정 시 중요한 텐서에 집중할 수 있다. 본 포스트에서는 신경망의 가중치 행렬을 서브 매니폴드에 제약하는 방법을 제안하며, Muon 최적화 알고리즘의 매니폴드 버전을 제안한다.
가중치 행렬의 정규화를 통해 신경망의 학습 효율성을 높일 수 있는 방법을 제시한다.
원문 출처
Thinking Machines Lab Blog