MaskSense: Confronting the Visual Exploration Trap in Masked Image Generation
Not on the map. Why?
Abstract (arXiv)
Reinforcement learning (RL) has shown strong potential for aligning generative models with human intent, but adapting RL to masked generative models (MGMs) remains largely underexplored. Existing approaches that directly adapt Group Relative Policy Optimization (GRPO) to MGMs leave a foundational conflict unaddressed: MGMs' exploitative decoding nature is inherently at odds with the exploratory diversity that GRPO presupposes. In this work, we reveal a structural decoding bias termed the visual exploration trap: confidence-based sampling trades the exploration of diverse subject realizations for the greedy resolution of low-uncertainty background regions, causing premature collapse of the generative exploration space and starving GRPO of the rollout diversity it depends on. To this end, we propose MaskSense, a novel RL framework for MGMs that confronts the visual exploration trap at both the sampling and optimization stages. Specifically, we introduce a semantic anchored routing sampling mechanism that leverages semantic priors to preserve high-entropy subject token exploration while amplifying intra-group reward variance through exploratory-exploitative routing to yield more discriminative advantage estimates. Furthermore, we design a global entropy transition anchoring strategy to identify the most consequential decoding steps and concentrate gradient updates on them. Extensive experiments on multiple text-to-image benchmarks demonstrate that MaskSense substantially improves upon the base model, with GenEval accuracy improving from 54% to 86% and HPS from 28.89 to 36.39, achieving state-of-the-art performance.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 52 of its 67 references that are in the map. First posted on arXiv 12 Oct 2026.
Builds on
52 of 67 references.
References · 52
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
38 more
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,387
- 2022MaskGIT: Masked Generative Image Transformercited by 1,099
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
- 2023Training Diffusion Models with Reinforcement Learningcited by 982
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Modelscited by 938
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
- 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 790
- 2024Emu3: Next-Token Prediction is All You Needcited by 781
- 2018Image Transformercited by 772
- 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 752
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
- 2023Muse: Text-To-Image Generation via Masked Generative Transformerscited by 648
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
- 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
- 2025DanceGRPO: Unleashing GRPO on Visual Generationcited by 413
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 250
- 2025T2I-R1: Reinforcing Image Generation with Collaborative Semantic-level and Token-level CoTcited by 166
- 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generationcited by 163
- 2024Lumina-mGPT: Illuminate Flexible Photorealistic Text-to-Image Generation with Multimodal Generative Pretrainingcited by 150
- 2025Teaching Large Language Models to Regress Accurate Image Quality Scores using Score Distributioncited by 141
- 2025Can We Generate Images with CoT? Let's Verify and Reinforce Image Generation Step by Stepcited by 131
- 2025Lumina-DiMOO: An Omni Diffusion Large Language Model for Multi-Modal Generation and Understandingcited by 88
- 2024Meissonic: Revitalizing Masked Generative Transformers for Efficient High-Resolution Text-to-Image Synthesiscited by 56
- 2024Improved Video VAE for Latent Video Diffusion Modelcited by 48
- 2024Benchmarking Large Vision-Language Models via Directed Scene Graph for Comprehensive Image Captioningcited by 31
- 2025The LLM Already Knows: Estimating LLM-Perceived Question Difficulty via Hidden Representationscited by 29
- 2024GREAT: Geometry-Intention Collaborative Inference for Open-Vocabulary 3D Object Affordance Groundingcited by 28
- 2026The Flexibility Trap: Rethinking the Value of Arbitrary Order in Diffusion Language Modelscited by 18
- 2025Group Critical-token Policy Optimization for Autoregressive Image Generationcited by 17
- 2025Reinforcement Learning Meets Masked Generative Models: Mask-GRPO for Text-to-Image Generationcited by 14
- 2025Consolidating Reinforcement Learning for Multimodal Discrete Diffusion Modelscited by 12
- 2025Towards Better & Faster Autoregressive Image Generation: From the Perspective of Entropycited by 12
- 2025Anchoring Values in Temporal and Group Dimensions for Flow Matching Model Alignmentcited by 6
- 2025MaskFocus: Focusing Policy Optimization on Critical Steps for Masked Image Generationcited by 6
- 2025See Different, Think Better: Visual Variations Mitigating Hallucinations in LVLMscited by 4
50 most cited of 52 All 52 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026From Broad Exploration to Stable Synthesis: Entropy-Guided Optimization for Autoregressive Image Generation
- 2026Optimizing Visual Generative Models via Distribution-wise Rewards
- 2025Accelerating Inference of Masked Image Generators via Reinforcement Learning
- 2025Learning What to Trust: Bayesian Prior-Guided Optimization for Visual Generation
- 2025Integrating Reinforcement Learning with Visual Generative Models: Foundations and Advances
- 2025Seeing What Matters: Visual Preference Policy Optimization for Visual Generation
- 2026VIGOR: Zero-Shot Visual Generalization via Latent-Space Consistency in Model-Based Reinforcement Learning
- 2026Step Back to Move Forward: Reflection-Aware Preference Optimization for Visual Generation