The Dark Side of Rich Rewards: Understanding and Mitigating Noise in VLM Rewards

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Huang, Sukai, Liu, Shu-Wei, Lipovetzky, Nir, Cohn, Trevor
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911254664183808
author Huang, Sukai
Liu, Shu-Wei
Lipovetzky, Nir
Cohn, Trevor
author_facet Huang, Sukai
Liu, Shu-Wei
Lipovetzky, Nir
Cohn, Trevor
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2409_15922
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle The Dark Side of Rich Rewards: Understanding and Mitigating Noise in VLM Rewards
Huang, Sukai
Liu, Shu-Wei
Lipovetzky, Nir
Cohn, Trevor
Machine Learning
Robotics
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
title The Dark Side of Rich Rewards: Understanding and Mitigating Noise in VLM Rewards
topic Machine Learning
Robotics
url https://arxiv.org/abs/2409.15922