Focusing Where Vision Matters: Selective Training for Large Vision Language Models via Visual Information Gain
Organizations: Department of Data Science, Seoul National University of Science and Technology, Seoul, Republic of Korea. · Currently with Tomocube, Daejeon, Republic of Korea.
Abstract
Large Vision Language Models (LVLMs) have achieved remarkable progress, yet they often suffer from language bias, producing answers without relying on visual evidence. While prior work attempts to mitigate this issue through decoding strategies, architectural modifications, or curated instruction data, they typically lack a quantitative measure of how much individual training samples or tokens actually benefit from the image. In this work, we introduce Visual Information Gain (VIG), a perplexity-based metric that measures the reduction in prediction uncertainty provided by visual input. VIG enables fine-grained analysis at both sample and token levels, effectively highlighting visually grounded elements such as colors, spatial relations, and attributes. Leveraging this, we propose a VIG-guided selective training scheme that prioritizes high-VIG samples and tokens. This approach improves visual grounding and mitigates language bias, achieving superior performance with significantly reduced supervision by focusing exclusively on visually informative samples and tokens.
Figures & tables
| Red | white (3.59), black (6.08), lying (5.99), flying (5.73) sitting (4.22), standing (5.10), reading (5.974), crowd (3.30) |
| Blue | a (0.03), of (0.01), the (0.03), ize (0.00), which (0.00), are (-0.02), The (0.04), - (-1.10) |
| Model | # Sample Tokens | # Active Tokens | Vision Understanding | Hallucination | |||||||||
| LLaVA | MMVet | MMBench | CV-Bench | DocVQA | POPE | CHAIR | MMHal | ||||||
| Score | Acc. | F1 | Acc. | Score | Hall. | ||||||||
| LLaVA-1.5 7B | 58.61M | 58.61M | 59.02 | 28.62 | 65.46 | 59.18 | 22.31 | 85.90 | 87.08 | 52.93 | 14.99 | 1.71 | 71.25 |
| + VIG training | 51.17M | 38.45M | 61.22 | 32.71 | 66.33 | 62.48 | 22.51 | 85.93 | 87.47 | 47.00 | 12.80 | 2.23 | 62.78 |
| -13% | -34% | +2.20 | +4.09 | +0.87 | +3.30 | +0.20 | +0.03 | +0.39 | +5.93 | +2.19 | +0.52 | +8.47 | |
| LLaVA-1.5 13B | 58.61M | 58.61M | 72.01 | 36.19 | 67.52 | 60.16 | 24.08 | 85.72 | 87.05 | 51.96 | 13.22 | 2.05 | 67.09 |
| Model | Vision Understanding | Hallucination | ||||||||
| LLaVA | MMVet | MMBench | DocVQA | POPE | CHAIR | MMHal | ||||
| Score | Acc. | F1 | Acc. | Score | Hall. | |||||
| LLaVA-1.5 7B | 59.02 | 28.62 | 65.46 | 22.31 | 85.90 | 87.08 | 52.93 | 14.99 | 1.71 | 71.25 |
| + VCD | 60.55 | 27.01 | 64.34 | 22.98 | 86.53 | 86.61 | 49.81 | 13.69 | 1.67 | 76.01 |
| + PAI | 57.54 | 27.99 | 65.45 | 21.98 | 85.98 | 85.84 | 35.46 | 12.11 | 1.78 | 72.31 |
| + VAR | 61.11 | 30.98 | 66.30 | 23.11 | 86.10 | 87.12 | 50.09 | 14.77 | 2.14 | 62.89 |
| Model | GSM8K | MMLU | HellaSwag | TruthfulQA |
| LLaVA 1.5 7B | 15.33 | 51.68 | 76.09 | 45.86 |
| + VIG training | 14.99 | 51.35 | 76.11 | 46.01 |
| Model | LLaVA | MMBench | CHAIR | MMHal |
| LLaVA1.5-7B | 59.02 | 65.46 | 52.93 | 71.25 |
| Random | 56.91 | 55.97 | 54.88 | 74.49 |
| SS | 58.12 | 57.56 | 50.23 | 68.14 |
| SS+TS | 61.19 | 66.33 | 49.10 | 61.82 |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | LLaVA-1.5 7B | LLaVA-1.5 13B | ShareGPT4V 7B | Open-Qwen2VL 2B |
| Threshold |
| Model | Epoch | Batch size | LR | LR schedule | LR warmup ratio | Optimizer | Training time | |
| Pretraining | ShareGPT4V | 1 | 256 | 2e-5 | cosine decay | 0.03 | AdamW | 12 hours |
| Instruction Tuning | LLaVA-1.5 Family, ShareGPT4V | 1 | 128 | 2e-5 | cosine decay | 0.03 | AdamW | 7 hours |
| Open-Qwen2VL 2B | 1 | 128 | 2e-5 | linear-warmup + cosine decay | 0.03 | AdamW | 10 hours |
| User | Describe this photo in detail. | |
| LLaVA-1.5 7B | Vanilla | The image features a painting of a dog dressed up as a woman, wearing a dress and a hat. The dog is the main focus of the painting, and it appears to be the subject of the artwork. The painting is set in a room with a dining table visible in the background. There are also two people in the scene, one on the left side and another on the right side of the painting. The person on the left is partially visible, while the person on the right is more prominently featured in the scene. |
| +VIG Training | The image features a painting of a dog wearing a Renaissance-style dress , complete with a hood and a cape. The dog is sitting in a pose that resembles a woman , giving the impression of a human-like figure. The dog’s attire and the way it is positioned in the painting create a unique and intriguing scene. | |
| LLaVA-1.5 13B | Vanilla | The image is a painting featuring a dog dressed up as Mona Lisa, wearing a bandana on its head and holding its hands together, posing for the portrait. The background includes mountains and a river, adding to the composition’s artistic appeal. In addition to the dog, there are three people present in the scene. One person is positioned near the left edge, another person can be seen at the bottom right corner, and the third person is located on the right side of the painting. |
| +VIG Training | The image features a painting of a dog dressed as the famous painting of the Mona Lisa. The dog is wearing a bandana on its head and is positioned in a way that resembles the iconic pose of the Mona Lisa . The painting is quite detailed, capturing the essence of the original artwork while incorporating the unique twist of the dog as the subject. The dog’s facial expression and posture are reminiscent of the Mona Lisa’s enigmatic smile and pose, making the painting a creative and amusing take on the classic artwork . | |
| ShareGPT4V 7B | Vanilla | The image presents a surrealistic portrait of a dog, painted in the style of the Renaissance period. The dog, the central figure of the painting, is adorned with a red hat and a blue robe, adding a touch of whimsy to the otherwise realistic style. The background is a landscape, painted in a realistic style, with a river meandering through it, adding depth to the composition. The painting is executed in oil on canvas, a medium that allows for rich colors and intricate details. The genre of the painting is surrealism, a style that combines elements of the real and the imaginary, creating a dreamlike quality. The painting is a testament to the artist’s skill in blending realism with surrealism, creating a unique and captivating piece of art. |
| User | Is this airplane taking off or landing? | |
| LLaVA-1.5 7B | Vanilla | Based on the image, it is not possible to definitively determine whether the airplane is taking off or landing. However, we can observe that the airplane is in the process of either ascending or descending, as it is flying low over the runway. The presence of smoke coming from the airplane’s wheels suggests that it could be either taking off or landing. To determine the exact phase of the flight, we would need more information or a clearer view of the airplane’s position relative to the ground. |
| +VIG Training | The airplane is taking off from the runway . | |
| LLaVA-1.5 13B | Vanilla | The airplane in the image is taking off from the runway. |
| +VIG Training | The airplane in the image is taking off from the runway. | |
| ShareGPT4V 7B | Vanilla | The airplane is taking off from the runway. |
| Model | Base | Corruption | Norm |
| LLaVA-1.5 7B (Baseline) | 77.9 | 32.1 | 41.2 |
| + VCD | 78.1 | 33.5 | 42.9 |
| + VIG training | 78.2 | 42.3 | 54.1 |
| Model | GSM8K | MMLU | HellaSwag | TruthfulQA |
| LLaVA 1.5 13B | 15.33 | 51.68 | 76.09 | 45.86 |
| + VIG training | 14.99 | 51.35 | 76.11 | 46.01 |
| ShareGPT4V 7B | 18.01 | 47.62 | 54.78 | 46.53 |
| + VIG training | 18.03 | 47.66 | 56.14 | 46.42 |
| Model | Vision Understanding | Hallucination | ||||||||
| LLaVA | MMVet | MMBench | DocVQA | POPE | CHAIR | MMHal | ||||
| Score | Acc. | F1 | Acc. | Score | Hall. | |||||
| LLaVA-1.5 7B | 59.02 | 28.62 | 65.46 | 22.31 | 85.90 | 87.08 | 52.93 | 14.99 | 1.71 | 71.25 |
| Random | 58.50 | 27.27 | 55.97 | 21.38 | 85.48 | 86.63 | 53.11 | 15.00 | 1.79 | 71.33 |
| SS | 58.71 | 32.29 | 57.56 | 22.06 | 85.22 | 86.53 | 52.10 | 14.61 | 1.84 | 63.54 |
| SS+TS | 61.22 | 32.71 | 66.33 | 22.51 | 85.93 | 87.47 | 47.00 | 12.80 | 2.23 | 62.78 |
| Ratio | 30 | 50 | 70 |
| Threshold |
| Ratio | Vision Understanding | Hallucination | ||||||||||
| # Sample Tokens | # Active Tokens | LLaVA | MMVet | MMBench | DocVQA | POPE | CHAIR | MMHal | ||||
| Score | Acc. | F1 | Acc. | Score | Hall. | |||||||
| 30% | 8.58M | 3.19M | 60.90 | 34.03 | 56.87 | 22.66 | 82.80 | 84.28 | 48.71 | 13.31 | 1.99 | 63.01 |
| 50% | 26.32M | 10.00M | 61.81 | 31.19 | 56.44 | 22.49 | 85.09 | 86.14 | 47.63 | 12.99 | 1.95 | 62.91 |
| 70% | 51.17M | 38.45M | 61.22 | 32.71 | 66.33 | 22.51 | 85.93 | 87.47 | 47.00 | 12.80 | 2.23 | 62.78 |
| 100% | 58.61M | 58.61M | 59.02 | 28.62 | 65.46 | 22.31 | 85.90 | 87.08 | 52.93 | 14.99 | 1.71 | 71.25 |
Vision-Language Models
Vision-language models connect visual inputs with text for retrieval, question answering, generation, and grounded multimodal reasoning.
See all 724 papers