TLDR: Token-Level Detective Reward Model for Large Vision Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Fu, Deqing, Xiao, Tong, Wang, Rui, Zhu, Wang, Zhang, Pengchuan, Pang, Guan, Jia, Robin, Chen, Lawrence
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909509040996352
author Fu, Deqing
Xiao, Tong
Wang, Rui
Zhu, Wang
Zhang, Pengchuan
Pang, Guan
Jia, Robin
Chen, Lawrence
author_facet Fu, Deqing
Xiao, Tong
Wang, Rui
Zhu, Wang
Zhang, Pengchuan
Pang, Guan
Jia, Robin
Chen, Lawrence
contents Although reward models have been successful in improving multimodal large language models, the reward models themselves remain brutal and contain minimal information. Notably, existing reward models only mimic human annotations by assigning only one binary feedback to any text, no matter how long the text is. In the realm of multimodal language models, where models are required to process both images and texts, a naive reward model may learn implicit biases toward texts and become less grounded in images. In this paper, we propose a $\textbf{T}$oken-$\textbf{L}$evel $\textbf{D}$etective $\textbf{R}$eward Model ($\textbf{TLDR}$) to provide fine-grained annotations to each text token. We first introduce a perturbation-based method to generate synthetic hard negatives and their token-level labels to train TLDR models. Then we show the rich usefulness of TLDR models both in assisting off-the-shelf models to self-correct their generations, and in serving as a hallucination evaluation tool. We show that TLDR automatically trains a token-level likelihood optimization, and can improve the base model's performance significantly. Finally, we show that TLDR models can significantly speed up human annotation by 3 times to acquire a broader range of high-quality vision language data.
format Preprint
id arxiv_https___arxiv_org_abs_2410_04734
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TLDR: Token-Level Detective Reward Model for Large Vision Language Models
Fu, Deqing
Xiao, Tong
Wang, Rui
Zhu, Wang
Zhang, Pengchuan
Pang, Guan
Jia, Robin
Chen, Lawrence
Machine Learning
Computation and Language
Computer Vision and Pattern Recognition
Although reward models have been successful in improving multimodal large language models, the reward models themselves remain brutal and contain minimal information. Notably, existing reward models only mimic human annotations by assigning only one binary feedback to any text, no matter how long the text is. In the realm of multimodal language models, where models are required to process both images and texts, a naive reward model may learn implicit biases toward texts and become less grounded in images. In this paper, we propose a $\textbf{T}$oken-$\textbf{L}$evel $\textbf{D}$etective $\textbf{R}$eward Model ($\textbf{TLDR}$) to provide fine-grained annotations to each text token. We first introduce a perturbation-based method to generate synthetic hard negatives and their token-level labels to train TLDR models. Then we show the rich usefulness of TLDR models both in assisting off-the-shelf models to self-correct their generations, and in serving as a hallucination evaluation tool. We show that TLDR automatically trains a token-level likelihood optimization, and can improve the base model's performance significantly. Finally, we show that TLDR models can significantly speed up human annotation by 3 times to acquire a broader range of high-quality vision language data.
title TLDR: Token-Level Detective Reward Model for Large Vision Language Models
topic Machine Learning
Computation and Language
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.04734