RLLaVA: An RL-central Framework for Language and Vision Assistants

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhao, Lei, Ma, Zihao, Lin, Boyu, Liu, Yuhe, Wu, Wenjun, Huang, Lei
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909975792582656
author Zhao, Lei
Ma, Zihao
Lin, Boyu
Liu, Yuhe
Wu, Wenjun
Huang, Lei
author_facet Zhao, Lei
Ma, Zihao
Lin, Boyu
Liu, Yuhe
Wu, Wenjun
Huang, Lei
contents We present an RL-central framework for Language and Vision Assistants (RLLaVA) with its formulation of Markov decision process (MDP). RLLaVA decouples RL algorithmic logic from model architecture and distributed execution, supporting researchers in implementing new RL algorithms with minimal code, and to plug in a broad family of RL methods and vision-language models (VLMs) while remaining agnostic to specific training and inference engines. RLLaVA makes resource-efficient training of 1B--7B models feasible on common GPUs; notably, 4B-scale models can be trained end-to-end with full-parameter updates on a single 24GB GPU. Experiments on multi-modal and agentic tasks demonstrate that RLLaVA has task extensibility, and the models trained with it consistently improve performance over base models, competitive with other specially engineered RL frameworks. The code is available at https://github.com/TinyLoopX/RLLaVA.
format Preprint
id arxiv_https___arxiv_org_abs_2512_21450
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RLLaVA: An RL-central Framework for Language and Vision Assistants
Zhao, Lei
Ma, Zihao
Lin, Boyu
Liu, Yuhe
Wu, Wenjun
Huang, Lei
Machine Learning
We present an RL-central framework for Language and Vision Assistants (RLLaVA) with its formulation of Markov decision process (MDP). RLLaVA decouples RL algorithmic logic from model architecture and distributed execution, supporting researchers in implementing new RL algorithms with minimal code, and to plug in a broad family of RL methods and vision-language models (VLMs) while remaining agnostic to specific training and inference engines. RLLaVA makes resource-efficient training of 1B--7B models feasible on common GPUs; notably, 4B-scale models can be trained end-to-end with full-parameter updates on a single 24GB GPU. Experiments on multi-modal and agentic tasks demonstrate that RLLaVA has task extensibility, and the models trained with it consistently improve performance over base models, competitive with other specially engineered RL frameworks. The code is available at https://github.com/TinyLoopX/RLLaVA.
title RLLaVA: An RL-central Framework for Language and Vision Assistants
topic Machine Learning
url https://arxiv.org/abs/2512.21450