English
Related papers

Related papers: STAGE: Stable and Generalizable GRPO for Autoregre…

200 papers

Reinforcement learning with verifiable rewards (RLVR) has become a central paradigm for improving reasoning and code generation in large language models, and GRPO-style training is widely adopted for its simplicity and effectiveness.…

Machine Learning · Computer Science 2026-05-07 Zhiyuan Zeng , Jiameng Huang , Zhangyue Yin , Jiashuo Liu , Ziniu Li , Bingrui Li , Yuhao Wu , Yining Zheng , Ge Zhang , Wenhao Huang , Xipeng Qiu

Reinforcement learning (RL) has become a powerful paradigm for optimizing large language models (LLMs) to handle complex reasoning tasks. A core challenge in this process lies in managing policy entropy, which reflects the balance between…

Machine Learning · Computer Science 2026-04-24 Zhenpeng Su , Leiyu Pan , Minxuan Lv , Yuntao Li , Wenping Hu , Fuzheng Zhang , Kun Gai , Guorui Zhou

Autoregressive (AR) visual generation relies on tokenizers to map images to and from discrete sequences. However, tokenizers are trained to reconstruct clean images from ground-truth tokens, while AR generators are optimized only for token…

Computer Vision and Pattern Recognition · Computer Science 2025-12-23 Xinyao Liao , Qiyuan He , Kai Xu , Xiaoye Qu , Yicong Li , Wei Wei , Angela Yao

Reinforcement learning (RL) has emerged as the predominant paradigm for training large language model (LLM)-based AI agents. However, existing backbone RL algorithms lack verified convergence guarantees in agentic scenarios, especially in…

Artificial Intelligence · Computer Science 2026-02-09 Tianyi Hu , Qingxu Fu , Yanxi Chen , Zhaoyang Liu , Bolin Ding

Reinforcement learning (RL) has become a standard approach for post-training large language models and, more recently, for improving image generation models, which uses reward functions to enhance generation quality and human preference…

Computer Vision and Pattern Recognition · Computer Science 2026-01-08 Yunqi Hong , Kuei-Chun Kao , Hengguang Zhou , Cho-Jui Hsieh

Graph generation is a fundamental task with broad applications, such as drug discovery. Recently, discrete flow matching-based graph generation, \aka, graph flow model (GFM), has emerged due to its superior performance and flexible…

Machine Learning · Computer Science 2026-03-12 Baoheng Zhu , Deyu Bo , Delvin Ce Zhang , Xiao Wang

The instability of Generative Adversarial Network (GAN) training has frequently been attributed to gradient descent. Consequently, recent methods have aimed to tailor the models and training procedures to stabilise the discrete updates. In…

Recent advances in text-to-image (T2I) generation via reinforcement learning (RL) have benefited from reward models that assess semantic alignment and visual quality. However, most existing reward models pay limited attention to…

Computer Vision and Pattern Recognition · Computer Science 2026-03-24 Sashuai Zhou , Qiang Zhou , Junpeng Ma , Yue Cao , Ruofan Hu , Ziang Zhang , Xiaoda Yang , Zhibin Wang , Jun Song , Cheng Yu , Bo Zheng , Zhou Zhao

Reinforcement learning (RL) is a key paradigm for post-training large language models (LLMs), but the widely used Group Relative Policy Optimization (GRPO) often suffers from entropy collapse: exploration quickly disappears, policies…

Machine Learning · Computer Science 2026-05-19 Chen Wang , Zhaochun Li , Jionghao Bai , Hexuan Deng , Ge Lan , Yue Wang

Training stability remains a critical bottleneck for Group Relative Policy Optimization (GRPO), often manifesting as a trade-off between reasoning plasticity and general capability retention. We identify a root cause as the geometric…

Machine Learning · Computer Science 2026-02-09 Wenwen Qiang , Ziyin Gu , Jiahuan Zhou , Jie Hu , Jingyao Wang , Changwen Zheng , Hui Xiong

Graph neural networks (GNNs) have shown advantages in graph-based analysis tasks. However, most existing methods have the homogeneity assumption and show poor performance on heterophilic graphs, where the linked nodes have dissimilar…

Machine Learning · Computer Science 2024-04-16 Tianhao Peng , Wenjun Wu , Haitao Yuan , Zhifeng Bao , Zhao Pengrui , Xin Yu , Xuetao Lin , Yu Liang , Yanjun Pu

Recent advances in video world modeling have enabled large-scale generative models to simulate embodied environments with high visual fidelity, providing strong priors for prediction, planning, and control. Yet, despite their realism, these…

Computer Vision and Pattern Recognition · Computer Science 2026-02-10 Haoyang He , Jay Patrikar , Dong-Ki Kim , Max Smith , Daniel McGann , Ali-akbar Agha-mohammadi , Shayegan Omidshafiei , Sebastian Scherer

We focus on the foundational task of Scene Staging: given a reference scene image and a text condition specifying an actor category to be generated in the scene and its spatial relation to the scene, the goal is to synthesize an output…

Computer Vision and Pattern Recognition · Computer Science 2026-05-19 Cong Xie , Che Wang , Yan Zhang , Ruiqi Yu , Han Zou , Zheng Pan , Zhenpeng Zhan

Reinforcement Learning with Verifiable Rewards (RLVR) improves the reasoning ability of Large Language Models (LLMs), but sparse outcome rewards make token-level credit assignment difficult. We study token-level credit as a…

Machine Learning · Computer Science 2026-05-27 Yuhang He , Haodong Wu , Siyi Liu , Hongyu Ge , Hange Zhou , Keyi Wu , Zhuo Zheng , Qihong Lin , Zixin Zhong , Yongqi Zhang

Group Relative Policy Optimization (GRPO) assigns a single scalar advantage to all tokens in a completion. For structured generations with explicit segments and objectives, this couples unrelated reward signals across segments, leading to…

Machine Learning · Computer Science 2026-02-12 Kirill Pavlenko , Alexander Golubev , Simon Karasik , Boris Yangel

Current approaches to pose generation rely heavily on intermediate representations, either through two-stage pipelines with quantization or autoregressive models that accumulate errors during inference. This fundamental limitation leads to…

Computer Vision and Pattern Recognition · Computer Science 2025-07-25 Yayuan Li , Filippos Bellos , Jason Corso

State-of-the-art text-to-image models suffer from a persistent identity crisis when generating scenes with multiple humans: producing duplicate faces, merging identities, and miscounting individuals. We present DisCo (Reinforcement with…

Computer Vision and Pattern Recognition · Computer Science 2026-04-02 Shubhankar Borse , Farzad Farhadzadeh , Munawar Hayat , Fatih Porikli

Recent advances in autoregressive (AR) generative models have produced increasingly powerful systems for media synthesis. Among them, next-scale prediction has emerged as a popular paradigm, where models generate images in a coarse-to-fine…

Computer Vision and Pattern Recognition · Computer Science 2025-12-09 Gengze Zhou , Chongjian Ge , Hao Tan , Feng Liu , Yicong Hong

Reinforcement learning has become a widely used post-training approach for LLM agents, where training commonly relies on outcome-level rewards that provide only coarse supervision. While finer-grained credit assignment is promising for…

Machine Learning · Computer Science 2026-05-15 Sijia Li , Yuchen Huang , Zifan Liu , Yanping Li , Jingjing Fu , Li Zhao , Jiang Bian , Ling Zhang , Jun Zhang , Rui Wang

In this work, we consider one-shot imitation learning for object rearrangement tasks, where an AI agent needs to watch a single expert demonstration and learn to perform the same task in different environments. To achieve a strong…

Machine Learning · Computer Science 2022-11-29 Aviv Netanyahu , Tianmin Shu , Joshua Tenenbaum , Pulkit Agrawal