English

Can Active Label Correction Improve LLM-based Modular AI Systems?

Machine Learning 2024-10-04 v3 Artificial Intelligence

Abstract

Modular AI systems can be developed using LLM-prompts-based modules to minimize deployment time even for complex tasks. However, these systems do not always perform well and improving them using the data traces collected from a deployment remains an open challenge. The data traces contain LLM inputs and outputs, but the annotations from LLMs are noisy. We hypothesize that Active Label Correction (ALC) can be use on the collected data to train smaller task-specific improved models that can replace LLM-based modules. In this paper, we study the noise in three GPT-3.5-annotated datasets and their denoising with human feedback. We also propose a novel method ALC3 that iteratively applies three updates to the training dataset: auto-correction, correction using human feedback and filtering. Our results show that ALC3 can lead to oracle performance with feedback on 17-24% fewer examples than the number of noisy examples in the dataset across three different NLP tasks.

Keywords

Cite

@article{arxiv.2401.05467,
  title  = {Can Active Label Correction Improve LLM-based Modular AI Systems?},
  author = {Karan Taneja and Ashok Goel},
  journal= {arXiv preprint arXiv:2401.05467},
  year   = {2024}
}

Comments

EMNLP (Main) 2024, 13 pages, 6 figures

R2 v1 2026-06-28T14:13:38.990Z