AI & Computational Science

DiffImaginE: Imagine to Verify Entity Types with Diffusion

How the science connects

Multimodal learningDiffusion modelNamed entity recog…

AI Insight

This paper introduces DiffImaginE, a new approach for multimodal named entity recognition that uses diffusion models to verify whether text spans match proposed entity types by analyzing both textual and visual evidence. Unlike existing methods that compress visual information into single prototypes, DiffImaginE formulates verification as a probabilistic inference problem where competing entity-type hypotheses are ranked based on how well they explain observed visual evidence through a denoising process. Testing on Twitter datasets showed consistent improvements over deterministic baseline methods using the same encoder architecture.


This work advances the ability of AI systems to accurately identify and classify entities in social media posts and other multimodal content by better leveraging visual context alongside text. Improved entity recognition has applications in information extraction, content moderation, and knowledge graph construction from real-world multimedia data.


⚠️ Preprint – Noch nicht peer-reviewed

Dieser Artikel wurde noch nicht von unabhängigen Experten begutachtet. Die Ergebnisse sind vorläufig und sollten mit Vorsicht interpretiert werden.

Abstract: Multimodal named entity recognition (MNER) determines whether each candidate span and entity-type hypothesis is supported by joint textual and visual evidence. Existing imagine-and-compare verifiers map each (span, type) pair to one predicted visual feature, compressing diverse visual realisations into a single prototype and providing a compatibility score without explicit probabilistic semantics. We introduce DiffImaginE, which formulates MNER type verification as conditional latent diffusion inference. Given span-localised visual evidence, a type-conditioned denoiser predicts noise injected into its standardised latent. The resulting denoising error provides an ELBO-consistent surrogate for type-conditional negative log-likelihood, allowing competing type hypotheses to be ranked by how well they explain the observation. DiffImaginE retains a standard multimodal encoder stack and replaces the deterministic verifier with a classifier-free-guided diffusion scorer trained using Min-SNR weighting. We directly supervise per-type diffusion scores as classification logits, learn aggregation across noise levels, and use antithetic sampling to reduce Monte Carlo comparison variance. Our analysis shows that classifier-free guidance sharpens the induced type posterior and characterises when antithetic pairing reduces variance at equal denoiser cost. Experiments on Twitter-2015 and Twitter-2017 show consistent gains over a matched deterministic ImaginE control under the same encoder, auxiliary objectives, and evaluation protocol, supported by ablations and paired significance tests.

Source: DiffImaginE: Imagine to Verify Entity Types with Diffusion