arXiv · 2409.15922
The Dark Side of Rich Rewards: Understanding and Mitigating Noise in VLM Rewards
Abstract
While Vision-Language Models (VLMs) are increasingly used to generate reward signals for training embodied agents to follow instructions, our research reveals that agents guided by VLM rewards often underperform compared to those employing only intrinsic (exploration-driven) rewards, contradicting expectations set by recent work. We hypothesize that false positive rewards -- instances where unintended trajectories are incorrectly rewarded -- are more detrimental than false negatives. Our analysis confirms this hypothesis, revealing that the widely used cosine similarity metric is prone to false positive reward estimates. To address this, we introduce BiMI ({Bi}nary {M}utual {I}nformation), a novel reward function designed to mitigate noise. BiMI significantly enhances learning efficiency across diverse and challenging embodied navigation environments. Our findings offer a nuanced understanding of how different types of reward noise impact agent learning and highlight the importance of addressing multimodal reward signal noise when training embodied agents
Explore related subjects
Keep this discovery
Sukai Huang, Shu-Wei Liu, Nir Lipovetzky, Trevor Cohn. 2024-09-24. The Dark Side of Rich Rewards: Understanding and Mitigating Noise in VLM Rewards. https://arxiv.org/abs/2409.15922
Cite the original work for its findings. Save a collection to share your selection of sources.