LLaVAFlow: Preserving Latent Alignment Flow for Parameter-Efficient Multimodal Fine-Tuning
Authors: Muyao Yuan, Muyan Jiao, Jiangyong Ying, Weizhan Zhang, Yuanhong Zhang, Lan Ma, Yuan Gao, Haipeng Du
Organizations: MOEKLINNS, and School of Computer Science and Technology Xi’an Jiaotong University Xi’an, China · E-surfing Vision Technology Co., Ltd China Telecom Hangzhou, China · China Telecom Xi’an, China
While Multimodal Large Language Models (MLLMs) exhibit strong generalization, visual instruction tuning for downstream tasks inevitably causes catastrophic forgetting, impairing overall generalization. While existing methods regulate weight updates to reduce forgetting, they overlook the fundamental cross-modal alignment in MLLMs. Based on prior work and our observations, we argue that cross-modal alignment is implicitly captured in the information-compression trajectory. To preserve the alignment flow embedded in the trajectory, we propose LLaVAFlow, an information-theoretic distillation framework. First, we compress the mutual information between the extracted relations and MLLM embeddings, encouraging a learnable module to produce a refined alignment flow that benefits downstream tasks. Second, we maximize the mutual information between the extracted alignment flows of the pretrained and fine-tuned MLLMs, enabling the transfer of compact alignment information. Extensive experiments show that LLaVAFlow is an effective plug-and-play framework that preserves alignment flow and enhances both downstream performance and generalization.
Figures & tables
Figure 1. Motivation. We map visual tokens to semantic words along the forward pass using Logit Lens ( Nostalgebraist, 2020 ) . Existing methods ( Liang et al., 2025 ) disrupt the alignment within the feature flow, leading to incorrect predictions. By preserving the alignment flow, we effectively mitigate forgetting.
Figure 2. Overview of LLaVAFlow. LLaVAFlow is a distillation framework that effectively transfers alignment flow from a pretrained MLLM to its fine-tuned counterpart from an information-theoretic perspective. The framework comprises an Alignment Flow Module (AFM) and an Information Bottleneck Distillation (IBD) objective that jointly regulate MLLM fine-tuning: the Flow Refinement Loss drives the AFM to capture refined alignment flow, while the Alignment Preservation Loss effectively transfers it to the student. The T-MLLM (teacher) and S-MLLM (student) share the same AFM parameters.
Methods
ScienceQA
COCO-Caption
OKVQA
OCRVQA
GQA
TextVQA
Source
Target
Avg
OKVQA
OCRVQA
GQA
TextVQA
Source
Target
Avg
Zero-shot
58.00
66.20
61.94
58.21
61.09
70.22
65.65
58.00
66.20
61.94
58.21
61.09
40.34
50.71
LoRA Rank = 32
LoRA
56.99
65.25
61.52
56.64
60.10
87.57
73.84
48.60
66.00
56.73
43.91
53.81
128.46
91.14
LoRA+Ours
56.94
65.65
61.19
57.07
60.21
89.46
74.84
52.18
67.25
59.33
51.20
57.49
130.04
93.76
DoRA
55.10
61.05
59.68
54.18
57.50
88.94
73.22
49.69
66.50
57.62
45.81
54.90
128.26
91.58
Table 1. Comparison with state-of-the-art methods on the visual question answering task ScienceQA and the image captioning task COCO-Caption. Source indicates the average performance across upstream datasets, Target reports the performance on the fine-tuned dataset, and Avg is the mean of Source and Target .
Figure 3. Alignment trends across layers. (a) and (b) show the average cosine similarity between features at the original vision and text embedding positions within each layer on the target and source datasets, respectively. (c) and (d) show the corresponding normalized Euclidean distance. All fine-tuned models are fine-tuned on ScienceQA ( Lu et al., 2022 ) . Our model preserves the pre-trained modality alignment in the shallow layers (1–25), indicating retention of general cross-modal alignment, while exhibiting divergence in the deeper layers as they adapt to task-specific objectives.
Figure 4. Comparison of semantic trajectories across models, obtained via Logit Lens ( Nostalgebraist, 2020 ) . Existing methods, such as LoRASculpt, fail to preserve alignment within the feature flow, leading to incorrect predictions. In contrast, our method retains most of the trajectory of the pre-trained LLaVA, effectively mitigating catastrophic forgetting. The trajectory is derived from the image patch indicated by the red box. Full trajectories are provided in Appendix B.1.
LFR
LAP
LoRA Rank =32
LoRA Rank =64
Target
Source
Avg
Target
Source
Avg
Zero-shot
70.22
61.09
65.65
70.22
61.09
65.65
✗
✗
87.57
60.10
73.84
88.49
55.67
72.08
✗
✓
89.08
60.04
74.56
87.86
58.71
73.28
✓
✗
89.37
60.05
74.71
88.52
57.11
72.81
✓
✓
89.46
60.21
74.84
89.46
58.13
73.79
Table 2. Ablation study of Flow Refinement Loss LFR and Alignment Preservation Loss LAP . All experiments are conducted with LoRA on the ScienceQA dataset.
Method
LoRA Rank =32
LoRA Rank =64
Target
Source
Avg
Target
Source
Avg
Zero-shot
70.22
61.09
65.65
70.22
61.09
65.65
Scale
88.92
59.97
74.45
87.97
57.18
72.58
Linear
88.73
59.34
74.03
87.55
34.99
61.27
Attn-2
89.39
59.70
74.54
87.34
52.02
69.68
Attn-3
88.23
60.12
74.18
86.68
38.61
62.64
Table 3. Comparison of different alignment flow modeling strategies. All experiments are conducted with LoRA on the ScienceQA dataset. Attn- L denotes stacking L−1 standard attention layers followed by AFM as the final scoring layer.
Figure 5. Hyperparameter analysis of λFR and λAP . All results are obtained with LLaVAFlow applied.
Multimodal Large Language Models (MLLMs) rely on a projector to align visual representations with the language embedding space, making it central to cross-modal understanding. In Multimodal Continual Instruction Tuning (MCIT), however, shifting visual distributions and evolving instruction semantics cause this shared projector to drift, leading to projector-level forgetting, an issue largely overlooked by methods that focus primarily on the LLM backbone. We introduce Progressive Multimodal Alignment (PMA), a framework that enables the projector to adapt continually while preserving previously learned alignment. PMA detects multimodal distribution shifts via a lightweight representation descriptor and progressively expands projector experts only when needed. An expandable router integrates expert outputs based on multimodal features, while the original pretrained projector is retained as a stable alignment anchor. This progressive mechanism balances stability and plasticity with sub-linear parameter growth and serves as a method-agnostic add-on to existing MCIT approaches. Extensive experiments on two recent MCIT benchmarks demonstrate that mitigating projector-level forgetting yields consistent gains over prior state-of-the-art methods when combined with PMA. Moreover, PMA scales across diverse MLLM backbones, demonstrating robust and broadly applicable MCIT performance.
Duzhen Zhang, Yahan Yu, Qiaoyi Su +2
Mohamed bin Zayed University of Artificial Intelligence · Center for Excellence in Brain Science and Intelligence Technology, Chinese Academy of Sciences · Kyoto University +2
Multimodal Continual Instruction Tuning (MCIT) is essential for sequential task adaptation of Multimodal Large Language Models (MLLMs) but is severely restricted by catastrophic forgetting. While existing literature focuses on the reasoning language backbone, in this work, we expose a critical yet neglected dual-forgetting phenomenon across both perception drift in Cross-modal Projection Space and reasoning collapse in Low-rank Parameter Space. To resolve this, we present MAny (Merge Anything), a framework that merges task-specific knowledge through Cross-modal Projection Merging (CPM) and Low-rank Parameter Merging (LPM). Specifically, CPM recovers perceptual alignment by adaptively merging cross-modal visual representations via visual-prototype guidance, ensuring accurate feature recovery during inference. Simultaneously, LPM eliminates mutual interference among task-specific low-rank modules by recursively merging low-rank weight matrices. By leveraging recursive least squares, LPM provides a closed-form solution that mathematically guarantees an optimal fusion trajectory for reasoning stability. Notably, MAny operates as a training-free paradigm that achieves knowledge merging via efficient CPU-based algebraic operations, eliminating additional gradient-based optimization beyond initial tuning. Our extensive evaluations confirm the superior performance and robustness of MAny across multiple MLLMs and benchmarks. Specifically, on the UCIT benchmark, MAny achieves significant leads of up to 8.57% and 2.85% in final average accuracy over state-of-the-art methods across two different MLLMs, respectively.
Zijian Gao, Wangwang Jia, Xingxing Zhang +6
College of Computer Science and Technology, National University of Defense Technology · State Key Laboratory of Complex & Critical Software Environment · National Key Laboratory of Parallel and Distributed Computing, National University of Defense Technology +1
Fine-tuning Multimodal Large Language Models (MLLMs) on specialized tasks often leads to catastrophic forgetting of their general capabilities. Existing model merging methods to combat this are often heuristic or use sub-optimal objectives. We propose CurvatureGuided Mixing (CGM), a theoretically grounded framework that merges pre-trained and fine-tuned models. CGM formulates a joint optimization objective and uses a second-order (Hessian) approximation of the loss landscapes to analytically derive an optimal, closed-form "soft mixing" ratio. This ratio intelligently blends parameters based on their relative task-specific curvatures. We also introduce CGM†, a robust "hard mixing" variant that performs sparse parameter selection guided by a novel, curvature-aware score. Experiments on LLaVA-1.5 and Qwen2.5VL across multiple downstream tasks show that CGM and CGM† consistently improve the trade-off between task specialization and general knowledge retention over existing methods. Code is available at github.com/zzsyjl/CGM-ECCV-2026.
Jinglong Yang, Jiaxuan He, Wenjian Huang +2
Research Institute of Trustworthy Autonomous Systems and Department of Computer Science and Engineering, Southern University of Science and Technology · Department of Computer Science, City University of Hong Kong