Joint Velocity-Growth Flow Matching for Single-Cell Dynamics Modeling

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wang, Dongyi, Jiang, Yuanwei, Zhang, Zhenyi, Gu, Xiang, Zhou, Peijie, Sun, Jian
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912694774267904
author Wang, Dongyi
Jiang, Yuanwei
Zhang, Zhenyi
Gu, Xiang
Zhou, Peijie
Sun, Jian
author_facet Wang, Dongyi
Jiang, Yuanwei
Zhang, Zhenyi
Gu, Xiang
Zhou, Peijie
Sun, Jian
contents Learning the underlying dynamics of single cells from snapshot data has gained increasing attention in scientific and machine learning research. The destructive measurement technique and cell proliferation/death result in unpaired and unbalanced data between snapshots, making the learning of the underlying dynamics challenging. In this paper, we propose joint Velocity-Growth Flow Matching (VGFM), a novel paradigm that jointly learns state transition and mass growth of single-cell populations via flow matching. VGFM builds an ideal single-cell dynamics containing velocity of state and growth of mass, driven by a presented two-period dynamic understanding of the static semi-relaxed optimal transport, a mathematical tool that seeks the coupling between unpaired and unbalanced data. To enable practical usage, we approximate the ideal dynamics using neural networks, forming our joint velocity and growth matching framework. A distribution fitting loss is also employed in VGFM to further improve the fitting performance for snapshot data. Extensive experimental results on both synthetic and real datasets demonstrate that VGFM can capture the underlying biological dynamics accounting for mass and state variations over time, outperforming existing approaches for single-cell dynamics modeling.
format Preprint
id arxiv_https___arxiv_org_abs_2505_13413
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Joint Velocity-Growth Flow Matching for Single-Cell Dynamics Modeling
Wang, Dongyi
Jiang, Yuanwei
Zhang, Zhenyi
Gu, Xiang
Zhou, Peijie
Sun, Jian
Machine Learning
Learning the underlying dynamics of single cells from snapshot data has gained increasing attention in scientific and machine learning research. The destructive measurement technique and cell proliferation/death result in unpaired and unbalanced data between snapshots, making the learning of the underlying dynamics challenging. In this paper, we propose joint Velocity-Growth Flow Matching (VGFM), a novel paradigm that jointly learns state transition and mass growth of single-cell populations via flow matching. VGFM builds an ideal single-cell dynamics containing velocity of state and growth of mass, driven by a presented two-period dynamic understanding of the static semi-relaxed optimal transport, a mathematical tool that seeks the coupling between unpaired and unbalanced data. To enable practical usage, we approximate the ideal dynamics using neural networks, forming our joint velocity and growth matching framework. A distribution fitting loss is also employed in VGFM to further improve the fitting performance for snapshot data. Extensive experimental results on both synthetic and real datasets demonstrate that VGFM can capture the underlying biological dynamics accounting for mass and state variations over time, outperforming existing approaches for single-cell dynamics modeling.
title Joint Velocity-Growth Flow Matching for Single-Cell Dynamics Modeling
topic Machine Learning
url https://arxiv.org/abs/2505.13413