activity
20232026
most citedMitigating the Alignment Tax of RLHF

10 citations · 22 across the 11 of their papers we have counts for

collaborators

11 papers

cs.LG2026

Beyond Uniform Token-Level Trust Region in LLM Reinforcement Learning

Renjie Mao, Xiangxin Zhou, Lvfang Tao +7

Reinforcement learning with verifiable rewards (RLVR) has become standard for improving LLM reasoning. However, existing PPO-style trust-region mechanisms remain position-agnostic…

cs.LG2024★ 1 cited

Entropy-Regularized Process Reward Model

Hanning Zhang, Pengcheng Wang, Shizhe Diao +6

Large language models (LLMs) have shown promise in performing complex multi-step reasoning, yet they continue to struggle with mathematical reasoning, often making systematic error…

cs.LG2024

On the Limited Generalization Capability of the Implicit Reward Model Induced by Direct Preference Optimization

Yong Lin, Skyler Seto, Maartje ter Hoeve +6

Reinforcement Learning from Human Feedback (RLHF) is an effective approach for aligning language models to human preferences. Central to RLHF is learning a reward function for scor…

cs.CL2024

Regularizing Hidden States Enables Learning Generalizable Reward Model for LLMs

Rui Yang, Ruomeng Ding, Yong Lin +2

Reward models trained on human preference data have been proven to effectively align Large Language Models (LLMs) with human intent within the framework of reinforcement learning f…

cs.LG2024

Arithmetic Control of LLMs for Diverse User Preferences: Directional Preference Alignment with Multi-Objective Rewards

Haoxiang Wang, Yong Lin, Wei Xiong +5

Fine-grained control over large language models (LLMs) remains a significant challenge, hindering their adaptability to diverse user needs. While Reinforcement Learning from Human…

cs.CV2024

The Instinctive Bias: Spurious Images lead to Illusion in MLLMs

Tianyang Han, Qing Lian, Rui Pan +5

Large language models (LLMs) have recently experienced remarkable progress, where the advent of multi-modal large language models (MLLMs) has endowed LLMs with visual capabilities,…

cs.CL2023★ 8 cited

R-Tuning: Instructing Large Language Models to Say `I Don't Know'

Hanning Zhang, Shizhe Diao, Yong Lin +6

Large language models (LLMs) have revolutionized numerous domains with their impressive performance but still face their challenges. A predominant issue is the propensity for these…

cs.LG2023★ 1 cited

Continuous Invariance Learning

Yong Lin, Fan Zhou, Lu Tan +8

Invariance learning methods aim to learn invariant features in the hope that they generalize under distributional shifts. Although many tasks are naturally characterized by continu…

stat.ML2023

Optimal Sample Selection Through Uncertainty Estimation and Its Application in Deep Learning

Yong Lin, Chen Liu, Chenlu Ye +3

Modern deep learning heavily relies on large labeled datasets, which often comse with high costs in terms of both manual labeling and computational resources. To mitigate these cha…

cs.LG2023★ 2 cited

Spurious Feature Diversification Improves Out-of-distribution Generalization

Yong Lin, Lu Tan, Yifan Hao +5

Generalization to out-of-distribution (OOD) data is a critical challenge in machine learning. Ensemble-based methods, like weight space ensembles that interpolate model parameters,…

cs.LG2023★ 10 cited

Mitigating the Alignment Tax of RLHF

Yong Lin, Hangyu Lin, Wei Xiong +14

LLMs acquire a wide range of abilities during pre-training, but aligning LLMs under Reinforcement Learning with Human Feedback (RLHF) can lead to forgetting pretrained abilities, w…