Conditional diffusions for neural posterior estimation
Created by
Haebom
저자
Tianyu Chen, Vansh Bansal, James G. Scott
개요
본 논문은 베이지안 추론을 위한 시뮬레이션 기반 계산 방법인 신경 후방 추정(NPE)에서, 기존의 정규화 흐름(normalizing flows) 기반 방법들의 한계(훈련 불안정성, 표현력과 계산 비용 간의 상충 관계)를 극복하기 위해 조건부 확산 모델(conditional diffusions)과 고용량 요약 네트워크(high-capacity summary networks)를 결합한 새로운 NPE 방법을 제시합니다. 다양한 NPE 벤치마킹 문제에 대한 실험 결과, 제안된 방법은 기존의 흐름 기반 방법보다 안정성, 정확도, 훈련 속도 면에서 우수한 성능을 보였으며, 더 단순하고 얕은 모델을 사용하더라도 성능 저하가 적었습니다. 다양한 요약 네트워크 아키텍처에서도 이러한 성능 향상이 지속됨을 확인했습니다. 소스 코드는 https://github.com/TianyuCodings/cDiff 에서 확인할 수 있습니다.