cs.LGOct 15, 2026

Generative Value Classifier

Authors: Alexander Swerdlow, Perry Dong, Chelsea Finn

Organizations: Stanford University

Abstract

Q-learning is premised on learning a function to predict the value of a state-action pair, with the goal of using such a function to guide policy improvement. Most approaches learn this function from scratch using scalar or distributional return targets. This discriminative approach leaves another source of supervision for learning representations unused: the actions themselves. Rather than directly predict the return of a state-action pair, we ask the reverse question: how likely is the action under each possible return? This perspective provides a critic whose generative component is a return-conditioned policy, which can share the actor's architecture. Training this policy supervises the critic with every action rather than with returns alone, and a squared-error loss anchors the critic's Q-value to the bootstrapped target. We evaluate our approach on both offline and online RL tasks and show gains over leading discriminative methods, including distributional RL methods.

Figures & tables

Appendix figures & tables7 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Bringing Value Models Back: Generative Critics for Value Modeling in LLM Reinforcement Learning

    Apr 12, 2026Zikang Shan, Han Zhong, Liwei Wang +1RL for Language ModelsReinforcement Learning

  2. Per-Domain Generalizing Policies: On Learning Efficient and Robust Q-Value Functions (Extended Version with Technical Appendix)

    Mar 18, 2026Nicola J. Müller, Moritz Oster, Isabel Valera +2Generalization in RLQ-Learning

  3. Decoupling Policy Extraction for Offline Reinforcement Learning

    Aug 21, 2026Xuyao Lin, Yixiang Shan, Jinru Duan +7Offline RLPolicy Learning