Lune

ICLR2023Top-tier venue

Learning Uncertainty for Unknown Domains with Zero-Target-Assumption

Yu Yu, Hassan Sajjad, Jia Xu

Abstract

We introduce our Maximum-Entropy Rewarded Reinforcement Learning (MERRL) framework that selects training data for more accurate Natural Language Processing (NLP). Because conventional data selection methods select training samples based on the test domain knowledge and not on real life data, they frequently fail in unknown domains like patent and Twitter. Our approach selects training samples that maximize information uncertainty measured by entropy, including observation entropy like empirical Shannon entropy, Min-entropy, Rényi entropy, and prediction entropy using mutual information, to cover more possible queries that may appear in unknown worlds. Our MERRL using regularized A2C and SAC achieves up to -99.7 perplexity decrease (-43.4% relatively) in language modeling, +25.0 accuracy increase (+40.0% relatively) in sentiment analysis, and +5.0 F1 score increase (+30.8% relatively) in named entity recognition over various domains, demonstrating strong generalization power on unknown test sets.

Ask Lune

Ask anything. The answer shows its sources.

Lune reads the papers around it, then shows you the passages behind every answer.

Questions to start from

Lune answers from

its abstract · 3 top-tier papers that cite it · related work

Get the answer

Free to start. No credit card required.

Cited by top-tier papers3

  1. ProbVLM: Probabilistic Adapter for Frozen Vison-Language ModelsUddeshya Upadhyay, Shyamgopal Karthik, Massimiliano Mancini, Zeynep AkataICCV 2023 · 41 citations
  2. Can Large Language Models Capture Dissenting Human Voices?Noah Lee, Na An, James ThorneEMNLP 2023 · 8 citations
  3. Is Your Model Fairly Certain? Uncertainty-Aware Fairness Evaluation for LLMsYinong Oliver Wang, Nivedha Sivakumar, Falaah Arif Khan, Katherine Metcalf et al.ICML 2025

Related papers

  1. Breaking Consensus Bias: Unsupervised Reinforcement Learning for Machine TranslationShuting Jiang, Ran Song, Siqi Zhang, Yuxin Huang et al.ACL Findings 2026
  2. A Max-Min Entropy Framework for Reinforcement LearningSeungyul Han, Youngchul SungNeurIPS 2021 · 44 citations
  3. APS: Active Pretraining with Successor FeaturesHao Liu, Pieter AbbeelICML 2021 · 147 citations
  4. Maximum Entropy Reinforcement Learning with Diffusion PolicyXiaoyi Dong, Jian Cheng, Xi Sheryl ZhangICML 2025
  5. From Exploration to Exploitation: A Two-Stage Entropy RLVR Approach for Noise-Tolerant MLLM TrainingDonglai Xu, Hongzheng Yang, Yuzhi Zhao, Pingping Zhang et al.CVPR 2026 · 4 citations
Dusk over the sea between two cliffs drawn in fine vertical lines