English

Advancing Autonomous VLM Agents via Variational Subgoal-Conditioned Reinforcement Learning

Machine Learning 2025-05-22 v2 Artificial Intelligence

Abstract

State-of-the-art (SOTA) reinforcement learning (RL) methods have enabled vision-language model (VLM) agents to learn from interaction with online environments without human supervision. However, these methods often struggle with learning inefficiencies when applied to complex, real-world decision-making tasks with sparse rewards and long-horizon dependencies. We propose a novel framework, Variational Subgoal-Conditioned Reinforcement Learning (VSC-RL), advancing the VLM agents in resolving challenging decision-making tasks. Fundamentally distinct from existing methods, VSC-RL reformulates the decision-making problem as a variational subgoal-conditioned RL problem with the newly derived optimization objective, Subgoal Evidence Lower BOund (SGC-ELBO), which comprises two key components: (a) maximizing the subgoal-conditioned return, and (b) minimizing the divergence from a reference goal-conditioned policy. We theoretically and empirically demonstrate that the VSC-RL can efficiently improve the learning efficiency without compromising performance guarantees. Across a diverse set of challenging benchmarks, including mobile device and web control tasks, VSC-RL consistently outperforms existing SOTA methods, achieving superior learning efficiency and performance.

Keywords

Cite

@article{arxiv.2502.07949,
  title  = {Advancing Autonomous VLM Agents via Variational Subgoal-Conditioned Reinforcement Learning},
  author = {Qingyuan Wu and Jianheng Liu and Jianye Hao and Jun Wang and Kun Shao},
  journal= {arXiv preprint arXiv:2502.07949},
  year   = {2025}
}
R2 v1 2026-06-28T21:40:52.645Z