mlsanityData to 7 Oct 2026

CoMat: Aligning Text-to-Image Diffusion Model with Image-to-Text Concept Matching

arXiv 2404.03653 ↗First posted 4 Apr 2024cs.CV cs.AI cs.CLcited by 60

On the map: Diffusion models → Image generation, text-to-image and editing

also in 1 more
Abstract (arXiv)

Diffusion models have demonstrated great success in the field of text-to-image generation. However, alleviating the misalignment between the text prompts and images is still challenging. The root reason behind the misalignment has not been extensively investigated. We observe that the misalignment is caused by inadequate token attention activation. We further attribute this phenomenon to the diffusion model's insufficient condition utilization, which is caused by its training paradigm. To address the issue, we propose CoMat, an end-to-end diffusion model fine-tuning strategy with an image-to-text concept matching mechanism. We leverage an image captioning model to measure image-to-text alignment and guide the diffusion model to revisit ignored tokens. A novel attribute concentration module is also proposed to address the attribute binding problem. Without any image or human preference data, we use only 20K text prompts to fine-tune SDXL to obtain CoMat-SDXL. Extensive experiments show that CoMat-SDXL significantly outperforms the baseline model SDXL in two text-to-image alignment benchmarks and achieves start-of-the-art performance.

Key facts. Cited by 60 papers in the map (data through 7 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 70 of its 94 references that are in the map. First posted on arXiv 4 Apr 2024.

Builds on

70 of 94 references.

Closest ancestors · 23

Same direction, newest first

  1. 2024Be Yourself: Bounded Attention for Multi-Subject Text-to-Image Generation25 Mar
  2. 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignment8 Mar
  3. 2024Mastering Text-to-Image Diffusion: Recaptioning, Planning, and Generating with Multimodal LLMs22 Jan
  4. 2023CONFORM: Contrast is All You Need For High-Fidelity Text-to-Image Diffusion Models11 Dec
  5. 2023Self-correcting LLM-controlled Diffusion Models27 Nov
  6. 2023UFOGen: You Forward Once Large Scale Text-to-Image Generation via Diffusion GANs14 Nov
  7. 2023Improving Compositional Text-to-image Generation with Large Vision-Language Models10 Oct
  8. 2023Dense Text-to-Image Generation with Attention Modulation24 Aug
  9. 2023Divide & Bind Your Attention for Improved Generative Semantic Nursing20 Jul
  10. 2023BoxDiff: Text-to-Image Synthesis with Training-Free Box-Constrained Diffusion20 Jul
  11. 2023A-STAR: Test-time Attention Segregation and Retention for Text-to-image Synthesis26 Jun
  12. 2023Linguistic Binding in Diffusion Models: Enhancing Attribute Correspondence through Attention Map Alignment15 Jun
11 more
  1. 2023Grounded Text-to-Image Synthesis with Attention Refocusing8 Jun
  2. 2023RAPHAEL: Text-to-Image Generation via Large Mixture of Diffusion Paths29 May
  3. 2023LLM-grounded Diffusion: Enhancing Prompt Understanding of Text-to-Image Diffusion Models with Large Language Models23 May
  4. 2023HRS-Bench: Holistic, Reliable and Scalable Benchmark for Text-to-Image Models11 Apr
  5. 2023Attend-and-Excite: Attention-Based Semantic Guidance for Text-to-Image Diffusion Models31 Jan
  6. 2023GLIGEN: Open-Set Grounded Text-to-Image Generation17 Jan
  7. 2022Training-Free Structured Diffusion Guidance for Compositional Text-to-Image Synthesis9 Dec
  8. 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generation25 Aug
  9. 2022Compositional Visual Generation with Composable Diffusion Models3 Jun
  10. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understanding23 May
  11. 2022Hierarchical Text-Conditional Image Generation with CLIP Latents13 Apr

Other references · 47

  1. 2017Attention Is All You Needcited by 59,932
  2. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  3. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  4. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  5. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  6. 2023GPT-4 Technical Reportcited by 18,132
  7. 2014Microsoft COCO: Common Objects in Contextcited by 17,498
  8. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  9. 2023Visual Instruction Tuningcited by 7,792
  10. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
  11. 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,069
  12. 2022BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generationcited by 4,310
35 more
  1. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
  2. 2021Zero-Shot Text-to-Image Generationcited by 4,276
  3. 2023MiniGPT-4: Enhancing Vision-Language Understanding with Advanced Large Language Modelscited by 2,689
  4. 2019ViLBERT: Pretraining Task-Agnostic Visiolinguistic Representations for Vision-and-Language Taskscited by 2,219
  5. 2021Align before Fuse: Vision and Language Representation Learning with Momentum Distillationcited by 1,612
  6. 2019LXMERT: Learning Cross-Modality Encoder Representations from Transformerscited by 1,548
  7. 2015Microsoft COCO Captions: Data Collection and Evaluation Servercited by 1,447
  8. 2022CoCa: Contrastive Captioners are Image-Text Foundation Modelscited by 1,265
  9. 2021ViLT: Vision-and-Language Transformer Without Convolution or Region Supervisioncited by 1,224
  10. 2020Oscar: Object-Semantics Aligned Pre-training for Vision-Language Taskscited by 1,213
  11. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
  12. 2023ShareGPT4V: Improving Large Multi-Modal Models with Better Captionscited by 1,000
  13. 2023Training Diffusion Models with Reinforcement Learningcited by 939
  14. 2024Grounded SAM: Assembling Open-World Models for Diverse Visual Taskscited by 914
  15. 2024Are We on the Right Way for Evaluating Large Vision-Language Models?cited by 893
  16. 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesiscited by 854
  17. 2023LLaMA-Adapter: Efficient Fine-tuning of Language Models with Zero-init Attentioncited by 776
  18. 2023Adversarial Diffusion Distillationcited by 741
  19. 2023Aligning Large Multimodal Models with Factually Augmented RLHFcited by 703
  20. 2022When and why vision-language models behave like bags-of-words, and what to do about it?cited by 603
  21. 2024MathVerse: Does Your Multi-modal LLM Truly See the Diagrams in Visual Math Problems?cited by 573
  22. 2023Segment Everything Everywhere All at Oncecited by 546
  23. 2022GIT: A Generative Image-to-text Transformer for Vision and Languagecited by 536
  24. 2022Image as a Foreign Language: BEiT Pretraining for All Vision and Vision-Language Taskscited by 510
  25. 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewardscited by 431
  26. 2023TIFA: Accurate and Interpretable Text-to-Image Faithfulness Evaluation with Question Answeringcited by 416
  27. 2023Training-Free Layout Control with Cross-Attention Guidancecited by 312
  28. 2023T2I-CompBench++: An Enhanced and Comprehensive Benchmark for Compositional Text-to-image Generationcited by 253
  29. 2023Personalize Segment Anything Model with One Shotcited by 250
  30. 2023Davidsonian Scene Graph: Improving Reliability in Fine-grained Evaluation for Text-to-Image Generationcited by 159
  31. 2021Simple multi-dataset detectioncited by 126
  32. 2023DreamSync: Aligning Text-to-Image Generation with Image Understanding Feedbackcited by 57
  33. 2024Be-Your-Outpainter: Mastering Video Outpainting through Input-Specific Adaptationcited by 29
  34. 2023Stable Diffusion is Unstablecited by 27
  35. 2024Lumen: Unleashing Versatile Vision-Centric Capabilities of Large Multimodal Modelscited by 24

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 18; 2025: 30; 2026: 12.

60 papers.

Sort:

By area:

  1. 2025T2I-R1: Reinforcing Image Generation with Collaborative Semantic-level and Token-level CoTcited by 162
  2. 2025Can We Generate Images with CoT? Let's Verify and Reinforce Image Generation Step by Stepcited by 130
  3. 2024MMSearch: Benchmarking the Potential of Large Models as Multi-modal Search Enginescited by 98
  4. 2025Echo-4o: Harnessing the Power of GPT-4o Synthetic Images for Improved Image Generationcited by 95
  5. 2024Phased Consistency Modelscited by 83
  6. 2024MoVA: Adapting Mixture of Vision Experts to Multimodal Contextcited by 82
  7. 2025WebGen-Bench: Evaluating LLMs on Generating Interactive and Functional Websites from Scratchcited by 66
  8. 2024Exploring the Role of Large Language Models in Prompt Encoding for Diffusion Modelscited by 45
  9. 2024Token Merging for Training-Free Semantic Binding in Text-to-Image Synthesiscited by 35
  10. 2025Diffusion Model as a Noise-Aware Latent Reward Model for Step-Level Preference Optimizationcited by 35
  11. 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Futurecited by 34
  12. 2024Scalable Ranked Preference Optimization for Text-to-Image Generationcited by 33
38 more
  1. 2025DenseDPO: Fine-Grained Temporal Preference Optimization for Video Diffusion Modelscited by 32
  2. 2024PixWizard: Versatile Image-to-Image Visual Assistant with Open-Language Instructionscited by 31
  3. 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewardscited by 29
  4. 2024EasyRef: Omni-Generalized Group Image Reference for Diffusion Models via Multimodal LLMcited by 25
  5. 2024Robust CLIP-Based Detector for Exposing Diffusion Model-Generated Imagescited by 20
  6. 2025D-Fusion: Direct Preference Optimization for Aligning Diffusion Models with Visually Consistent Samplescited by 14
  7. 2025Detect-and-Guide: Self-regulation of Diffusion Models for Safe Text-to-Image Generation via Guideline Token Optimizationcited by 12
  8. 2025Dual-Process Image Generationcited by 12
  9. 2024Enhancing MMDiT-Based Text-to-Image Models for Similar Subject Generationcited by 11
  10. 2024Mask Approximation Net: A Novel Diffusion Model Approach for Remote Sensing Change Captioningcited by 10
  11. 2024Text-to-Image Synthesis: A Decade Surveycited by 9
  12. 2024Attention Overlap Is Responsible for The Entity Missing Problem in Text-to-image Diffusion Models!cited by 8
  13. 2025PiSA: A Self-Augmented Data Engine and Training Strategy for 3D Understanding with Large Modelscited by 8
  14. 2025Turning Internal Gap into Self-Improvement: Promoting the Generation-Understanding Unification in MLLMscited by 8
  15. 2024CoMPaSS: Enhancing Spatial Understanding in Text-to-Image Diffusion Modelscited by 7
  16. 2025SUDER: Self-Improving Unified Large Multimodal Models for Understanding and Generation with Dual Self-Rewardscited by 6
  17. 2025Vision-Language Models as Differentiable Semantic and Spatial Rewards for Text-to-3D Generationcited by 6
  18. 2025Geometrical Properties of Text Token Embeddings for Strong Semantic Binding in Text-to-Image Generationcited by 5
  19. 2026Beyond the Dirac Delta: Mitigating Diversity Collapse in Reinforcement Fine-Tuning for Versatile Image Generationcited by 5
  20. 2024GraPE: A Generate-Plan-Edit Framework for Compositional T2I Synthesiscited by 4
  21. 2025Detail++: Training-Free Detail Enhancer for T2I Diffusion Modelscited by 4
  22. 2025Guiding What Not to Generate: Automated Negative Prompting for Text-Image Alignmentcited by 4
  23. 2025IMG: Calibrating Diffusion Models via Implicit Multimodal Guidancecited by 3
  24. 2025CalibCLIP: Contextual Calibration of Dominant Semantics for Text-Driven Image Retrievalcited by 3
  25. 2025AgeBooth: Controllable Facial Aging and Rejuvenation via Diffusion Modelscited by 3
  26. 2025Bridging Fidelity-Reality with Controllable One-Step Diffusion for Image Super-Resolutioncited by 3
  27. 2024Diffusion-RSCC: Diffusion Probabilistic Model for Change Captioning in Remote Sensing Imagescited by 2
  28. 2025CARINOX: Inference-time Scaling with Category-Aware Reward-based Initial Noise Optimization and Explorationcited by 2
  29. 2025Free Lunch Alignment of Text-to-Image Diffusion Models without Preference Image Pairscited by 2
  30. 2025DOS: Directional Object Separation in Text Embeddings for Multi-Object Image Generationcited by 2
  31. 2025Reg-DPO: SFT-Regularized Direct Preference Optimization with GT-Pair for Improving Video Generationcited by 2
  32. 2026Selective Fine-Tuning for Targeted and Robust Concept Unlearningcited by 2
  33. 2024Low-Biased General Annotated Dataset Generationcited by 1
  34. 2025Coffee: Controllable Diffusion Fine-tuningcited by 1
  35. 2026Say Cheese! Detail-Preserving Portrait Collection Generation via Natural Language Editscited by 1
  36. 2026Diagnosing and Correcting Concept Omission in Multimodal Diffusion Transformerscited by 1
  37. 2025ISAC: Training-Free Instance-to-Semantic Attention Control for Multi-Instance Generationcited by 0
  38. 2025ComposeAnything: Composite Object Priors for Text-to-Image Generationcited by 0

50 most cited of 60 All 60 (JSON)

Similar, not cited

Close in wording, no citation either way

  1. 2024AlignIT: Enhancing Prompt Alignment in Customization of Text-to-Image Models
  2. 2024Bridging the Gap: Aligning Text-to-Image Diffusion Models with Specific Feedback
  3. 2024Towards Better Text-to-Image Generation Alignment via Attention Modulation
  4. 2023Divide, Evaluate, and Refine: Evaluating and Improving Text-to-Image Alignment with Iterative VQA Feedback
  5. 2025Aligning Text to Image in Diffusion Models is Easier Than You Think
  6. 2023RealignDiff: Boosting Text-to-Image Diffusion Model with Coarse-to-fine Semantic Re-alignment
  7. 2026Alignment-Guided Score Matching for Text-to-Image Alignment in Diffusion Models
  8. 2024Information Theoretic Text-to-Image Alignment