English
Related papers

Related papers: Stabilizing MoE Reinforcement Learning by Aligning…

200 papers

Recent advancements have shown that the Mixture of Experts (MoE) approach significantly enhances the capacity of large language models (LLMs) and improves performance on downstream tasks. Building on these promising results, multi-modal…

Computation and Language · Computer Science 2025-06-02 Linglin Jing , Yuting Gao , Zhigang Wang , Wang Lan , Yiwen Tang , Wenhai Wang , Kaipeng Zhang , Qingpei Guo

Mixture-of-Experts (MoE) enables efficient scaling of large language models (LLMs) with sparsely activated experts during inference. To effectively deploy large MoE models on memory-constrained devices, many systems introduce *expert…

Machine Learning · Computer Science 2026-03-03 Jingcong Liang , Siyuan Wang , Miren Tian , Yitong Li , Duyu Tang , Zhongyu Wei

Reinforcement learning (RL) has been successfully applied to autoregressive (AR) and diffusion models. However, extending RL to hybrid AR-diffusion frameworks remains challenging due to interleaved inference and noisy log-probability…

Computer Vision and Pattern Recognition · Computer Science 2026-04-09 Xiaoxiao Ma , Jiachen Lei , Tianfei Ren , Jie Huang , Siming Fu , Aiming Hao , Jiahong Wu , Xiangxiang Chu , Feng Zhao

Expert parallelism is vital for effectively training Mixture-of-Experts (MoE) models, enabling different devices to host distinct experts, with each device processing different input data. However, during expert parallel training, dynamic…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-13 Xinyi Liu , Yujie Wang , Fangcheng Fu , Xuefeng Xiao , Huixia Li , Jiashi Li , Bin Cui

Large Language Models (LLMs) increasingly rely on reinforcement learning with verifiable rewards (RLVR) to elicit reliable chain-of-thought reasoning. However, the training process remains bottlenecked by the computationally expensive…

Machine Learning · Computer Science 2026-01-13 Bingshuai Liu , Ante Wang , Zijun Min , Liang Yao , Haibo Zhang , Yang Liu , Xu Han , Peng Li , Anxiang Zeng , Jinsong Su

While Reinforcement Learning for Verifiable Rewards (RLVR) is powerful for training large reasoning models, its training dynamics harbor a critical challenge: RL overfitting, where models gain training rewards but lose generalization. Our…

Artificial Intelligence · Computer Science 2025-11-07 Zeng Zhiyuan , Jiashuo Liu , Zhangyue Yin , Ge Zhang , Wenhao Huang , Xipeng Qiu

Reinforcement Learning (RL) has become essential for eliciting complex reasoning capabilities in Large Language Models (LLMs). However, the substantial memory overhead of storing Key-Value (KV) caches during long-horizon rollouts acts as a…

Machine Learning · Computer Science 2026-03-31 Sijia Luo , Xiaokang Zhang , Yuxuan Hu , Bohan Zhang , Ke Wang , Jinbo Su , Mengshu Sun , Lei Liang , Jing Zhang

Sparse Mixture-of-Experts (MoE) architectures are increasingly popular for frontier large language models (LLM) but they introduce training challenges due to routing complexity. Fully leveraging parameters of an MoE model requires all…

Reinforcement learning (RL) has emerged as a promising strategy for finetuning small language models (SLMs) to solve targeted tasks such as math and coding. However, RL algorithms tend to be resource-intensive, taking a significant amount…

Machine Learning · Computer Science 2025-10-07 Lianghuan Huang , Sagnik Anupam , Insup Lee , Shuo Li , Osbert Bastani

Long chain-of-thought (CoT) significantly enhances the reasoning capabilities of large language models (LLMs). However, extensive reasoning traces lead to inefficiencies and increased time-to-first-token (TTFT). We propose a training…

Computation and Language · Computer Science 2026-01-08 Roy Xie , David Qiu , Deepak Gopinath , Dong Lin , Yanchao Sun , Chong Wang , Saloni Potdar , Bhuwan Dhingra

Mixture-of-Experts (MoE) language models route each token to a small subset of experts, but whether the routes selected by a trained top-$k$ router are good ones is rarely evaluated directly. Holding the model fixed, we compare each…

Machine Learning · Computer Science 2026-05-11 Youngsik Yoon , Siwei Wang , Wei Chen , Jungseul Ok

Deep Research agents tackle knowledge-intensive tasks through multi-round retrieval and decision-oriented generation. While reinforcement learning (RL) has been shown to improve performance in this paradigm, its contributions remain…

Computation and Language · Computer Science 2026-02-24 Yinuo Xu , Shuo Lu , Jianjie Cheng , Meng Wang , Qianlong Xie , Xingxing Wang , Ran He , Jian Liang

Reinforcement learning (RL) for large language models (LLMs) is increasingly bottlenecked by rollout (generation), where long output sequence lengths make attention and KV-cache memory dominate end-to-end step time. FP8 offers an attractive…

Machine Learning · Computer Science 2026-04-13 Zhaopeng Qiu , Shuang Yu , Jingqi Zhang , Shuai Zhang , Xue Huang , Jingyi Yang , Junjie Lai

Sparsely activated Mixture-of-Experts (MoE) models are widely adopted to scale up model capacity without increasing the computation budget. However, vanilla TopK routers are trained in a discontinuous, non-differentiable way, limiting their…

Machine Learning · Computer Science 2025-02-28 Ziteng Wang , Jun Zhu , Jianfei Chen

Reinforcement Learning from Verifiable Rewards (RLVR) improves the reasoning abilities of Large Language Models (LLMs) but it struggles with unstable exploration. We propose FR3E (First Return, Entropy-Eliciting Explore), a structured…

Artificial Intelligence · Computer Science 2025-07-10 Tianyu Zheng , Tianshun Xing , Qingshui Gu , Taoran Liang , Xingwei Qu , Xin Zhou , Yizhi Li , Zhoufutu Wen , Chenghua Lin , Wenhao Huang , Qian Liu , Ge Zhang , Zejun Ma

Sparse Mixture-of-Experts (MoE) have been widely adopted in recent large language models since it can efficiently scale up the model capability without increasing the inference cost. However, evaluations on broad downstream tasks reveal a…

Machine Learning · Computer Science 2025-11-13 Zhongyang Li , Ziyue Li , Tianyi Zhou

Research on applications of reinforcement learning (RL) to large language models has mostly been focused on single-turn problems, such as mathematical reasoning or single-shot code generation. While these problems can be viewed as…

Ride-hailing platforms face the challenge of balancing passenger waiting times with overall system efficiency under highly uncertain supply-demand conditions. Adaptive delayed matching, which controls the holding intervals for batched sets…

Machine Learning · Computer Science 2026-05-05 Yuhan Tang , Kangxin Cui , Jung Ho Park , Yibo Zhao , Xuan Jiang , Haoze He , Jiangbo Yu , Haris Koutsopoulos , Jinhua Zhao

Reinforcement learning has become a central paradigm for improving LLM reasoning. However, existing methods use a single policy to produce both inference responses and training optimization trajectories. The objective conflict between…

Machine Learning · Computer Science 2026-01-26 Jingchu Wang , Bingbing Xu , Yige Yuan , Bin Xie , Xiaoqian Sun , Huawei Shen

The development of state-of-the-art large language models is commonly understood as a two-stage process involving pre-training and post-training. We point out the need for an additional intermediate stage called reinforcement mid-training…

Computation and Language · Computer Science 2025-09-30 Yijun Tian , Shaoyu Chen , Zhichao Xu , Yawei Wang , Jinhe Bi , Peng Han , Wei Wang
‹ Prev 1 3 4 5 6 7 10 Next ›