MIRROR: Learning from the Other View for Multi-Modal Reasoning

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

MIRROR (Modality-Informed Reciprocal Reasoning Optimization) is a new reinforcement learning approach designed to enhance vision-language models' (VLMs) reasoning capabilities, particularly for geometry problems. Unlike large language models (LLMs) which excel at reasoning, VLMs often exhibit inconsistent performance across different input modalities—text, diagram, or combined views—for the same problem. This inconsistency suggests complementary reasoning paths. To address this, researchers developed ODA-Data, a high-quality paired multimodal geometry dataset. MIRROR leverages this by evaluating a model across all views for a given problem, identifying the best-performing view as a "teacher," and then training the other views using a reverse-KL objective to align with the teacher's superior performance. This self-supervised method significantly improves over standard reinforcement learning, leading to more accurate and consistent multimodal reasoning behavior on geometry benchmarks.

Key takeaway

For Machine Learning Engineers developing vision-language models for complex reasoning tasks, particularly geometry, standard multimodal post-training may not suffice. You should recognize that different input modalities expose complementary reasoning paths and inconsistencies. Implementing a self-supervised approach, such as MIRROR's method of training weaker modalities towards a stronger "teacher" view using a reverse-KL objective, can significantly improve reasoning accuracy and consistency across modalities in your models.

Key insights

MIRROR improves VLM reasoning consistency by using self-supervision from a problem's best-performing modality view.

Principles

Method

MIRROR evaluates a model across all problem views, selects the best-performing view as a teacher, and trains other views with a reverse-KL objective.

In practice

Topics

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.