Automated Multi-level Preference for MLLMs

Mengxi Zhang, Wenhao Wu, Yu Lu, Yuxin Song, Kang Rong +6 more
5/1/2024

No official code found. Best implementation: takomc/amp (22 stars, last commit 2024-09-26).

Abstract

Current multimodal Large Language Models (MLLMs) suffer from ``hallucination'', occasionally generating responses that are not grounded in the input images. To tackle this challenge, one promising path is to utilize reinforcement learning from human feedback (RLHF), which steers MLLMs towards learning superior responses while avoiding inferior ones. We rethink the common practice of using binary preferences (i.e., superior, inferior), and find that adopting multi-level preferences (e.g., superior, medium, inferior) is better for two benefits: 1) It narrows the gap between adjacent levels, thereby encouraging MLLMs to discern subtle differences. 2) It further integrates cross-level comparisons (beyond adjacent-level comparisons), thus providing a broader range of comparisons with hallucination examples. To verify our viewpoint, we present the Automated Multi-level Preference (AMP) framework for MLLMs. To facilitate this framework, we first develop an automated dataset generation pipeline that provides high-quality multi-level preference datasets without any human annotators. Furthermore, we design the Multi-level Direct Preference Optimization (MDPO) algorithm to robustly conduct complex multi-level preference learning. Additionally, we propose a new hallucination benchmark, MRHal-Bench. Extensive experiments across public hallucination and general benchmarks, as well as our MRHal-Bench, demonstrate the effectiveness of our proposed method. Code is available at https://github.com/takomc/amp.

View on arXivView PDF

Code Implementations(1)

takomc/ampCommunity100%

【NeurIPS 2024】The official code of paper "Automated Multi-level Preference for MLLMs"

221Shell, PythonMay 20, 20242 years ago

Similar Papers

Cite this paper

@article{zhang2024automated,
  title  = {Automated Multi-level Preference for MLLMs},
  author = {Mengxi Zhang and Wenhao Wu and Yu Lu and Yuxin Song and Kang Rong and Huanjin Yao and Jianbo Zhao and Fanglong Liu and Yifan Sun and Haocheng Feng and Jingdong Wang},
  year   = {2024},
  eprint = {2405.11165},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2405.11165},
  url    = {https://arxiv.org/abs/2405.11165},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.11165.svg)](https://codeofpaper.com/paper/2405.11165)

Discussion