AI4Meder
返回论文列表
论文ICLR 2026 Poster2026 年clinical prediction

面向因果推断的基础模型:基于先验数据拟合网络

ICLR 2026 Poster accepted paper at ICLR 2026. Prior-data fitted networks (PFNs) have recently been proposed as a promising way to train tabular foundation models. PFNs are transformers that are pre-trained on synthetic data generated from a prespecified prior distribution and that enable Bayesian inference through in-context learning. In this paper, we introduce CausalFM, a comprehensive framework for training PFN-based foundation models in various causal inference settings. First, we formalize the construction of Bayesian priors for causal inference based on structural causal models (SCMs) in a principled way and derive necessary criteria for the validity of such priors. Building on this, we propose a novel family of prior distributions using causality-inspired Bayesian neural networks that enable CausalFM to perform Bayesian causal inference in various settings, including for back-door, front-door, and instrumental variable adjustment.

论文默认配图 - EHR 与临床预测

论文详情

英文标题
Foundation Models for Causal Inference via Prior-Data Fitted Networks
作者
Yuchen Ma, Dennis Frauen, Emil Javurek, Stefan Feuerriegel
期刊/会议
ICLR 2026 Poster
发表年份
2026 年
研究方向
clinical prediction