Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

MergeMix: A Unified Augmentation Paradigm for Visual and Multi-Modal Understanding

About

Vision-language alignment in multi-modal large language models (MLLMs) relies on supervised fine-tuning (SFT) or reinforcement learning (RL). To align multi-modal large language models (MLLMs) in the post-training stage, supervised fine-tuning (SFT) is a stable choice but requires human annotations and lacks task generalizations, while Reinforcement Learning (RL) searches for better answers from reward signals but suffers from computational overhead and instability. To achieve balance among scalability, efficiency, and alignment generalizations, we propose MergeMix, a unified paradigm that bridges SFT and RL with an efficient Token Merge based Mixup augmentation. As for the Mixup policy, we generate contextual aligned mixed images with the corresponding labels according to the merged attention maps with cluster regions. Then, we enhance the preference-driven paradigm for MLLMs by building preference pairs with raw images and MergeMix-generated ones and optimizing the soft preference margin with the mixed SimPO loss. Extensive experiments demonstrate that MergeMix not only achieves dominant classification accuracy as an augmentation method but also improves generalization abilities and alignment of MLLMs, providing a new learning paradigm for preference alignment with training efficiency and stability.

Xin Jin, Siyuan Li, Siyong Jian, Kai Yu, Huan Wang• 2025

Related benchmarks

TaskDatasetResultRank
Object Hallucination EvaluationPOPE
Accuracy87.28
935
Image ClassificationCIFAR-100 (val)
Accuracy84.3
661
Image ClassificationCIFAR-100
Top-1 Accuracy81.44
622
Fine-grained Image ClassificationStanford Cars (test)
Accuracy92.2
348
Image ClassificationStanford Cars (test)--
306
Fine-grained visual classificationFGVC-Aircraft (test)
Top-1 Acc81.97
287
Model CalibrationCIFAR-100
ECE4.65
53
Fine-grained Image ClassificationCUB-200 (test)
Accuracy88.4
45
Visual Question AnsweringScienceVQA
ECE23.66
36
Image ClassificationCIFAR100 (test)
Top-1 Accuracy (0% Corruption)81.66
32
Showing 10 of 23 rows

Other info

Follow for update