핵심 개념
본 논문에서는 딥러닝 모델의 가중치 공간 대신 함수 공간에 직접 사전 확률을 적용하는 새로운 라플라스 근사 방법인 FSP-LAPLACE를 제안하여 해석 가능하고 효율적인 베이지안 딥러닝을 가능하게 합니다.
초록
FSP-LAPLACE: 베이지안 딥러닝에서 라플라스 근사를 위한 함수 공간 사전 확률
본 연구는 기존의 가중치 공간 기반 사전 확률의 한계를 극복하고자 딥 뉴럴 네트워크에 대한 해석 가능하고 효율적인 베이지안 추론 방법을 제시하는 것을 목표로 합니다. 특히, 가중치 공간 대신 함수 공간에 직접 가우시안 프로세스(GP) 사전 확률을 적용하여 라플라스 근사를 수행하는 새로운 방법론인 FSP-LAPLACE를 제안합니다.
FSP-LAPLACE는 크게 두 단계로 구성됩니다. 첫째, GP 사전 확률을 사용하여 함수 공간에서 정규화된 딥 뉴럴 네트워크의 목적 함수를 정의합니다. 이 목적 함수는 네트워크에 의해 표현될 수 있는 함수 공간에서 GP 사전 확률 하에서의 사후 확률 측정값의 약한 모드를 찾는 것과 동일합니다. 둘째, 목적 함수에 라플라스 근사를 적용하여 사후 확률 분포를 근사합니다. 이때, 행렬-Free 선형 대수학의 효율적인 방법을 활용하여 대규모 모델 및 데이터셋에 대한 확장성을 확보합니다.