← Search

Akshay Balsubramani

7 accepted papers

2021

WILDS: A Benchmark of in-the-Wild Distribution Shifts

ICML 2021oral

Distribution shifts—where the training distribution differs from the test distribution—can substantially degrade the accuracy of machine learning (ML) systems deployed in the wild. Despite their ubiquity in the real-world deployments, these distribution shifts are under-represented in the datasets w…

2019

An adaptive nearest neighbor rule for classification

NeurIPS 2019spotlight

We introduce a variant of the $k$-nearest neighbor classifier in which $k$ is chosen adaptively for each query, rather than supplied as a parameter. The choice of $k$ depends on properties of each neighborhood, and therefore may significantly vary between different points. (For example, the algorith…

Cited by 38SourcePDFScholar
2018

Semantically Decomposing the Latent Spaces of Generative Adversarial Networks

ICLR 2018poster

We propose a new algorithm for training generative adversarial networks to jointly learn latent codes for both identities (e.g. individual humans) and observations (e.g. specific photographs). In practice, this means that by fixing the identity portion of latent codes, we can generate diverse images…