Progressive Reasoning with Primitive Correction for Compositional Zero-Shot Learning

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, long

Summary

PRPC, a Progressive Reasoning framework with Primitive Correction, addresses challenges in Compositional Zero-Shot Learning (CZSL) by explicitly modeling bidirectional dependencies between attributes and objects. This framework mitigates error propagation common in prior unidirectional methods. PRPC formulates CZSL as a structured, Q&A-style Chain-of-Thought (CoT) reasoning process, guiding a Multimodal Large Language Model (MLLM) like Qwen-VL through five predefined semantic steps. These steps include initial object and attribute prediction, followed by mutual correction where attributes refine object recognition and vice versa. To enhance reasoning reliability, PRPC incorporates reinforcement learning post-training using a GRPO-based objective, which provides step-level rewards. Extensive experiments on three CZSL benchmarks, including MIT-States and C-GQA, demonstrate that PRPC achieves leading performance, validating its progressive reasoning and bidirectional correction strategy.

Key takeaway

For Machine Learning Engineers developing compositional zero-shot learning systems, consider adopting a progressive, bidirectional reasoning approach. Your models can significantly improve accuracy by implementing mutual correction steps between attribute and object predictions, rather than relying on unidirectional inference. Leverage Chain-of-Thought prompting with MLLMs and integrate reinforcement learning with step-level rewards to refine intermediate reasoning, leading to more robust and consistent compositional understanding.

Key insights

Bidirectional primitive correction via step-wise Chain-of-Thought reasoning enhances compositional zero-shot learning performance.

Principles

Method

PRPC uses a 5-step Q&A-style CoT process with an MLLM (e.g., Qwen-VL) for object/attribute prediction and mutual correction. RL post-training with GRPO loss provides step-level rewards.

In practice

Topics

Best for: Research Scientist, 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 cs.CV updates on arXiv.org.