mlsanityData to 13 Oct 2026

CRAFT: Constrained Reward via Attention Fine-Tuning for Subject Personalization without Composed Targets

arXiv 2608.14403 ↗First posted 14 Aug 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Subject-driven image personalization---generating new images that preserve the identity of one or several reference subjects in novel scenes---is a foundational capability for modern visual content creation. It is currently dominated by generalized methods that fine-tune a pretrained multimodal diffusion transformer (MMDiT) on hundreds of thousands to millions of paired (reference, composed-target) examples, where each composed target is a synthesized image of the subject in a novel scene. Producing such targets demands a costly multi-stage curation pipeline---LLM-based prompt generation, T2I-based composed-target synthesis, reference-subject extraction, VLM-based quality filtering, and correspondence labeling---and tightly couples each method to a particular target synthesizer and curation choice. We introduce CRAFT (Constrained Reward via Attention Fine-Tuning), a single-step ReFL framework that fine-tunes a pre-trained reference-aware MMDiT via LoRA adapters using a compact reference-only data construction---10K reference images and subject masks, with no composed-target supervision. CRAFT realizes a Where to look principle: attention-level rewards align noise- and phrase-token attention with the correct reference subject, and the resulting per-subject attention masks gate a pixel-level identity reward to keep image-space supervision consistent with the learned attention routing. Applied to FLUX.2-klein-9B, CRAFT achieves state-of-the-art performance on XVerseBench revwhile using no composed-target supervision---only 10K reference-only samples, whereas prior generalized methods require 150K to over 2M composed-target pairs. The same recipe transfers to other reference-aware backbones, consistently improving performance. Project page: https://jihun999.github.io/projects/CRAFT/.

Key facts. No citing papers in the map yet (data through 13 Oct 2026). Builds on 27 of its 43 references that are in the map. First posted on arXiv 14 Aug 2026.

Builds on

27 of 43 references.

References · 27

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,588
  2. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,857
  3. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,130
  4. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,637
  5. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,607
  6. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,570
  7. 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,421
  8. 2022An Image is Worth One Word: Personalizing Text-to-Image Generation using Textual Inversioncited by 2,240
  9. 2022Multi-Concept Customization of Text-to-Image Diffusioncited by 1,104
  10. 2025Emerging Properties in Unified Multimodal Pretrainingcited by 872
  11. 2023BLIP-Diffusion: Pre-trained Subject Representation for Controllable Text-to-Image Generation and Editingcited by 440
  12. 2023Directly Fine-Tuning Diffusion Models on Differentiable Rewardscited by 440
15 more
  1. 2024InstantID: Zero-shot Identity-Preserving Generation in Secondscited by 433
  2. 2023Florence-2: Advancing a Unified Representation for a Variety of Vision Taskscited by 406
  3. 2023Aligning Text-to-Image Diffusion Models with Reward Backpropagationcited by 286
  4. 2023Linguistic Binding in Diffusion Models: Enhancing Attribute Correspondence through Attention Map Alignmentcited by 171
  5. 2025Less-to-More Generalization: Unlocking More Controllability by In-Context Generationcited by 154
  6. 2023SSR-Encoder: Encoding Selective Subject Representation for Subject-Driven Generationcited by 143
  7. 2024MS-Diffusion: Multi-subject Zero-shot Image Personalization with Layout Guidancecited by 143
  8. 2025Echo-4o: Harnessing the Power of GPT-4o Synthetic Images for Improved Image Generationcited by 98
  9. 2025DreamO: A Unified Framework for Image Customizationcited by 83
  10. 2025Uniworld-V2: Reinforce Image Editing with Diffusion Negative-aware Finetuning and MLLM Implicit Feedbackcited by 79
  11. 2025XVerse: Consistent Multi-Subject Control of Identity and Semantic Attributes via DiT Modulationcited by 56
  12. 2024Resolving Multi-Condition Confusion for Finetuning-Free Personalized Image Generationcited by 43
  13. 2025USO: Unified Style and Subject-Driven Generation via Disentangled and Reward Learningcited by 27
  14. 2025UMO: Scaling Multi-Identity Consistency for Image Customization via Matching Rewardcited by 19
  15. 2025Scone: Bridging Composition and Distinction in Subject-Driven Image Generation via Unified Understanding-Generation Modelingcited by 10

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2023Subject-Diffusion:Open Domain Personalized Text-to-Image Generation without Test-time Fine-tuning
  2. 2025IMAGE-ALCHEMY: Advancing subject fidelity in personalised text-to-image generation
  3. 2024MM-Diff: High-Fidelity Image Personalization via Multi-Modal Condition Integration
  4. 2025DynASyn: Multi-Subject Personalization Enabling Dynamic Action Synthesis
  5. 2026SwiftPie: Lightning-fast Subject-driven Image Personalization via One step Diffusion
  6. 2024ComFusion: Personalized Subject Generation in Multiple Specific Scenes From Single Image
  7. 2023FastComposer: Tuning-Free Multi-Subject Image Generation with Localized Attention
  8. 2025Personalize Anything for Free with Diffusion Transformer