CellBRIDGE: Learning Cellular Trajectories via Interaction-Aware Alignment
- URL: http://arxiv.org/abs/2605.30635v1
- Date: Thu, 28 May 2026 22:37:54 GMT
- Title: CellBRIDGE: Learning Cellular Trajectories via Interaction-Aware Alignment
- Abstract summary: Inferring dynamics from population snapshots is a fundamental challenge in machine learning and biology.<n>We introduce CellBRIDGE, which augments feature-based OT with a directed, typed interaction cost derived from ligand-receptor activity.<n>By explicitly modeling cell-cell communication, CellBRIDGE improves cross-snapshot couplings and downstream trajectory estimates.
- Score: 46.660621971318996
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Inferring dynamics from population snapshots is a fundamental challenge in machine learning and biology. In scRNA-sequencing (scRNA-seq), destructive measurements preclude direct tracking of individual cells across time, making trajectory inference underdetermined. Optimal Transport (OT) provides a principled framework for snapshot alignment, but a long-standing modeling question is which cost functions yield biologically meaningful couplings. Standard OT approaches rely on gene-expression distances, implicitly treating cells as independent points and neglecting structured cell-cell communication mediated by ligand-receptor signaling. We introduce CellBRIDGE (Cell-Based Regularized Interaction-Driven Gene Expression), which augments feature-based OT with a directed, typed interaction cost derived from ligand-receptor activity. By explicitly modeling cell-cell communication, CellBRIDGE improves cross-snapshot couplings and downstream trajectory estimates across synthetic and real scRNA-seq datasets relative to feature-only baselines. Notably, CellBRIDGE enables mechanistically interpretable in silico perturbations: on lung cancer data, silencing specific ligand-receptor pairs induces trajectory shifts that recapitulate expected effects of targeted pathway inhibition.
Related papers
- FreeBridge: Variational Schrödinger Bridges for Cellular Transition Dynamics [48.14051378654045]
FreeBridge is a Schrdinger Bridge for single-cell transition modeling under endpoint-only supervision.<n>FreeBridge maintains competitive or improved endpoint fidelity and mechanism mechanism retention under unified evaluation protocol.
arXiv Detail & Related papers (2026-06-09T16:46:59Z) - BRIDGE: Biological Evidence Refinement and Heterogeneous Dynamic Gating for Gene Regulatory Networks [25.29216866424431]
We propose an innovative framework named Biological Evidence Refinement and Heterogeneous Dynamic Gating for Gene Regulatory Networks (BRIDGE)<n>BRIDGE extracts gene and cell representations from the expression matrix and its matrix dual, and performs contrastive learning in the gene space and cell space between self and neighbors.<n>It then applies heterogeneous gated encoding to adaptively regulate information transfer between genes and cells, enabling robust transcription factor-to-target gene prediction.
arXiv Detail & Related papers (2026-06-02T11:54:36Z) - Latent Causal Diffusions for Single-Cell Perturbation Modeling [83.47931153555321]
We present a generative model that frames single-cell gene expression as a stationary diffusion process observed under measurement noise.<n> LCD outperforms established approaches in predicting the distributional shifts of unseen perturbation combinations in single-cell RNA-sequencing screens.<n>We develop an approach we call causal linearization via perturbation responses (CLIPR), which yields an approximation of the direct causal effects between all genes.
arXiv Detail & Related papers (2026-01-20T16:15:38Z) - Departures: Distributional Transport for Single-Cell Perturbation Prediction with Neural Schrödinger Bridges [51.83259180910313]
A major bottleneck in gene function analysis is the unpaired nature of single-cell data.<n>We approximate Schrdinger Bridge (SB) to tackle unpaired single-cell perturbation data.<n>Our model effectively captures heterogeneous single-cell responses and achieves state-of-the-art performance.
arXiv Detail & Related papers (2025-11-17T08:27:13Z) - Learning Explicit Single-Cell Dynamics Using ODE Representations [33.16920280365721]
Cell-Mechanistic Neural Networks (Cell-MNN) is an encoder-decoder architecture whose latent representation is a locally linearized ODE governing the dynamics of cellular evolution from stem to tissue cells.<n>We show that Cell-MNN achieves competitive performance on single-cell benchmarks, surpasses state-of-the-art baselines in scaling to larger datasets and joint training across multiple datasets, while also learning interpretable gene interactions that we validate against the TRRUST database of gene interactions.
arXiv Detail & Related papers (2025-10-03T11:15:16Z) - STAGED: A Multi-Agent Neural Network for Learning Cellular Interaction Dynamics [8.659754814655303]
Single-cell technology has improved our understanding of cellular states and subpopulations in various tissues under normal and diseased conditions.<n>With spatial transcriptomics, we can represent cellular organization, along with dynamic cell-cell interactions that lead to changes in cell state.<n>We introduce Spatio Temporal Agent-Based Graph Evolution Dynamics(STAGED) to model intercellular communication, and its effect on the intracellular gene regulatory network.
arXiv Detail & Related papers (2025-07-15T18:46:07Z) - Clustering with Communication: A Variational Framework for Single Cell Representation Learning [2.275097126764287]
We propose CCCVAE, a variational autoencoder framework that incorporates CCC signals into single-cell representation learning.<n>We show that CCCVAE improves clustering performance, achieving higher evaluation scores than standard VAE baselines.
arXiv Detail & Related papers (2025-05-08T01:53:36Z) - A scalable gene network model of regulatory dynamics in single cells [88.48246132084441]
We introduce a Functional Learnable model of Cell dynamicS, FLeCS, that incorporates gene network structure into coupled differential equations to model gene regulatory functions.<n>Given (pseudo)time-series single-cell data, FLeCS accurately infers cell dynamics at scale.
arXiv Detail & Related papers (2025-03-25T19:19:21Z) - Cell as Point: One-Stage Framework for Efficient Cell Tracking [56.98130311648714]
We propose CAP, a novel end-to-end one-stage framework that reimagines cell tracking by treating Cell as Point.<n>Unlike traditional methods, CAP eliminates the need for explicit detection or segmentation, instead jointly tracking cells for sequences in one stage by leveraging the inherent correlations among their trajectories.<n>Cap demonstrates promising cell tracking performance and is 8 to 32 times more efficient than existing methods.
arXiv Detail & Related papers (2024-11-22T10:16:35Z) - DynaCLR: Contrastive Learning of Cellular Dynamics with Temporal Regularization [0.5385157903509136]
DynaCLR is a self-supervised method for embedding cell and organelle Dynamics via Contrastive Learning of Representations of time-lapse images.<n>We integrate single-cell tracking and time-aware contrastive sampling to learn robust, temporally regularized representations of cell dynamics.
arXiv Detail & Related papers (2024-10-15T05:01:25Z) - Granger causal inference on DAGs identifies genomic loci regulating
transcription [77.58911272503771]
GrID-Net is a framework based on graph neural networks with lagged message passing for Granger causal inference on DAG-structured systems.
Our application is the analysis of single-cell multimodal data to identify genomic loci that mediate the regulation of specific genes.
arXiv Detail & Related papers (2022-10-18T21:15:10Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.