Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimization

This is an official checkpoint from the paper: "Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimization " (link). See the official implementation for more information on how to use the models.

qwen2_2b_lora_expert_generalv2-102400

This model is a fine-tuned version of Qwen/Qwen2-VL-2B on a custom dataset with General VQA data (~100k samples).

It achieves the following results on the evaluation set:

Loss: 0.9787

Training hyperparameters

The following hyperparameters were used during training:

learning_rate: 2e-05
train_batch_size: 8
eval_batch_size: 8
seed: 42
distributed_type: multi-GPU
num_devices: 4
gradient_accumulation_steps: 4
total_train_batch_size: 128
total_eval_batch_size: 32
optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
lr_scheduler_type: cosine
lr_scheduler_warmup_ratio: 0.1
training_steps: 800

Training results

Training Loss	Epoch	Step	Validation Loss
1.0197	0.125	100	1.0769
0.9267	0.25	200	1.0109
0.9713	0.375	300	0.9950
0.9133	0.5	400	0.9865
0.9316	0.625	500	0.9828
0.9402	0.75	600	0.9797
0.93	0.875	700	0.9787
0.9383	1.0	800	0.9787