MAG Boosts Multi-modal In-Context Learning with Unlabeled Data.

Zirui Cheng, Xun Xu, Tiankai Chen, Fady Rezk, Bowen Zheng, Xiaodong Shi, Shijie Li, Kangkang Lu, Bharadwaj Veeravalli, Nancy F. Chen· August 14, 2026 View original

Key takeaways

  • MAG improves multi-modal in-context learning by using unlabeled data for demonstration selection.
  • It employs a two-stage process: relevance propagation for pseudo-labeling, then multi-modal selection.
  • Textual representations are key for initial propagation, while both modalities are vital for selection.
  • MAG significantly boosts performance in label-scarce multi-modal benchmarks.

Who benefits

MarketingE-commerceContent CreationHealthcareRobotics

Summary

This paper introduces MAG, a semi-supervised framework that enhances multi-modal in-context learning (ICL) by leveraging abundant unlabeled data for demonstration selection, significantly outperforming baselines in label-scarce regimes by efficiently identifying high-impact samples for pseudo-labeling.

Multi-modal large language models (MLLMs) can adapt to new tasks with few-shot in-context learning (ICL), but their performance heavily depends on the quality of selected demonstrations. The challenge lies in effectively using the vast amounts of unlabeled multi-modal data available. This research proposes MAG (MAnifold-Guided semi-supervised in-context demonstration selection), a novel framework designed to exploit this unlabeled data. MAG formulates demonstration selection as a semi-supervised propagation problem on a multi-modal graph, employing a two-stage strategy. First, it uses relevance score propagation to pinpoint a compact set of high-impact unlabeled samples for efficient pseudo-labeling, thereby reducing MLLM inference costs. Second, it utilizes multi-modal relevance to make the final demonstration selections. The study highlights that textual representations are more effective for the initial relevance propagation, while both visual and textual modalities are crucial for selecting high-quality demonstrations. Experiments across eight multi-modal benchmarks demonstrate that MAG consistently outperforms strong baselines, especially in scenarios where labeled data is scarce, achieving significant performance gains with limited pseudo-labeling budgets.

Why it matters

AI developers and researchers working with MLLMs can use MAG to significantly improve the efficiency and performance of few-shot learning, especially in domains where acquiring large amounts of labeled multi-modal data is costly or impractical.

How to implement this in your domain

  1. 1Integrate MAG's two-stage demonstration selection process into multi-modal in-context learning pipelines.
  2. 2Leverage unlabeled multi-modal datasets to enhance the performance of few-shot MLLM tasks.
  3. 3Prioritize textual representations for initial relevance propagation in semi-supervised learning.
  4. 4Combine both visual and textual modalities for final, high-quality demonstration selection.
  5. 5Benchmark MAG against existing ICL demonstration selection methods in label-scarce environments.

Original post by Zirui Cheng, Xun Xu, Tiankai Chen, Fady Rezk, Bowen Zheng, Xiaodong Shi, Shijie Li, Kangkang Lu, Bharadwaj Veeravalli, Nancy F. Chen

"arXiv:2608.12724v1 Announce Type: new Abstract: Few-shot in-context learning (ICL) with multi-modal large language models (MLLMs) enables task adaptation without parameter updates, but its performance is highly sensitive to the quality and coverage of the selected demonstrations.…"

View on X

Originally posted by Zirui Cheng, Xun Xu, Tiankai Chen, Fady Rezk, Bowen Zheng, Xiaodong Shi, Shijie Li, Kangkang Lu, Bharadwaj Veeravalli, Nancy F. Chen on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Research

AI Engineering & DevToolsAI ResearchAI Investing

FlowLOB Generates Realistic, Controllable Limit Order Books Efficiently

This paper introduces FlowLOB, a conditional flow-matching generator for Limit Order Book (LOB) trajectories that offers realistic market dynamics, efficient sampling, and controllable scenario generation, outperforming existing agent-based and deep generative simulators. FlowLOB achieves high fidelity with significantly fewer computational steps than diffusion models and transfers effectively to unseen instruments.

Zhuohan Wang, Andreea Bacalum, Ollie Olby, Carmine Ventre, Namid StillmanAug 14, 2026
AI Engineering & DevToolsAI Research

Auditing Reveals Bias in Neural Combinatorial Optimization Benchmarks

This paper audits test-time budget allocation in Neural Combinatorial Optimization (NCO) solvers, revealing that reported gains from non-uniform sampling often stem from "sampling luck" rather than true allocation benefits on in-distribution data. It proposes a correction procedure and demonstrates real gains under distribution shift, emphasizing the need for rigorous evaluation.

Jinhyung BaeAug 14, 2026
AI Engineering & DevToolsAI Research

Diffusion Models Solve Mixed-Integer Optimization Problems Faster

This paper introduces Constrained Graph Diffusion (CGD), a novel learning-based approach that uses a graph-based generative diffusion model to approximately solve mixed-integer optimization problems (MIPs). CGD integrates a training-free feasibility projection operator into the diffusion process, significantly improving solution quality and feasibility while achieving substantial speedups over traditional numerical solvers.

Vincenzo Di Vito, Mehdi Taghizadeh, Deepjyoti Deka, Kaarthik Sundar, Ferdinando FiorettoAug 14, 2026