Pareto LoRA boosts multimodal model image quality by 44.9%
Language gradients dominate training, but Pareto LoRA rebalances text-image generation for stunning gains.
Unified multimodal models (UMMs) that handle both text and image generation often suffer from a severe modality imbalance during parameter-efficient fine-tuning (PEFT) like LoRA. The authors show that language gradients dominate the optimization process, causing vision performance to degrade substantially more than text when compared to unimodal counterparts. In some layers, modality-specific gradients differ by orders of magnitude, leading to poor image generation quality even as text remains strong.
To solve this, the team reformulates multimodal instruction tuning as a bi-objective optimization problem and proposes Pareto LoRA. This strategy integrates gradients from text and image objectives by adjusting both direction and magnitude to find a Pareto-optimal balance. Tested on the CoMM benchmark with the Emu2 architecture, Pareto LoRA achieved up to 44.9% gains in perceptual image quality (e.g., FID and CLIP scores) while keeping text performance on par with vanilla LoRA. This approach offers a principled way to boost multimodal generation without sacrificing efficiency.
- Language gradients can dominate LoRA fine-tuning, causing up to 44.9% worse image quality compared to unimodal baselines.
- Pareto LoRA treats multimodal training as a bi-objective problem, modulating gradient direction and strength for balanced learning.
- Achieves 44.9% improvement in perceptual image quality on CoMM benchmark with Emu2, with no loss in text performance.
Why It Matters
Better multimodal generation means AI can produce coherent text and high-quality images, enabling more capable chatbots and creative tools.