about
Logistic Regression makes small LLMs strong "tens-of-shot" classifiers (arxiv.org)
1 point by PaulHoule on Aug 18, 2024 | hide | past | pdf | discuss on HN

In plain words: Instead of prompting a big chatbot with examples, this trains a simple linear classifier on the numbers a small local language model gives each sentence. Across 17 sentence-sorting tasks it matched or usually beat the big model with tens of labels, and explains its decisions.

Abstract · Logistic Regression makes small LLMs strong and explainable "tens-of-shot" classifiers

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.

Marcus Buckmann, Edward Hill
arXiv:2408.03414 · cs.CL, cs.LG, stat.ML · submitted Aug 6, 2024 · updated Oct 4, 2024
abstract · pdf · html · 48 pages, 24 figures

add comment on HN