← Search

Rong Bao

9 accepted papers

2025

Fixing Distribution Shifts of LLM Self-Critique via On-Policy Self-Play Training

ACL 2025long

Self-critique mechanisms significantly improve the performance of language models in complex reasoning tasks by giving them the ability to correct errors, conduct induction and deduction, and switch thinking insights. However, synthetic data methods often require human-introduced errors or sampling…

2025

Implicit Reward as the Bridge: A Unified View of SFT and DPO Connections

NeurIPS 2025poster

Post-training processes are essential phases in grounding pre-trained language models to real-world tasks, with learning from demonstrations or preference signals playing a crucial role in this adaptation. We present a unified theoretical framework bridging Supervised Fine-Tuning (SFT) and preferenc…

Cited by 0SourceScholar
2025

RMB: Comprehensively benchmarking reward models in LLM alignment

ICLR 2025poster

Reward models (RMs) guide the alignment of large language models (LLMs), steering them toward behaviors preferred by humans. Evaluating RMs is the key to better aligning LLMs. However, the current evaluation of RMs may not directly correspond to their alignment performance due to the limited distrib…

2024

InfoRM: Mitigating Reward Hacking in RLHF via Information-Theoretic Reward Modeling

NeurIPS 2024poster

Despite the success of reinforcement learning from human feedback (RLHF) in aligning language models with human values, reward hacking, also termed reward overoptimization, remains a critical challenge. This issue primarily arises from reward misgeneralization, where reward models (RMs) compute rew…

2023

CASN:Class-Aware Score Network for Textual Adversarial Detection

ACL 2023long

Adversarial detection aims to detect adversarial samples that threaten the security of deep neural networks, which is an essential step toward building robust AI systems. Density-based estimation is widely considered as an effective technique by explicitly modeling the distribution of normal data an…

Cited by 4SourcePDFScholar
2023

Orthogonal Subspace Learning for Language Model Continual Learning

EMNLP 2023long findings

Benefiting from massive corpora and advanced hardware, large language models (LLMs) exhibit remarkable capabilities in language understanding and generation. However, their performance degrades in scenarios where multiple tasks are encountered sequentially, also known as catastrophic forgetting. In…

Cited by 0SourcecodeScholar
2022

PlugAT: A Plug and Play Module to Defend against Textual Adversarial Attack

COLING 2022main

Adversarial training, which minimizes the loss of adversarially perturbed examples, has received considerable attention. However, these methods require modifying all model parameters and optimizing the model from scratch, which is parameter inefficient and unfriendly to the already deployed models.…