World Models in Words: Auditing Physical State-Transition Commitments in Vision-Language Models
This paper introduces \wmw, an evaluation framework and the \tracebank dataset that audit Vision-Language Models by requiring them to generate typed physical state-transition traces, thereby revealing hidden reasoning failures and enabling improvements in physical consistency that traditional answer-only benchmarks miss.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
Imagine you are hiring a student to solve a physics problem. In the old way of testing, you only looked at the final answer they wrote on the paper. If they got the right number, you gave them an A. If they got it wrong, you gave them an F.
The problem is, a student could get the right answer by pure luck, or by guessing, or by using a completely wrong logic that just happened to land on the correct number. You wouldn't know how they got there, only what they got.
This paper introduces a new way of testing AI models (specifically Vision-Language Models) called World Models in Words (WMW). Instead of just checking the final answer, the researchers force the AI to show its work step-by-step, like a detailed diary of its thoughts.
The New "Show Your Work" Test
The researchers ask the AI to produce a Trace, which is a four-part story:
- The Starting Scene (): What does the AI see right now? (e.g., "A ball is sitting on a ramp.")
- The Action (): What rule is happening? (e.g., "Gravity is pulling the ball down.")
- The Result (): What happens next? (e.g., "The ball rolls down the ramp.")
- The Answer (): The final conclusion.
The AI has to write this story in a very specific, structured format (like a JSON code block) so a computer can check it.
The "Hybrid Verifier" (The Strict Teacher)
Once the AI writes its story, a special computer program called a Hybrid Verifier reads it. This isn't just a simple spell-checker. It acts like a strict physics teacher who checks:
- Did you see the right objects? (Is that a ball or a box?)
- Is the physics real? (Does gravity pull up or down?)
- Does the story make sense? (If the ball rolls down, does it end up at the bottom?)
- Does the answer match the story? (If the story says the ball rolls left, but the answer says "Right," the AI is lying.)
If the AI gets the right answer but tells a fake story (e.g., "The ball rolled left because of magic"), the Verifier catches it. The paper calls this a "Hidden Inconsistency."
The Big Discovery: "The Lucky Guessers"
The researchers tested 7 different AI models. They found something surprising:
- Many models got the right final answer.
- But, for about 18% to 42% of those "correct" answers, the story the AI told to get there was physically impossible.
It's like a student who guesses the right answer on a math test but writes down a solution that involves dividing by zero. The old test would have given them an A; this new test reveals they don't actually understand the math.
The "World Bank" of Physics Stories
To make this test possible, the authors created a massive library called WMW-TRACEBANK.
- It contains 200 carefully crafted physics scenarios (like ramps, springs, and collisions).
- It has 3,200 pairs of "Good Stories" vs. "Bad Stories." In a "Bad Story," they changed just one tiny thing (like flipping a force direction) to see if the AI could spot the error.
- This library helps train AI to stop guessing and start telling the truth about how the physical world works.
Fixing the AI
The paper also tested ways to fix these "liars."
- Reranking: They let the AI generate 8 different stories and picked the one that passed the Verifier's check. This improved the quality of the stories without changing the final answers.
- Preference Training: They taught the AI to prefer "Good Stories" over "Bad Stories" using the 3,200 pairs. This reduced the number of hidden lies by 41%.
The Bottom Line
The main point of this paper is not to build a better physics calculator. It is to build a truth detector.
Before we trust AI to help us with real-world tasks (like planning for robots or teaching students), we need to know if their internal story of the world matches reality. This paper gives us a tool to audit that story, ensuring that when an AI says "The ball rolls down," it actually understands why and how, rather than just guessing the right word.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.