mlsanityData to 14 Oct 2026

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

arXiv 2609.38485 ↗First posted 29 Sep 2026cs.CV cs.LGcited by 0

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  5. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  6. 2023Visual Instruction Tuningcited by 7,869
  7. 2017Neural Discrete Representation Learningcited by 4,712
  8. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  9. 2021Zero-Shot Text-to-Image Generationcited by 4,288
  10. 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
  11. 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,109
  12. 2023Evaluating Object Hallucination in Large Vision-Language Modelscited by 1,729
38 more
  1. 2023MME: A Comprehensive Evaluation Benchmark for Multimodal Large Language Modelscited by 1,652
  2. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  3. 2018On the Spectral Bias of Neural Networkscited by 1,341
  4. 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,295
  5. 2020Gradient Surgery for Multi-Task Learningcited by 1,256
  6. 2022MaskGIT: Masked Generative Image Transformercited by 1,099
  7. 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,023
  8. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
  9. 2025Emerging Properties in Unified Multimodal Pretrainingcited by 897
  10. 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
  11. 2018Multi-Task Learning as Multi-Objective Optimizationcited by 821
  12. 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 777
  13. 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 752
  14. 2023Language Model Beats Diffusion -- Tokenizer is Key to Visual Generationcited by 640
  15. 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 459
  16. 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
  17. 2019What Makes Training Multi-Modal Classification Networks Hard?cited by 344
  18. 2024LoRA Learns Less and Forgets Lesscited by 328
  19. 2024VILA-U: a Unified Foundation Model Integrating Visual Understanding and Generationcited by 295
  20. 2022Balanced Multimodal Learning via On-the-fly Gradient Modulationcited by 264
  21. 2024Playground v2.5: Three Insights towards Enhancing Aesthetic Quality in Text-to-Image Generationcited by 256
  22. 2025Show-o2: Improved Native Unified Multimodal Modelscited by 250
  23. 2022Multi-Task Learning as a Bargaining Gamecited by 215
  24. 2024TokenFlow: Unified Image Tokenizer for Multimodal Understanding and Generationcited by 202
  25. 2020Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropoutcited by 201
  26. 2023Making LLaMA SEE and Draw with SEED Tokenizercited by 185
  27. 2020Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Modelscited by 169
  28. 2024Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Modelscited by 163
  29. 2025UniTok: A Unified Tokenizer for Visual Generation and Understandingcited by 153
  30. 2024Rho-1: Not All Tokens Are What You Needcited by 133
  31. 2024Randomized Autoregressive Visual Generationcited by 127
  32. 2024LMFusion: Adapting Pretrained Language Models for Multimodal Generationcited by 120
  33. 2018Adapting Auxiliary Losses Using Gradient Similaritycited by 100
  34. 2022In Defense of the Unitary Scalarization for Deep Multi-Task Learningcited by 67
  35. 2022Do Current Multi-Task Optimization Methods in Deep Learning Even Help?cited by 60
  36. 2025DualToken: Towards Unifying Visual Understanding and Generation with Dual Visual Vocabulariescited by 42
  37. 2025SemHiTok: A Unified Image Tokenizer via Semantic-Guided Hierarchical Codebook for Multimodal Understanding and Generationcited by 21
  38. 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.