LLM Chain Ensembles for Scalable and Accurate Data Annotation

David Farr, Nico Manzonelli, Iain Cruickshank, Kate Starbird, Jevin West
10/16/2024
cs.LGcs.SI

No official code found. Best implementation: davidthfarr/ChainEnsembles (3 stars, last commit 2025-02-20).

Abstract

The ability of large language models (LLMs) to perform zero-shot classification makes them viable solutions for data annotation in rapidly evolving domains where quality labeled data is often scarce and costly to obtain. However, the large-scale deployment of LLMs can be prohibitively expensive. This paper introduces an LLM chain ensemble methodology that aligns multiple LLMs in a sequence, routing data subsets to subsequent models based on classification uncertainty. This approach leverages the strengths of individual LLMs within a broader system, allowing each model to handle data points where it exhibits the highest confidence, while forwarding more complex cases to potentially more robust models. Our results show that the chain ensemble method often exceeds the performance of the best individual model in the chain and achieves substantial cost savings, making LLM chain ensembles a practical and efficient solution for large-scale data annotation challenges.

View on arXivView PDF

Code Implementations(1)

Github repo for LLM Chain Ensembles for Scalable and Accurate Data Annotation

31PythonSep 5, 20241 years ago

Similar Papers

Cite this paper

@article{farr2024chain,
  title  = {LLM Chain Ensembles for Scalable and Accurate Data Annotation},
  author = {David Farr and Nico Manzonelli and Iain Cruickshank and Kate Starbird and Jevin West},
  year   = {2024},
  eprint = {2410.13006},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.13006v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.13006.svg)](https://codeofpaper.com/paper/2410.13006)

Discussion