Beyond Layers: Position-Resolved Gradient Conflict and Position-Aware Modulation for Unified Multimodal Models
Not on the map. Why?
Abstract (arXiv)
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× 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 vs. 0.35 for layer on Show-o; 0.15 vs. 0.30 on Janus-Pro): the first quarter of the sequence has a mean gradient cosine of -0.18 against understanding, the last quarter -0.02. The dependence survives per-position gradient-norm normalization, retaining 80% of its effect size, and conflict strength tracks semantic content (Spearman ρ=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 MME and +2.4 GenEval points on Show-o while matching it on POPE and overall FID; a random-position control recovers about 31% of the gain. Position-based and layer-based separation are complementary degrees of freedom and can be combined.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 62 of its 71 references that are in the map. First posted on arXiv 29 Sep 2026.
Builds on
62 of 71 references.
References · 62
- 2017Attention Is All You Needcited by 60,361
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2023Visual Instruction Tuningcited by 7,869
- 2017Neural Discrete Representation Learningcited by 4,712
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2021Zero-Shot Text-to-Image Generationcited by 4,288
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,109
- 2023Evaluating Object Hallucination in Large Vision-Language Modelscited by 1,729
38 more
- 2023MME: A Comprehensive Evaluation Benchmark for Multimodal Large Language Modelscited by 1,652
- 2016Understanding intermediate layers using linear classifier probescited by 1,353
- 2018On the Spectral Bias of Neural Networkscited by 1,341
- 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,295
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2022MaskGIT: Masked Generative Image Transformercited by 1,099
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,023
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
- 2025Emerging Properties in Unified Multimodal Pretrainingcited by 897
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 821
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 777
- 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 752
- 2023Language Model Beats Diffusion -- Tokenizer is Key to Visual Generationcited by 640
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 459
- 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
- 2019What Makes Training Multi-Modal Classification Networks Hard?cited by 344
- 2024LoRA Learns Less and Forgets Lesscited by 328
- 2024VILA-U: a Unified Foundation Model Integrating Visual Understanding and Generationcited by 295
- 2022Balanced Multimodal Learning via On-the-fly Gradient Modulationcited by 264
- 2024Playground v2.5: Three Insights towards Enhancing Aesthetic Quality in Text-to-Image Generationcited by 256
- 2025Show-o2: Improved Native Unified Multimodal Modelscited by 250
- 2022Multi-Task Learning as a Bargaining Gamecited by 215
- 2024TokenFlow: Unified Image Tokenizer for Multimodal Understanding and Generationcited by 202
- 2020Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropoutcited by 201
- 2023Making LLaMA SEE and Draw with SEED Tokenizercited by 185
- 2020Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Modelscited by 169
- 2024Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Modelscited by 163
- 2025UniTok: A Unified Tokenizer for Visual Generation and Understandingcited by 153
- 2024Rho-1: Not All Tokens Are What You Needcited by 133
- 2024Randomized Autoregressive Visual Generationcited by 127
- 2024LMFusion: Adapting Pretrained Language Models for Multimodal Generationcited by 120
- 2018Adapting Auxiliary Losses Using Gradient Similaritycited by 100
- 2022In Defense of the Unitary Scalarization for Deep Multi-Task Learningcited by 67
- 2022Do Current Multi-Task Optimization Methods in Deep Learning Even Help?cited by 60
- 2025DualToken: Towards Unifying Visual Understanding and Generation with Dual Visual Vocabulariescited by 42
- 2025SemHiTok: A Unified Image Tokenizer via Semantic-Guided Hierarchical Codebook for Multimodal Understanding and Generationcited by 21
- 2025Uni-X: Mitigating Modality Conflict with a Two-End-Separated Architecture for Unified Multimodal Modelscited by 18
50 most cited of 62 All 62 (JSON)
Built on by
Not cited yet.