Search papers, labs, and topics across Lattice.
Distribution-Conditioned Transport (DCT) is introduced as a framework for learning transport maps between distributions, conditioned on learned embeddings of the source and target distributions. This enables generalization to unseen distribution pairs, addressing a key limitation in traditional transport models. DCT's versatility is showcased by its compatibility with various transport mechanisms, including flow matching and distributional divergence-based models, and its application to diverse biological problems.
Generalize transport maps to unseen source and target distributions by conditioning on learned distribution embeddings, unlocking applications in single-cell genomics, mass cytometry, and more.
Learning a transport model that maps a source distribution to a target distribution is a canonical problem in machine learning, but scientific applications increasingly require models that can generalize to source and target distributions unseen during training. We introduce distribution-conditioned transport (DCT), a framework that conditions transport maps on learned embeddings of source and target distributions, enabling generalization to unseen distribution pairs. DCT also allows semi-supervised learning for distributional forecasting problems: because it learns from arbitrary distribution pairs, it can leverage distributions observed at only one condition to improve transport prediction. DCT is agnostic to the underlying transport mechanism, supporting models ranging from flow matching to distributional divergence-based models (e.g. Wasserstein, MMD). We demonstrate the practical performance benefits of DCT on synthetic benchmarks and four applications in biology: batch effect transfer in single-cell genomics, perturbation prediction from mass cytometry data, learning clonal transcriptional dynamics in hematopoiesis, and modeling T-cell receptor sequence evolution.
