← Search

Nimit S. Sohoni

4 accepted papers

2022

Correct-N-Contrast: a Contrastive Approach for Improving Robustness to Spurious Correlations

ICML 2022oral

Spurious correlations pose a major challenge for robust machine learning. Models trained with empirical risk minimization (ERM) may learn to rely on correlations between class labels and spurious attributes, leading to poor performance on data groups without these correlations. This is challenging t…

2022

Monarch: Expressive Structured Matrices for Efficient and Accurate Training

ICML 2022oral

Large neural networks excel in many domains, but they are expensive to train and fine-tune. A popular approach to reduce their compute or memory requirements is to replace dense weight matrices with structured ones (e.g., sparse, low-rank, Fourier transform). These methods have not seen widespread a…

2021

Low-Shot Validation: Active Importance Sampling for Estimating Classifier Performance on Rare Categories

ICCV 2021poster

For machine learning models trained with limited labeled training data, validation stands to become the main bottleneck to reducing overall annotation costs. We propose a statistical validation algorithm that accurately estimates the F-score of binary classifiers for rare categories, where finding r…

Cited by 9PDFScholar
2021

Mandoline: Model Evaluation under Distribution Shift

ICML 2021spotlight

Machine learning models are often deployed in different settings than they were trained and validated on, posing a challenge to practitioners who wish to predict how well the deployed model will perform on a target distribution. If an unlabeled sample from the target distribution is available, along…