English

Self Iterative Label Refinement via Robust Unlabeled Learning

Computation and Language 2025-12-01 v2

Abstract

Recent advances in large language models (LLMs) have yielded impressive performance on various tasks, yet they often depend on high-quality feedback that can be costly. Self-refinement methods attempt to leverage LLMs' internal evaluation mechanisms with minimal human supervision; however, these approaches frequently suffer from inherent biases and overconfidence, especially in domains where the models lack sufficient internal knowledge, resulting in performance degradation. As an initial step toward enhancing self-refinement for broader applications, we introduce an iterative refinement pipeline that employs the Unlabeled-Unlabeled learning framework to improve LLM-generated pseudo-labels for classification tasks. By exploiting two unlabeled datasets with differing positive class ratios, our approach iteratively denoises and refines the initial pseudo-labels, thereby mitigating the adverse effects of internal biases with minimal human supervision. Evaluations on diverse datasets, including low-resource language corpora, patent classifications, and protein structure categorizations, demonstrate that our method consistently outperforms both initial LLM's classification performance and the self-refinement approaches by cutting-edge models (e.g., GPT-4o and DeepSeek-R1). Moreover, we experimentally confirm that our refined classifier facilitates effective post-training alignment for safety in LLMs and demonstrate successful self-refinement in generative tasks as well.\footnote{Our code is available at https://github.com/HikaruAsano/self-iterative-label-refinement.}

Keywords

Cite

@article{arxiv.2502.12565,
  title  = {Self Iterative Label Refinement via Robust Unlabeled Learning},
  author = {Hikaru Asano and Tadashi Kozuno and Yukino Baba},
  journal= {arXiv preprint arXiv:2502.12565},
  year   = {2025}
}

Comments

To appear in the Thirty-Ninth Annual Conference on Neural Information Processing Systems (NeurIPS 2025)

R2 v1 2026-06-28T21:48:17.573Z