← Search

Jonas Mueller

17 accepted papers

2025

Stratify or Die: Rethinking Data Splits in Image Segmentation

NeurIPS 2025poster

Random splitting of datasets in image segmentation often leads to unrepresentative test sets, resulting in biased evaluations and poor model generalization. While stratified sampling has proven effective for addressing label distribution imbalance in classification tasks, extending these ideas to se…

Cited by 0SourceScholar
2024

Quantifying Uncertainty in Answers from any Language Model and Enhancing their Trustworthiness

ACL 2024long

We introduce BSDetector, a method for detecting bad and speculative answers from a pretrained Large Language Model by estimating a numeric confidence score for any output it generated. Our uncertainty quantification technique works for any LLM accessible only via a black-box API, whose training data…

2024

SAND: Smooth imputation of sparse and noisy functional data with Transformer networks

NeurIPS 2024poster

Although the transformer architecture has come to dominate other models for text and image data, its application to irregularly-spaced longitudinal data has been limited. We introduce a variant of the transformer that enables it to more smoothly impute such functional data. We augment the vanilla tr…

Cited by 0SourcePDFScholar
2024

Time-Varying Propensity Score to Bridge the Gap between the Past and Present

ICLR 2024poster

Real-world deployment of machine learning models is challenging because data evolves over time. While no model can work when data evolves in an arbitrary fashion, if there is some pattern to these changes, we might be able to design methods to address it. This paper addresses situations when data ev…

Cited by 2SourcePDFScholar
2023

DataPerf: Benchmarks for Data-Centric AI Development

NeurIPS 2023poster

Machine learning research has long focused on models rather than datasets, and prominent datasets are used for common ML tasks without regard to the breadth, difficulty, and faithfulness of the underlying problems. Neglecting the fundamental importance of data has given rise to inaccuracy, bias, and…

2022

Adaptive Interest for Emphatic Reinforcement Learning

NeurIPS 2022accept

Emphatic algorithms have shown great promise in stabilizing and improving reinforcement learning by selectively emphasizing the update rule. Although the emphasis fundamentally depends on an interest function which defines the intrinsic importance of each state, most approaches simply adopt a unifor…

Cited by 3SourcePDFScholar
2022

Does your graph need a confidence boost? Convergent boosted smoothing on graphs with tabular node features

ICLR 2022spotlight

Many practical modeling tasks require making predictions using tabular data composed of heterogeneous feature types (e.g., text-based, categorical, continuous, etc.). In this setting boosted decision trees and related ensembling techniques generally dominate real-world applications involving iid tr…

2021

Benchmarking Multimodal AutoML for Tabular Data with Text Fields

NeurIPS 2021poster

We consider the use of automated supervised learning systems for data tables that not only contain numeric/categorical columns, but one or more text fields as well. Here we assemble 18 multimodal data tables that each contain some text fields and stem from a real business application. Our publicly-a…

Cited by 45SourceScholar
2021

Continuous Doubly Constrained Batch Reinforcement Learning

NeurIPS 2021poster

Reliant on too many experiments to learn good actions, current Reinforcement Learning (RL) algorithms have limited applicability in real-world settings, which can be too expensive to allow exploration. We propose an algorithm for batch RL, where effective policies are learned using only a fixed offl…

2021

Deep Learning for Functional Data Analysis with Adaptive Basis Layers

ICML 2021spotlight

Despite their widespread success, the application of deep neural networks to functional data remains scarce today. The infinite dimensionality of functional data means standard learning algorithms can be applied only after appropriate dimension reduction, typically achieved via basis expansions. Cur…

2021

Overinterpretation reveals image classification model pathologies

NeurIPS 2021poster

Image classifiers are typically scored on their test set accuracy, but high accuracy can mask a subtle type of model failure. We find that high scoring convolutional neural networks (CNNs) on popular benchmarks exhibit troubling pathologies that allow them to display high accuracy even in the absenc…

2021

Pervasive Label Errors in Test Sets Destabilize Machine Learning Benchmarks

NeurIPS 2021poster

We identify label errors in the test sets of 10 of the most commonly-used computer vision, natural language, and audio datasets, and subsequently study the potential for these label errors to affect benchmark results. Errors in test sets are numerous and widespread: we estimate an average of at lea…

Cited by 673SourcecodeScholar
2020

Educating Text Autoencoders: Latent Representation Guidance via Denoising

ICML 2020poster

Generative autoencoders offer a promising approach for controllable text generation by leveraging their learned sentence representations. However, current models struggle to maintain coherent latent spaces required to perform meaningful text manipulations via latent vector operations. Specifically,…

2019

What made you do this? Understanding black-box decisions with sufficient input subsets

AISTATS 2019poster

Local explanation frameworks aim to rationalize particular decisions made by a black-box prediction model. Existing techniques are often restricted to a specific type of predictor or based on input saliency, which may be undesirably sensitive to factors unrelated to the model’s decision making proc…

2017

Sequence to Better Sequence: Continuous Revision of Combinatorial Structures

ICML 2017poster

We present a model that, after learning on observations of (sequence, outcome) pairs, can be efficiently used to revise a new sequence in order to improve its associated outcome. Our framework requires neither example improvements, nor additional evaluation of outcomes for proposed revisions. To avo…

Cited by 124SourcePDFScholar