SOTAVerified

LlamBERT: Large-scale low-cost data annotation in NLP

2024-03-23Code Available1· sign in to hype

Bálint Csanády, Lajos Muzsai, Péter Vedres, Zoltán Nádasdy, András Lukács

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

Large Language Models (LLMs), such as GPT-4 and Llama 2, show remarkable proficiency in a wide range of natural language processing (NLP) tasks. Despite their effectiveness, the high costs associated with their use pose a challenge. We present LlamBERT, a hybrid approach that leverages LLMs to annotate a small subset of large, unlabeled databases and uses the results for fine-tuning transformer encoders like BERT and RoBERTa. This strategy is evaluated on two diverse datasets: the IMDb review dataset and the UMLS Meta-Thesaurus. Our results indicate that the LlamBERT approach slightly compromises on accuracy while offering much greater cost-effectiveness.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
IMDbRoBERTa-large with LlamBERTAccuracy96.68Unverified
IMDbRoBERTa-largeAccuracy96.54Unverified
IMDbLlama-2-70b-chat (0-shot)Accuracy95.39Unverified

Reproductions