ActiveLLM: Large Language Model-based Active Learning for Textual Few-Shot Scenarios

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bayer, Markus, Lutz, Justin, Reuter, Christian
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911370318970880
author Bayer, Markus
Lutz, Justin
Reuter, Christian
author_facet Bayer, Markus
Lutz, Justin
Reuter, Christian
contents Active learning is designed to minimize annotation efforts by prioritizing instances that most enhance learning. However, many active learning strategies struggle with a `cold-start' problem, needing substantial initial data to be effective. This limitation reduces their utility in the increasingly relevant few-shot scenarios, where the instance selection has a substantial impact. To address this, we introduce ActiveLLM, a novel active learning approach that leverages Large Language Models such as GPT-4, o1, Llama 3, or Mistral Large for selecting instances. We demonstrate that ActiveLLM significantly enhances the classification performance of BERT classifiers in few-shot scenarios, outperforming traditional active learning methods as well as improving the few-shot learning methods ADAPET, PERFECT, and SetFit. Additionally, ActiveLLM can be extended to non-few-shot scenarios, allowing for iterative selections. In this way, ActiveLLM can even help other active learning strategies to overcome their cold-start problem. Our results suggest that ActiveLLM offers a promising solution for improving model performance across various learning setups.
format Preprint
id arxiv_https___arxiv_org_abs_2405_10808
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ActiveLLM: Large Language Model-based Active Learning for Textual Few-Shot Scenarios
Bayer, Markus
Lutz, Justin
Reuter, Christian
Computation and Language
Active learning is designed to minimize annotation efforts by prioritizing instances that most enhance learning. However, many active learning strategies struggle with a `cold-start' problem, needing substantial initial data to be effective. This limitation reduces their utility in the increasingly relevant few-shot scenarios, where the instance selection has a substantial impact. To address this, we introduce ActiveLLM, a novel active learning approach that leverages Large Language Models such as GPT-4, o1, Llama 3, or Mistral Large for selecting instances. We demonstrate that ActiveLLM significantly enhances the classification performance of BERT classifiers in few-shot scenarios, outperforming traditional active learning methods as well as improving the few-shot learning methods ADAPET, PERFECT, and SetFit. Additionally, ActiveLLM can be extended to non-few-shot scenarios, allowing for iterative selections. In this way, ActiveLLM can even help other active learning strategies to overcome their cold-start problem. Our results suggest that ActiveLLM offers a promising solution for improving model performance across various learning setups.
title ActiveLLM: Large Language Model-based Active Learning for Textual Few-Shot Scenarios
topic Computation and Language
url https://arxiv.org/abs/2405.10808