arXiv · 2510.26799
Masked Diffusion Captioning for Visual Feature Learning
Abstract
We learn visual features by captioning images with an image-conditioned masked diffusion language model, a formulation we call masked diffusion captioning (MDC). During training, text tokens in each image-caption pair are masked at a randomly chosen ratio, and a decoder conditioned on visual features is trained to reconstruct the original text. After training, the learned visual features can be applied to downstream vision tasks. Unlike autoregressive captioning, the strength of the visual learning signal in MDC does not depend on each token's position in the sequence, reducing the need for auxiliary objectives. Linear probing experiments across a variety of academic-scale models and datasets show that the learned visual features are competitive with those produced by autoregressive and contrastive approaches.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Chao Feng, Zihao Wei, Andrew Owens. 2025-10-30. Masked Diffusion Captioning for Visual Feature Learning. https://arxiv.org/abs/2510.26799
Cite the original work for its findings. Save a collection to share your selection of sources.