Which Model Shall I Choose? Cost/Quality Trade-offs for Text Classification Tasks
Not on the map. Why?
Abstract (arXiv)
Industry practitioners always face the problem of choosing the appropriate model for deployment under different considerations, such as to maximize a metric that is crucial for production, or to reduce the total cost given financial concerns. In this work, we focus on the text classification task and present a quantitative analysis for this challenge. Using classification accuracy as the main metric, we evaluate the classifiers' performances for a variety of models, including large language models, along with their associated costs, including the annotation cost, training (fine-tuning) cost, and inference cost. We then discuss the model choices for situations like having a large number of samples needed for inference. We hope our work will help people better understand the cost/quality trade-offs for the text classification task.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 12 of its 16 references that are in the map. First posted on arXiv 17 Jan 2023.
Builds on
12 of 16 references.
References · 12
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
- 2014Convolutional Neural Networks for Sentence Classificationcited by 1,124
- 2019Green AIcited by 620
- 2020Deep Learning Based Text Classification: A Comprehensive Reviewcited by 140
- 2020The Cost of Training NLP Models: A Concise Overviewcited by 97
- 2020A Survey on Text Classification: From Shallow to Deep Learningcited by 80
- 2021How to Train BERT with an Academic Budgetcited by 80
- 2021Pre-train or Annotate? Domain Adaptation with a Constrained Budgetcited by 27
Built on by
Citing papers per year: 2023: 1; 2024: 3.
4 papers.
- 2023Fly-Swat or Cannon? Cost-Effective Language Model Choice via Meta-Modelingcited by 94
- 2024OptLLM: Optimal Assignment of Queries to Large Language Modelscited by 14
- 2024JurEE not Judges: safeguarding llm interactions with small, specialised Encoder Ensemblescited by 3
- 2024EventChat: Implementation and user-centric evaluation of a large language model-driven conversational recommender system for exploring leisure events in an SME contextcited by 0
- 2024JurEE not Judges: safeguarding llm interactions with small, specialised Encoder Ensembles11 Oct 2024
- 2024EventChat: Implementation and user-centric evaluation of a large language model-driven conversational recommender system for exploring leisure events in an SME context5 Jul 2024
- 2024OptLLM: Optimal Assignment of Queries to Large Language Models24 May 2024
- 2023Fly-Swat or Cannon? Cost-Effective Language Model Choice via Meta-Modeling11 Aug 2023