LLM Chain Ensembles for Scalable and Accurate Data Annotation

David Farr,Nico Manzonelli,Iain Cruickshank,Kate Starbird,Jevin West
2024-10-17
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.
Machine Learning,Social and Information Networks
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to label large - scale data efficiently and accurately in the situation where high - quality labeled data is scarce and costly in rapidly evolving fields. Specifically, the paper proposes a large language model (LLM) chain ensemble method (LLM chain ensemble), aiming to solve this problem in the following ways: 1. **Zero - sample classification ability**: Utilize the zero - sample classification ability of LLM, enabling it to classify on data without prior training, thus providing a feasible data - labeling solution in rapidly changing fields. 2. **Cost - effectiveness**: Traditional data - labeling methods rely on manual labeling, which is both expensive and time - consuming, especially in rapidly evolving fields that require large - scale datasets. The method proposed in the paper achieves significant cost savings by routing data subsets to subsequent models to reduce the use of computational resources. 3. **Performance improvement**: The chain - ensemble method proposed in the paper can not only surpass the best performance of a single LLM, but also improve the overall classification accuracy and robustness by combining the prediction results of multiple LLMs. 4. **Flexibility and adaptability**: This method allows each LLM to handle the data points it is best at and forwards more complex cases to potentially more powerful models, thus showing good adaptability and flexibility on different tasks and datasets. In conclusion, the main goal of the paper is to solve the problem of scarce and costly high - quality labeled data in dynamic fields and improve the efficiency and accuracy of data - labeling by proposing a new LLM chain - ensemble method.