← Search

Jake Snell

12 accepted papers

2024

A Metalearned Neural Circuit for Nonparametric Bayesian Inference

NeurIPS 2024poster

Most applications of machine learning to classification assume a closed set of balanced classes. This is at odds with the real world, where class occurrence statistics often follow a long-tailed power-law distribution and it is unlikely that all classes are seen in a single sample. Nonparametric Bay…

2024

Implicit Maximum a Posteriori Filtering via Adaptive Optimization

ICLR 2024poster

Bayesian filtering approximates the true underlying behavior of a time-varying system by inverting an explicit generative model to convert noisy measurements into state estimates. This process typically requires matrix storage, inversion, and multiplication or Monte Carlo estimation, none of which a…

2024

Prompt Risk Control: A Rigorous Framework for Responsible Deployment of Large Language Models

ICLR 2024poster

With the explosion of the zero-shot capabilities of (and thus interest in) pre-trained large language models, there has come accompanying interest in how best to prompt a language model to perform a given task. While it may be tempting to choose a prompt based on empirical results on a validation se…

2023

Distribution-Free Statistical Dispersion Control for Societal Applications

NeurIPS 2023spotlight

Explicit finite-sample statistical guarantees on model performance are an important ingredient in responsible machine learning. Previous work has focused mainly on bounding either the expected loss of a predictor or the probability that an individual prediction will incur a loss value in a specifie…

Cited by 5SourcePDFScholar
2023

Im-Promptu: In-Context Composition from Image Prompts

NeurIPS 2023poster

Large language models are few-shot learners that can solve diverse tasks from a handful of demonstrations. This implicit understanding of tasks suggests that the attention mechanisms over word tokens may play a role in analogical reasoning. In this work, we investigate whether analogical reasoning c…

Cited by 3SourcePDFScholar
2023

Quantile Risk Control: A Flexible Framework for Bounding the Probability of High-Loss Predictions

ICLR 2023poster

Rigorous guarantees about the performance of predictive algorithms are necessary in order to ensure their responsible use. Previous work has largely focused on bounding the expected loss of a predictor, but this is not sufficient in many risk-sensitive applications where the distribution of errors i…

2021

Bayesian Few-Shot Classification with One-vs-Each Pólya-Gamma Augmented Gaussian Processes

ICLR 2021poster

Few-shot classification (FSC), the task of adapting a classifier to unseen classes given a small labeled dataset, is an important step on the path toward human-like machine learning. Bayesian methods are well-suited to tackling the fundamental issue of overfitting in the few-shot scenario because th…

2019

Dimensionality Reduction for Representing the Knowledge of Probabilistic Models

ICLR 2019poster

Most deep learning models rely on expressive high-dimensional representations to achieve good performance on tasks such as classification. However, the high dimensionality of these representations makes them difficult to interpret and prone to over-fitting. We propose a simple, intuitive and scalabl…

Cited by 15SourcePDFScholar
2018

Meta-Learning for Semi-Supervised Few-Shot Classification

ICLR 2018poster

In few-shot classification, we are interested in learning algorithms that train a classifier from only a handful of labeled examples. Recent progress in few-shot classification has featured meta-learning, in which a parameterized model for a learning algorithm is defined and trained on episodes repr…