CausalPFN

用上下文学习,秒级估算因果效应,告别复杂建模

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 16 🐛 待处理 2 👀 关注 3
GitHub ↗
GitHub 星标 ★ 137
维护状态 维护中
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0许可证,可免费使用和部署。

项目预览

CausalPFN 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队vdblm
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0许可证,可免费使用和部署。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Jupyter Notebook
技术栈/模型causal-inference,causalpfn,foundation-models,machine-learning
GitHub 星标★ 137
上线时间2025-06-09 00:00:00
代码更新2026-09-27
仓库体积45.8 MB
本站数据更新2026-11-09
维护状态维护中
收录时间2026-08-09
浏览次数5

项目介绍

CausalPFN 是一个基于上下文学习(In-Context Learning)的因果效应估计开源模型,旨在解决传统因果推断方法计算成本高、需要大量手工建模的问题。它借鉴了 Prior-Fitted Networks (PFN) 的思想,通过在海量合成因果数据上预训练一个 Transformer 模型,使其能够在新任务中仅通过少量样本(即上下文)快速输出平均处理效应(ATE)等因果估计结果,而无需重新训练或复杂的贝叶斯推断。核心能力包括:支持多种因果结构(如混杂、工具变量等)、提供不确定性量化、以及高效的摊销推断。项目以 Jupyter Notebook 形式提供代码和实验,适合研究因果推断与元学习交叉领域的学者快速上手。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 137

核心亮点

  • 预训练模型实现摊销推断,新任务无需重训,大幅降低计算成本
  • 通过合成数据生成覆盖多种因果场景,提升模型泛化能力
  • 提供不确定性估计,帮助判断因果效应结果的可靠性

不足之处

  • 项目处于早期阶段,文档和API尚不完善,上手门槛较高
  • 依赖合成数据训练,真实场景下的表现需进一步验证

适用场景

  • 学术研究中快速探索因果效应假设
  • 小样本观测数据下的因果效应初筛
  • 作为基线对比传统因果推断方法

替代项目

DoWhy、EconML、CausalML

上一篇:ts-icl

下一篇:get_model

同类项目推荐

LoongSage 开源

用验证过的配方,快速训出会调工具的智能体

A production-grade, high-efficiency Agentic RL framework for frontier LLMs with vali···

★ 600 2026-10-07
reasoning-gym 开源

给AI推理训练提供可验证奖励的现成环境,直接上手RL

[NeurIPS 2025 Spotlight] Reasoning Environments for Reinforcement Learning with Veri···

★ 1548 2026-08-09
e4t-diffusion 开源

输入一张图,秒级生成个性化图像,免去繁琐微调

Implementation of Encoder-based Domain Tuning for Fast Personalization of Text-to-Im···

★ 322 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 464 2026-08-09
PERSIA 开源

多机并行训练大规模推荐模型

High performance distributed framework for training deep learning recommendation mod···

★ 410 2026-10-31
brewery-ai 开源

零经验也能在控制台微调 AI 模型

An intuitive console agent to fine-tune AI models with little to no experience requi···

★ 213 2026-10-09
LyCORIS 开源

超越 LoRA 的微调工具箱,让 Stable Diffusion 定制更灵活高效

Lora beYond Conventional methods, Other Rank adaptation Implementations for Stable d···

★ 2539 2026-08-09
transformerlab-app 开源

一站式搞定大模型训练、微调和评估,从笔记本到集群无缝扩展

The open source research environment for AI researchers to seamlessly train, evaluat···

★ 5200 2026-08-10
DiffusionART 开源

给扩散模型做稳定高效的强化学习后训练

Official Implementation of "DiffusionART: Efficient, Stable and Scalable Reinforceme···

★ 227 2026-10-21