Doloris 用双扩散模型无配对预测单细胞扰动,提升药物研发效率。
Doloris: Dual Conditional Diffusion Implicit Bridges with Sparsity Masking Strategy for Unpaired Single-Cell Perturbation Estimation
- 用双扩散模型分别学习对照与扰动数据分布,隐式对齐不需配对
- 引入稀疏掩码策略,聚焦表达基因而非零值,保持高维数据多样性
- 适用于无配对单细胞数据,助力基因发现与药物筛选
单细胞扰动响应估计有助于识别关键基因并提升药物筛选效率。然而,单细胞测序具有破坏性,无法获取同一细胞扰动前后的表型,导致扰动与未扰动数据天然无配对,成为单细胞扰动建模中的核心难题。此外,单细胞表达数据维度高且稀疏,直接建模易聚焦于零值而忽略有意义模式。为此,我们提出一种新范式:利用双扩散模型分别学习对照和扰动分布,并通过共享高斯隐空间隐式对齐,无需显式细胞配对。进一步提出稀疏掩码策略,掩码模型学习预测零表达基因,使扩散模型专注捕捉表达基因间的有效模式,从而在高维稀疏数据中保持多样性。我们构建了名为 Doloris 的生成框架,定义了建模无配对、高维、稀疏单细胞扰动数据的新范式。其基于双条件扩散模型分别学习控制与扰动分布,并结合稀疏掩码策略提升零值基因预测能力。在公开数据集上的结果表明,该模型能有效捕捉单细胞扰动的多样性,达到当前最优性能。为促进可复现性,代码已包含在补充材料中。
原文摘要 · Abstract (English)
Estimating single-cell responses across various perturbations facilitates the identification of key genes and enhances drug screening, significantly boosting experimental efficiency. However, single-cell sequencing is a destructive process, making it impossible to capture the same cell's phenotype before and after perturbation. Consequently, data collected under perturbed and unperturbed conditions are inherently unpaired, creating a critical yet unresolved problem in single-cell perturbation modeling. Moreover, the high dimensionality and sparsity of single-cell expression make direct modeling prone to focusing on zeros and neglecting meaningful patterns. To address these problems, we propose a new paradigm for single-cell perturbation modeling. Specifically, we leverage dual diffusion models to learn the control and perturbed distributions separately, and implicitly align them through a shared Gaussian latent space, without requiring explicit cell pairing. Furthermore, we introduce a sparsity masking strategy in which the mask model learns to predict zero-expressed genes, allowing the diffusion model to focus on capturing meaningful patterns among expressed genes and thereby preserving diversity in high-dimensional sparse data. We introduce \textbf{Doloris}, a generative framework that defines a new paradigm for modeling unpaired, high-dimensional, and sparse single-cell perturbation data. It leverages dual conditional diffusion models for separate learning of control and perturbed distributions, complemented by a sparsity masking strategy to enhance prediction of zero-valued genes. The results on publicly available datasets show that our model effectively captures the diversity of single-cell perturbations and achieves state-of-the-art performance. To facilitate reproducibility, we include the code in the supplementary materials.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。