On the Reliability of Majority-Vote Pseudo-Labels in Test-Time Reinforcement Learning
Haoyu Wang, Ru Wang, Yuanyuan Wang, Jiaxian Guo, Mingming Gong
Abstract
Test-time reinforcement learning (TTRL) adapts pretrained language models to unlabeled target-domain data using self-generated pseudo-labels, whose reliability remains poorly understood. We analyze majority-vote supervision for TTRL with online Direct Preference Optimization (DPO) and decompose its error into population-majority bias and prompt-dependent finite-rollout variation. Because population-majority bias is generally unidentifiable without external supervision, we focus on mitigating the downstream effects of finite-rollout variation. We propose Prompt-Conditional Distributionally Robust DPO (PC-Dr.DPO), which conditions within-prompt robust reweighting on observed vote reliability to make updates from sampling-sensitive prompts more conservative. We further introduce a cache-based mechanism to reduce self-supervised forgetting during online adaptation. Across six mathematical and knowledge-intensive reasoning benchmarks and five model backbones, PC-Dr.DPO matches or outperforms the strongest TTRL baseline in 24 of 30 settings.
Create a lesson
Related papers
CPU-ARC: Agentic Reasoning Caching for on-CPU Reasoning Reuse
Peter Belcak, Greg Heinrich, Saurav Muralidharan et al.
Flow-Guided Target-Space Alignment via Path Consistency
Ruizhi Yuan, Zeqiu Yu, Wei Gao et al.
Ensemble Methodology for Reference-Free Confidence Estimation in LLM-based Code Summarization
Marco Colombo, Balaji A Jayaram, Sowmyakumari Selvaraj et al.
Can Agentic Systems Close the Gap in Radiology with Production AI Tools?
Aubin Charley, Julien Khlaut, Charles Corbière et al.
BitDecoding-FP4: Native FP4 KV Cache Decoding via Algorithm-System Co-Design
Dayou Du, Woon Yee Ng, Guangming Sheng et al.
Generative Latent Priors for Interpreting and Steering Protein Language Models
Daniel Hinjos, Lucía Urcelay