bayesflow-org/bayesflow

A Python library for simulation-based inference with deep learning

해결하는 문제

BayesFlow는 시뮬레이션 기반 추론(SBI) 프로세스를 단순화합니다. 연구자들은 복잡한 과학 시뮬레이션에서 종종 사용할 수 없는 우도 함수를 수동으로 유도할 필요 없이 데이터에서 매개변수를 추정하거나 다른 모델을 비교하는 것과 같은 복잡한 베이지안 분석을 수행할 수 있습니다.

작동 방식

딥러닝을 사용하여 추론 프로세스를 "상각"합니다. 새 데이터 세트마다 느린 추론 프로세스를 실행하는 대신 시뮬레이션 매개변수와 결과 데이터 간의 관계를 학습하는 신경망을 훈련합니다. 훈련이 완료되면 이 네트워크는 거의 즉각적인 추정을 제공할 수 있습니다. Keras 3를 활용하여 여러 백엔드를 지원하므로 JAX, PyTorch 또는 TensorFlow를 사용하여 모델을 실행할 수 있습니다.

대상 사용자

복잡한 시뮬레이터 또는 매개변수 모델을 사용하는 연구자와 데이터 과학자를 위해 설계되었으며, 베이지안 추론, 모델 비교 또는 실험 설계를 수행해야 합니다. 현대 생성 AI 기술을 과학 워크플로우에 통합하려는 사람들에게 특히 유용합니다.

주요 기능

  • Keras 3를 통한 멀티 백엔드 지원(JAX, PyTorch 또는 TensorFlow).
  • 확산 모델 및 일관성 모델을 포함한 고급 생성 모델을 지원합니다.
  • 추론 결과를 검증하고 이해하기 위한 내장 진단 도구를 포함합니다.
  • 최소한의 코드로 워크플로우를 정의하기 위한 고수준 API를 제공합니다.
  • 유연한 다운스트림 분석을 위해 PyMC 생태계와 통합됩니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트