Flow-Matching Refined Dirichlet Process Priors for Structurally Coherent Single-Cell Representation Learning
Fu, Z.
Show abstract
Standard autoencoders for single-cell transcriptomics learn latent spaces whose cluster structure emerges only post hoc through K-means or community detection, leaving cluster count and boundary quality uncontrolled during training. Here we ask whether imposing an adaptive nonparametric prior can shift this balance. We equip a feedforward autoen-coder with an online Dirichlet Process Mixture Model (DPMM) prior that refits cluster assignments throughout training and directly regularizes latent compactness and separation. Across 56 scRNA-seq datasets the DPMM prior produces a pronounced geometry- concordance trade-off : cluster compactness (ASW) improves by 127% and Davies-Bouldin overlap drops by 47%, but label-recovery metrics decline (NMI -17%, ARI -21%) and downstream kNN accuracy falls from 0.784 to 0.725. Wilcoxon signed-rank tests confirm that the geometry gains are significant with large Cliffs{delta} effects while concordance losses remain bounded and non-significant. A second-stage conditional-flow refinement (DPMM-FM) further improves projection fidelity (DRE 0.751, LSE 0.695, DREX 0.873) at additional concordance cost, revealing a three-tier operating regime: prior-free for label recovery, DPMM for manifold geometry, and DPMM-FM for visualization fidelity. Against 18 external baselines DPMM-Base wins 70.5% of core-metric comparisons (p<0.05). Gene Ontology enrichment confirms that geometry-improved latent components recover coherent biological programs. Rather than claiming universal superiority, this study characterizes the operating envelope of nonparametric mixture priors and identifies the task contexts-- trajectory analysis, manifold visualization, and program-level annotation--where adaptive geometric structure outweighs label-counting accuracy.
Matching journals
The top 4 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- Evaluating discrepancies in dimensionality reduction for time-series single-cell RNA-sequencing data 96%
- Novel multi-omics deconfounding variational autoencoders can obtain meaningful disease subtyping 95%
- Synthetic observations from deep generative models and binary omics data with limited sample size 94%
Similar papers in this journal
- Single-Cell Multi-Modal GAN (scMMGAN) reveals spatial patterns in single-cell data from triple negative breast cancer 96%
- Tokenized and Continuous Embedding Compressions of Protein Sequence and Structure 96%
- Generating hard-to-obtain information from easy-to-obtain information: applications in drug discovery and clinical inference 95%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.