Modeling Cell Dynamics and Interactions with Unbalanced Mean Field Schrödinger Bridge

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Zhenyi, Wang, Zihan, Sun, Yuhao, Li, Tiejun, Zhou, Peijie
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918170744324096
author Zhang, Zhenyi
Wang, Zihan
Sun, Yuhao
Li, Tiejun
Zhou, Peijie
author_facet Zhang, Zhenyi
Wang, Zihan
Sun, Yuhao
Li, Tiejun
Zhou, Peijie
contents Modeling the dynamics from sparsely time-resolved snapshot data is crucial for understanding complex cellular processes and behavior. Existing methods leverage optimal transport, Schrödinger bridge theory, or their variants to simultaneously infer stochastic, unbalanced dynamics from snapshot data. However, these approaches remain limited in their ability to account for cell-cell interactions. This integration is essential in real-world scenarios since intercellular communications are fundamental life processes and can influence cell state-transition dynamics. To address this challenge, we formulate the Unbalanced Mean-Field Schrödinger Bridge (UMFSB) framework to model unbalanced stochastic interaction dynamics from snapshot data. Inspired by this framework, we further propose CytoBridge, a deep learning algorithm designed to approximate the UMFSB problem. By explicitly modeling cellular transitions, proliferation, and interactions through neural networks, CytoBridge offers the flexibility to learn these processes directly from data. The effectiveness of our method has been extensively validated using both synthetic gene regulatory data and real scRNA-seq datasets. Compared to existing methods, CytoBridge identifies growth, transition, and interaction patterns, eliminates false transitions, and reconstructs the developmental landscape with greater accuracy. Code is available at: https://github.com/zhenyiizhang/CytoBridge-NeurIPS.
format Preprint
id arxiv_https___arxiv_org_abs_2505_11197
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Modeling Cell Dynamics and Interactions with Unbalanced Mean Field Schrödinger Bridge
Zhang, Zhenyi
Wang, Zihan
Sun, Yuhao
Li, Tiejun
Zhou, Peijie
Machine Learning
Optimization and Control
Quantitative Methods
Modeling the dynamics from sparsely time-resolved snapshot data is crucial for understanding complex cellular processes and behavior. Existing methods leverage optimal transport, Schrödinger bridge theory, or their variants to simultaneously infer stochastic, unbalanced dynamics from snapshot data. However, these approaches remain limited in their ability to account for cell-cell interactions. This integration is essential in real-world scenarios since intercellular communications are fundamental life processes and can influence cell state-transition dynamics. To address this challenge, we formulate the Unbalanced Mean-Field Schrödinger Bridge (UMFSB) framework to model unbalanced stochastic interaction dynamics from snapshot data. Inspired by this framework, we further propose CytoBridge, a deep learning algorithm designed to approximate the UMFSB problem. By explicitly modeling cellular transitions, proliferation, and interactions through neural networks, CytoBridge offers the flexibility to learn these processes directly from data. The effectiveness of our method has been extensively validated using both synthetic gene regulatory data and real scRNA-seq datasets. Compared to existing methods, CytoBridge identifies growth, transition, and interaction patterns, eliminates false transitions, and reconstructs the developmental landscape with greater accuracy. Code is available at: https://github.com/zhenyiizhang/CytoBridge-NeurIPS.
title Modeling Cell Dynamics and Interactions with Unbalanced Mean Field Schrödinger Bridge
topic Machine Learning
Optimization and Control
Quantitative Methods
url https://arxiv.org/abs/2505.11197