Test-Time Training for Modality Order Consistency in Vision-Language Models

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Vision-Language Models · Depth: Expert, quick

Summary

Vision-language models exhibit sensitivity to the presentation order of images and questions, a semantically irrelevant change. Across three distinct models and three benchmarks, image-first prompting consistently outperformed question-first prompting, revealing a repeatable modality order failure. Researchers designed an order-consistent test-time training method that substantially closes this modality-order gap across all evaluated settings. This method also surprisingly yields consistent performance improvements in the stronger image-first branch compared to the baseline, effectively bootstrapping both orderings towards mutual consistency. Activation patching localized the ordering failure to a narrow mid-network region where representations sharply diverge between prompt orders, a misalignment that the test-time training method successfully repairs across layers. These findings identify modality-order sensitivity as a circuit-level failure in VLMs, demonstrating that simple, asymmetric test-time adaptation can effectively mitigate it and enhance baseline performance.

Key takeaway

For Machine Learning Engineers deploying vision-language models, you should rigorously test your models for modality order sensitivity. If your VLM exhibits performance degradation with question-first inputs, consider implementing asymmetric test-time training. This approach can not only close the performance gap between input orderings but also consistently improve your model's overall accuracy beyond its baseline. Your deployment strategy should account for potential input order variations.

Key insights

Vision-language models exhibit a circuit-level failure due to input modality order, which test-time training can effectively mitigate.

Principles

Method

An order-consistent test-time training method adapts VLMs by repairing mid-network representation misalignment that arises from different input modality orders.

In practice

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.