Progressive Reasoning with Primitive Correction for Compositional Zero-Shot Learning
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
- Contextual dependency is crucial for attribute-object pairs.
- Bidirectional correction reduces error propagation.
- Structured reasoning improves compositional generalization.
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
- Implement 5-step reasoning for attribute-object classification.
- Use RL with step-level rewards for MLLM fine-tuning.
- Generate CoT supervision with advanced LLMs like GPT-4o.
Topics
- Compositional Zero-Shot Learning
- Multimodal Large Language Models
- Chain-of-Thought Reasoning
- Reinforcement Learning
- Attribute-Object Recognition
- Error Propagation Mitigation
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.