用量子桥模型直接预测基因扰动后单细胞变化,突破数据不配对难题。
Departures: Distributional Transport for Single-Cell Perturbation Prediction with Neural Schrödinger Bridges
- 基于最小批量最优传输配对,构建单向神经量子桥模型
- 在多个公开数据集上实现当前最佳扰动预测效果
- 适合药物研发与基因功能分析研究者使用
预测单细胞扰动结果可直接推动基因功能解析和候选药物筛选,是基础与转化医学的关键。但单细胞测序具有破坏性,无法观测同一细胞扰动前后状态,导致数据多为非配对。现有神经生成运输模型或缺乏显式条件控制,或依赖先验空间间接对齐分布,难以精确建模扰动。本文通过逼近薛定谔桥(Schrödinger Bridge, SB),定义熵正则化最优传输的随机动态映射,直接对齐不同扰动条件下对照组与处理组的细胞分布。不同于以往需双向建模以推断源-目标样本对应关系的方法,我们采用最小批量最优传输(Minibatch-OT)进行配对,避免反向过程定义带来的病态问题,直接引导桥模型学习,实现可扩展的SB近似。我们构建了两个SB模型:一个建模离散基因激活状态,另一个建模连续表达分布。联合训练可准确捕捉扰动响应与单细胞异质性。在多个公开遗传与药物扰动数据集上的实验表明,该模型有效建模细胞异质性响应,并达到当前最优性能。
原文摘要 · Abstract (English)
Predicting single-cell perturbation outcomes directly advances gene function analysis and facilitates drug candidate selection, making it a key driver of both basic and translational biomedical research. However, a major bottleneck in this task is the unpaired nature of single-cell data, as the same cell cannot be observed both before and after perturbation due to the destructive nature of sequencing. Although some neural generative transport models attempt to tackle unpaired single-cell perturbation data, they either lack explicit conditioning or depend on prior spaces for indirect distribution alignment, limiting precise perturbation modeling. In this work, we approximate Schrödinger Bridge (SB), which defines stochastic dynamic mappings recovering the entropy-regularized optimal transport (OT), to directly align the distributions of control and perturbed single-cell populations across different perturbation conditions. Unlike prior SB approximations that rely on bidirectional modeling to infer optimal source-target sample coupling, we leverage Minibatch-OT based pairing to avoid such bidirectional inference and the associated ill-posedness of defining the reverse process. This pairing directly guides bridge learning, yielding a scalable approximation to the SB. We approximate two SB models, one modeling discrete gene activation states and the other continuous expression distributions. Joint training enables accurate perturbation modeling and captures single-cell heterogeneity. Experiments on public genetic and drug perturbation datasets show that our model effectively captures heterogeneous single-cell responses and achieves state-of-the-art performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。