Masked Diffusion Captioning for Visual Feature Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Feng, Chao, Wei, Zihao, Owens, Andrew
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917050810630144
author Feng, Chao
Wei, Zihao
Owens, Andrew
author_facet Feng, Chao
Wei, Zihao
Owens, Andrew
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2510_26799
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Masked Diffusion Captioning for Visual Feature Learning
Feng, Chao
Wei, Zihao
Owens, Andrew
Computer Vision and Pattern Recognition
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.
title Masked Diffusion Captioning for Visual Feature Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2510.26799