From Prompts to Tokens: Internalizing Causal Supervision in Vision-Language Model for Multi-Image Causal Reasoning
Abstract
Lay Summary
AI systems that can look at images and answer questions have become increasingly capable, but they still struggle with cause-and-effect reasoning. For example, if two images show the same scene before and after a change, the system may need to identify what was changed and predict how that change should affect other parts of the scene. Current models often receive this kind of causal information only as long written prompts, but they may not reliably connect those instructions to the visual evidence. We introduce BridgeVLM, a method that gives such models an internal representation of the important variables in an image and how they are related. Instead of treating causal knowledge only as text, BridgeVLM uses it to shape internal representations that the model can consult when answering questions. This makes the model better at intervention and counterfactual reasoning across several visual causal reasoning benchmarks. Our results show that BridgeVLM improves over strong baselines, especially when causal supervision is provided internally rather than only through prompts.