cs.CVSep 29, 2026

Beyond Layers: Position-Resolved Gradient Conflict and Position-Aware Modulation for Unified Multimodal Models

Authors: Shuyang Jiang, Fucheng Deng, Yuchuan Luo, Zhenyu Wu

Organizations: University of California, Los Angeles · Aimakj · College of Computer Science and Technology National University of Defense Technology · Key Laboratory of Advanced Microprocessor Chips and Systems College of Computer Science and Technology National University of Defense Technology

Abstract

Unified multimodal models (UMMs) train image understanding and autoregressive image generation on shared parameters, and the two objectives are known to interfere. Existing diagnoses and remedies operate at the resolution of layers or experts, measuring conflict per layer and resolving it by separating parameters. We argue that this resolution hides an orthogonal axis. Generation in a UMM is next-token prediction over a raster sequence of visual tokens whose roles vary systematically with position, so how strongly a generation gradient interferes with understanding should depend on where in the sequence it originates. We introduce a position-resolved interference map that attributes understanding-generation gradient conflict to visual-token positions within every layer, computed from a single backward pass at 1.2×1.2\times the cost of a standard backward pass. On Show-o and Janus-Pro, position explains a large share of conflict variance after controlling for depth (partial η2=0.31η^2=0.31 vs. 0.350.35 for layer on Show-o; 0.150.15 vs. 0.300.30 on Janus-Pro): the first quarter of the sequence has a mean gradient cosine of −0.18-0.18 against understanding, the last quarter −0.02-0.02. The dependence survives per-position gradient-norm normalization, retaining 8080% of its effect size, and conflict strength tracks semantic content (Spearman ρ=0.64ρ=0.64). Building on the map, we propose position-aware modulation (PAM), which removes the anti-aligned component of generation gradients only at high-conflict positions without changing the architecture. Under a matched trainable-parameter budget, PAM improves over layer-wise separation by +21+21 MME and +2.4+2.4 GenEval points on Show-o while matching it on POPE and overall FID; a random-position control recovers about 3131% of the gain. Position-based and layer-based separation are complementary degrees of freedom and can be combined.

Figures & tables

Appendix figures & tables5 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Does Gradient Conflict Predict the Understanding--Generation Trade-off? A Controlled Audit of Conflict-Metric Validity in Unified Multimodal Models

    Sep 29, 2026Shuyang Jiang, Fucheng Deng, Yuchuan Luo +1Dose

  2. Uncovering Understanding-Generation Synergy in Native Unified Multimodal Models: From Representation, Task to System

    Sep 1, 2026Penghao Wu, Haiwen Diao, Weichen Fan +3Data Generation

  3. Representation Forcing for Bottleneck-Free Unified Multimodal Models

    May 29, 2026Yuqing Wang, Zhijie Lin, Ceyuan Yang +10Multimodal ModelVideo Latents