ScDiVa: Masked Discrete Diffusion for Joint Modeling of Single-Cell Identity and Expression
Mingxuan Wang, Gaoyang Jiang, ZiJia Ren, Cheng Chen, Chuangxin Zhao, Lu Shi, Yanbiao Ma
Abstract
Single-cell RNA-seq profiles are high-dimensional, sparse, and unordered, causing autoregressive generation to impose an artificial ordering bias and suffer from error accumulation. To address this, we propose scDiVa, a masked discrete diffusion foundation model that aligns generation with the dropout-like corruption process by defining a continuous-time forward masking mechanism in token space. ScDiVa features a bidirectional denoiser that jointly models discrete gene identities and continuous values, utilizing entropy-normalized serialization and a latent anchor token to maximize information efficiency and preserve global cell identity. The model is trained via depth-invariant time sampling and a dual denoising objective to simulate varying sparsity levels while ensuring precise recovery of both identity and magnitude. Pre-trained on 59 million cells, scDiVa achieves strong transfer performance across major benchmarks, including batch integration, cell type annotation, and perturbation response prediction. These results suggest that masked discrete diffusion serves as a biologically coherent and effective alternative to autoregression.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a693de77-36c0-4c23-bafd-a009f941fe14Builds on6
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Structured Denoising Diffusion Models in Discrete State-SpacesJacob Austin, Daniel D. Johnson, Jonathan Ho, Daniel Tarlow et al.NeurIPS 2021 · 2,256 citations
- Large Language Diffusion ModelsShen Nie, Fengqi Zhu, Zebin You, Xiaolu Zhang et al.NeurIPS 2025 · 949 citations
- Simplified and Generalized Masked Diffusion for Discrete DataJiaxin Shi, Kehang Han, Zhe Wang, Arnaud Doucet et al.NeurIPS 2024 · 693 citations
- Vector Quantized Diffusion Model for Text-to-Image SynthesisShuyang Gu, Dong Chen, Jianmin Bao, Fang Wen et al.CVPR 2022 · 607 citations
Related papers
- Scalable Single-Cell Gene Expression Generation with Latent Diffusion ModelsGiovanni Palla, Sudarshan Babu, Payam Dibaeinia, James Pearce et al.ICML 2026 · 7 citations
- Multi-Modal and Multi-Attribute Generation of Single Cells with CFGenAlessandro Palma, Till Richter, Hanyi Zhang, Manuel Lubetzki et al.ICLR 2025
- CountsDiff: A diffusion model on the natural numbers for generation and imputation of count-based dataRenzo Soatto, Anders Hoel, Greycen Ren, Shorna Alam et al.ICML 2026
- Doloris: Dual Conditional Diffusion Implicit Bridges with Sparsity Masking Strategy for Unpaired Single-Cell Perturbation EstimationChangxi Chi, Jun Xia, Yufei Huang, Zhuoli Ouyang et al.ICLR 2026 · 4 citations
- scDEBART: Predicting in silico Single-Cell Perturbation Responses via Large-Scale Differential Expression LearningJieun Sung, Wankyu KimICML 2026
