Logistic Regression makes small LLMs strong and explainable "tens-of-shot" classifiers
Not on the map. Why?
Abstract (arXiv)
For simple classification tasks, we show that users can benefit from the advantages of using small, local, generative language models instead of large commercial models without a trade-off in performance or introducing extra labelling costs. These advantages, including those around privacy, availability, cost, and explainability, are important both in commercial applications and in the broader democratisation of AI. Through experiments on 17 sentence classification tasks (2-4 classes), we show that penalised logistic regression on the embeddings from a small LLM equals (and usually betters) the performance of a large LLM in the "tens-of-shot" regime. This requires no more labelled instances than are needed to validate the performance of the large LLM. Finally, we extract stable and sensible explanations for classification decisions.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 34 of its 88 references that are in the map. First posted on arXiv 6 Aug 2024.
Builds on
34 of 88 references.
References · 34
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 814
- 2023FrugalGPT: How to Use Large Language Models While Reducing Cost and Improving Performancecited by 791
- 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 740
- 2020Stanza: A Python Natural Language Processing Toolkit for Many Human Languagescited by 719
- 2022MTEB: Massive Text Embedding Benchmarkcited by 623
- 2020TweetEval: Unified Benchmark and Comparative Evaluation for Tweet Classificationcited by 414
22 more
- 2023SpQR: A Sparse-Quantized Representation for Near-Lossless LLM Weight Compressioncited by 359
- 2013Good Debt or Bad Debt: Detecting Semantic Orientations in Economic Textscited by 310
- 2023Language Models Represent Space and Timecited by 306
- 2024The Era of 1-bit LLMs: All Large Language Models are in 1.58 Bitscited by 289
- 2023Sentiment Analysis in the Era of Large Language Models: A Reality Checkcited by 262
- 2019SemEval-2019 Task 6: Identifying and Categorizing Offensive Language in Social Media (OffensEval)cited by 206
- 2023Fly-Swat or Cannon? Cost-Effective Language Model Choice via Meta-Modelingcited by 94
- 2022Shortcut Learning of Large Language Models in Natural Language Understandingcited by 83
- 2023Scaling Sentence Embeddings with Large Language Modelscited by 83
- 2020Impact of News on the Commodity Market: Dataset and Resultscited by 80
- 2023Trillion Dollar Words: A New Financial Dataset, Task & Market Analysiscited by 73
- 2016Stop Clickbait: Detecting and Preventing Clickbaits in Online News Mediacited by 39
- 2024Simple Techniques for Enhancing Sentence Embeddings in Generative Language Modelscited by 26
- 2023Localizing Lying in Llama: Understanding Instructed Dishonesty on True-False Questions Through Prompting, Probing, and Patchingcited by 24
- 2022Shoring Up the Foundations: Fusing Model Embeddings and Weak Supervisioncited by 18
- 2022Prompt-Augmented Linear Probing: Scaling beyond the Limit of Few-shot In-Context Learnerscited by 18
- 2017Extended Comparisons of Best Subset Selection, Forward Stepwise Selection, and the Lassocited by 16
- 2024Enhancing In-context Learning via Linear Probe Calibrationcited by 16
- 2021FedNLP: An interpretable NLP System to Decode Federal Reserve Communicationscited by 4
- 2024AI on AI: Exploring the Utility of GPT as an Expert Annotator of AI Publicationscited by 4
- 2023Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classificationcited by 3
- 2023CELDA: Leveraging Black-box Language Model as Enhanced Classifier without Labelscited by 2
Built on by
Citing papers per year: 2024: 1; 2025: 2.
3 papers.