arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2209.04744cs.LGstat.ME

因果模型中最优干预设计的主动学习

Active Learning for Optimal Intervention Design in Causal Models

Massachusetts Institute of Technology(麻省理工学院);Broad Institute of MIT and Harvard(麻省理工学院和哈佛大学布罗德研究所);Harvard University(哈佛大学)

Jiaqi Zhang, Louis Cammarata, Chandler Squires, Themistoklis P. Sapsis, Caroline Uhler

AI总结:

本文提出融合贝叶斯因果模型与闭式因果采集函数的主动学习方法,以更少实验高效识别最优干预,并在合成数据和 Perturb-CITE-seq 单细胞数据上验证其优势。

摘要:

通过序贯实验设计来发现能够实现预期结果的干预措施,是科学、工程和公共政策等多个领域中的关键问题。当可能干预的空间很大、使得穷举搜索不可行时,就需要实验设计策略。在此背景下,编码变量之间的因果关系,以及由此得到的干预对系统的影响,对于更高效地识别理想干预至关重要。本文提出一种因果主动学习策略,用于识别最优干预;其最优性由干预后分布均值与期望目标均值之间的差异来衡量。该方法对因果模型采用贝叶斯更新,并使用精心设计的、具备因果信息的采集函数对干预进行优先级排序。该采集函数可以闭式计算,从而实现快速优化。所得算法具有信息论界提供的理论基础,并在因果图已知的线性因果模型中具备可证明的一致性结果。我们将该方法应用于合成数据以及来自 Perturb-CITE-seq 实验的单细胞转录组数据,以识别诱导特定细胞状态转变的最优扰动。具备因果信息的采集函数总体上优于现有准则,能够用更少但经过审慎选择的样本完成最优干预设计。

英文摘要:

Sequential experimental design to discover interventions that achieve a desired outcome is a key problem in various domains including science, engineering and public policy. When the space of possible interventions is large, making an exhaustive search infeasible, experimental design strategies are needed. In this context, encoding the causal relationships between the variables, and thus the effect of interventions on the system, is critical for identifying desirable interventions more efficiently. Here, we develop a causal active learning strategy to identify interventions that are optimal, as measured by the discrepancy between the post-interventional mean of the distribution and a desired target mean. The approach employs a Bayesian update for the causal model and prioritizes interventions using a carefully designed, causally informed acquisition function. This acquisition function is evaluated in closed form, allowing for fast optimization. The resulting algorithms are theoretically grounded with information-theoretic bounds and provable consistency results for linear causal models with known causal graph. We apply our approach to both synthetic data and single-cell transcriptomic data from Perturb-CITE-seq experiments to identify optimal perturbations that induce a specific cell state transition. The causally informed acquisition function generally outperforms existing criteria allowing for optimal intervention design with fewer but carefully selected samples.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑