← Search

Neel Guha

10 accepted papers

2026

Cartridges: Lightweight and general-purpose long context representations via self-study

ICLR 2026poster

Large language models are often used to answer queries grounded in large text corpora (e.g. codebases, legal documents, or chat histories) by placing the entire corpus in the context window and leveraging in-context learning (ICL). Although current models support contexts of 100K-10M tokens, this se…

Cited by 0SourcecodeScholar
2025

An Architecture Search Framework for Inference-Time Techniques

ICML 2025poster

Inference-time techniques, such as repeated sampling or iterative revisions, are emerging as powerful ways to enhance large-language models (LLMs) at test time. However, best practices for developing systems that combine these techniques remain underdeveloped due to our limited understanding of the…

Cited by 0SourcePDFScholar
2024

Benchmarking and Building Long-Context Retrieval Models with LoCo and M2-BERT

ICML 2024poster

Retrieval pipelines are an integral component of many machine learning systems. However, they perform poorly in domains where documents are long (e.g., 10K tokens or more) and where identifying the relevant document requires synthesizing information across the entire text. Developing long-context re…

Cited by 21SourcePDFScholar
2024

Prospector Heads: Generalized Feature Attribution for Large Models & Data

ICML 2024poster

Feature attribution, the ability to localize regions of the input data that are relevant for classification, is an important capability for ML models in scientific and biomedical domains. Current methods for feature attribution, which rely on "explaining" the predictions of end-to-end classifiers, s…

2024

Smoothie: Label Free Language Model Routing

NeurIPS 2024poster

Large language models (LLMs) are increasingly used in applications where LLM inputs may span many different tasks. Recent work has found that the choice of LLM is consequential, and different LLMs may be good for different input samples. Prior approaches have thus explored how engineers might select…

2024

Stronger Than You Think: Benchmarking Weak Supervision on Realistic Tasks

NeurIPS 2024poster

Weak supervision (WS) is a popular approach for label-efficient learning, leveraging diverse sources of noisy but inexpensive *weak labels* to automatically annotate training data. Despite its wide usage, WS and its practical value are challenging to benchmark due to the many knobs in its setup, inc…

2023

Ask Me Anything: A simple strategy for prompting language models

ICLR 2023top-25%

Large language models (LLMs) transfer well to new tasks out-of-the-box simply given a natural language prompt that demonstrates how to perform the task and no additional training. Prompting is a brittle process wherein small modifications to the prompt can cause large variations in the model predict…

2023

Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification

NeurIPS 2023poster

Recent work has shown that language models' (LMs) prompt-based learning capabilities make them well suited for automating data labeling in domains where manual annotation is expensive. The challenge is that while writing an initial prompt is cheap, improving a prompt is costly---practitioners often…

Cited by 6SourcePDFScholar
2023

LegalBench: A Collaboratively Built Benchmark for Measuring Legal Reasoning in Large Language Models

NeurIPS 2023poster

The advent of large language models (LLMs) and their adoption by the legal community has given rise to the question: what types of legal reasoning can LLMs perform? To enable greater study of this question, we present LegalBench: a collaboratively constructed legal reasoning benchmark consisting of…

2022

Pile of Law: Learning Responsible Data Filtering from the Law and a 256GB Open-Source Legal Dataset

NeurIPS 2022accept

One concern with the rise of large language models lies with their potential for significant harm, particularly from pretraining on biased, obscene, copyrighted, and private information. Emerging ethical approaches have attempted to filter pretraining material, but such approaches have been ad hoc a…