Unraveling MMDiT Blocks: Training-free Analysis and Enhancement of Text-conditioned Diffusion

TL;DR

This paper systematically analyzes MMDiT-based diffusion models to understand their internal mechanisms and proposes training-free strategies to enhance text alignment, editing precision, and inference speed, leading to improved performance across multiple benchmarks.

Contribution

It introduces a comprehensive, training-free analysis pipeline for MMDiT models and proposes novel enhancement strategies based on these insights.

Findings

Semantic info appears in early blocks

Finer details are in later blocks

Enhancing text conditions improves semantic attributes

Abstract

Recent breakthroughs of transformer-based diffusion models, particularly with Multimodal Diffusion Transformers (MMDiT) driven models like FLUX and Qwen Image, have facilitated thrilling experiences in text-to-image generation and editing. To understand the internal mechanism of MMDiT-based models, existing methods tried to analyze the effect of specific components like positional encoding and attention layers. Yet, a comprehensive understanding of how different blocks and their interactions with textual conditions contribute to the synthesis process remains elusive. In this paper, we first develop a systematic pipeline to comprehensively investigate each block's functionality by removing, disabling and enhancing textual hidden-states at corresponding blocks. Our analysis reveals that 1) semantic information appears in earlier blocks and finer details are rendered in later blocks, 2) removing specific blocks is usually less disruptive than disabling text conditions, and 3) enhancing textual conditions in selective blocks improves semantic attributes. Building on these observations, we further propose novel training-free strategies for improved text alignment, precise editing, and acceleration. Extensive experiments demonstrated that our method outperforms various baselines and remains flexible across text-to-image generation, image editing, and inference acceleration. Our method improves T2I-Combench++ from 56.92% to 63.00% and GenEval from 66.42% to 71.63% on SD3.5, without sacrificing synthesis quality. These results advance understanding of MMDiT models and provide valuable insights to unlock new possibilities for further improvements.