bayesflow-org/bayesflow

해결하는 문제

BayesFlow은 딥러닝을 활용한 베이지안 추론을 간편하게 수행할 수 있도록 도와줍니다. 복잡한 시뮬레이터나 전통적인 파라미터 모델에서 생성된 데이터로부터 파라미터를 추정하는 과정을 단순화하여, 기존 베이지안 방법의 느린 반복적 프로세스를 피하고 '아모르타이즈드(amortized)' 워크플로우를 사용할 수 있게 합니다.

작동 방식

이 라이브러리는 신경망을 사용하여 관측값(데이터)과 파라미터 간의 관계를 학습합니다. 사후확률, 가능도, 비율, 포인트 추정기 등 다양한 신경망 추정기를 지원하며, 확산 모델과 일관성 모델과 같은 최신 생성형 AI를 활용해 사후 분포를 근사합니다. Keras 3 기반으로 구축되어 있어 JAX, PyTorch, TensorFlow 등 다양한 백엔드를 지원합니다.

대상 사용자

감염병 모델링(SIR), 인지 모델링, 공간 또는 이미지 데이터와 같은 복잡한 시뮬레이션 기반 데이터에 대해 베이지안 추론이 필요한 연구자 및 데이터 과학자에게 적합합니다.

주요 특징

• 다중 백엔드 지원: Keras 3를 통해 JAX, PyTorch, TensorFlow와 원활하게 통합 가능.
• 아모르타이즈드 워크플로우: 네트워크 학습이 완료되면 빠른 추론이 가능하며, 동일한 모델을 여러 데이터셋에 재사용할 수 있음.
• 생성형 AI 통합: 확산 모델과 일관성 모델을 포함한 풍부한 생성 모델을 제공.
• 크로스오버 기능: PyMC와 연동하여 가능도 추정 및 MCMC 샘플링 가능.
• 내장 진단 도구: 추론 과정의 결과를 검증하고 이해할 수 있도록 도와주는 도구 세트 제공.