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

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Taneja, Karan, Goel, Ashok
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916421052661760
author Taneja, Karan
Goel, Ashok
author_facet Taneja, Karan
Goel, Ashok
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2401_05467
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Can Active Label Correction Improve LLM-based Modular AI Systems?
Taneja, Karan
Goel, Ashok
Machine Learning
Artificial Intelligence
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.
title Can Active Label Correction Improve LLM-based Modular AI Systems?
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2401.05467