arXiv:2504.18522stat.MLcs.LG2025-04被引 1

通过隐空间线性叠加建模基因敲除效应,实现对未见组合的精准预测。

Extrapolation Guarantees for Perturbation Modeling Under the Additive Latent Shift Assumption

  • 假设扰动在隐空间中可加叠加,构建潜变量模型描述扰动机制。
  • 理论上证明:充分多样的训练数据下,可准确识别扰动效应并外推至线性组合的新扰动。
  • 提出PDAE模型,通过分布相似性训练,实现在单细胞数据上对新组合扰动的高精度预测。

我们研究基因敲除等扰动对单细胞RNA测序数据的影响建模问题。给定部分扰动的数据,目标是预测新扰动组合下的测量分布。为此,我们假设扰动在未知但合适的嵌入空间中呈加性作用,将数据生成过程建模为潜变量模型,其中扰动表现为隐空间中的均值偏移,且可线性叠加。我们证明,在训练扰动足够多样时,表示和扰动效应可被识别,仅受正交变换限制,并由此导出对可表为已见扰动线性组合的未见扰动的外推保证。为从数据中估计该模型,我们提出扰动分布自编码器(PDAE),通过最大化真实与模拟扰动分布间的分布相似性进行训练。训练后的模型可用于预测此前未见的扰动分布。仿真结果支持理论分析,表明PDAE能准确预测可识别的未见扰动;同时在组合基因扰动数据上展示了方法的有效性。

原文摘要 · Abstract (English)

We consider the problem of modeling the effects of perturbations like gene knockouts on measurements such as single-cell RNA counts. Given data for some perturbations, we aim to predict the distribution of measurements for new combinations of perturbations. To address this challenging extrapolation task, we posit that perturbations act additively in a suitable, unknown embedding space. We formulate the data-generating process as a latent variable model, in which perturbations amount to mean shifts in latent space and can be combined additively. We then prove that, given sufficiently diverse training perturbations, the representation and perturbation effects are identifiable up to orthogonal transformation and use this to derive extrapolation guarantees for unseen perturbations that can be expressed as linear combinations of seen ones. To estimate the model from data, we propose the perturbation distribution autoencoder (PDAE), which is trained by maximizing the distributional similarity between true and simulated perturbation distributions. The trained model can then be used to predict previously unseen perturbation distributions. In support of our theoretical results, we demonstrate through simulations that PDAE can accurately predict the effects of unseen but identifiable perturbations, and showcase the method on combinatorial gene perturbation data.

基因调控分布建模外推预测潜变量模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。