← Search

Vidya Muthukumar

14 accepted papers

2025

Task Shift: From Classification to Regression in Overparameterized Linear Models

AISTATS 2025poster

Modern machine learning methods have recently demonstrated remarkable capability to generalize under task shift, where latent knowledge is transferred to a different, often more difficult, task under a similar data distribution. We investigate this phenomenon in an overparameterized linear regressio…

Cited by 0SourcecodeScholar
2024

Balanced Data, Imbalanced Spectra: Unveiling Class Disparities with Spectral Imbalance

ICML 2024poster

Classification models are expected to perform equally well for different classes, yet in practice, there are often large gaps in their performance. This issue of class bias is widely studied in cases of datasets with sample imbalance, but is relatively overlooked in balanced datasets. In this work,…

Cited by 4SourcePDFScholar
2024

One Shot Inverse Reinforcement Learning for Stochastic Linear Bandits

UAI 2024poster

The paradigm of inverse reinforcement learning (IRL) is used to specify the reward function of an agent purely from its actions and is critical for value alignment and AI safety. While IRL is successful in practice, theoretical guarantees remain nascent. Motivated by the need for IRL in large action…

Cited by 1SourcePDFScholar
2024

Precise asymptotics of reweighted least-squares algorithms for linear diagonal networks

NeurIPS 2024poster

The classical iteratively reweighted least-squares (IRLS) algorithm aims to recover an unknown signal from linear measurements by performing a sequence of weighted least squares problems, where the weights are recursively updated at each step. Varieties of this algorithm have been shown to achieve f…

Cited by 1SourcePDFScholar
2024

The Group Robustness is in the Details: Revisiting Finetuning under Spurious Correlations

NeurIPS 2024poster

Modern machine learning models are prone to over-reliance on spurious correlations, which can often lead to poor performance on minority groups. In this paper, we identify surprising and nuanced behavior of finetuned models on worst-group accuracy via comprehensive experiments on four well-establish…

2023

Faster Margin Maximization Rates for Generic Optimization Methods

NeurIPS 2023spotlight

First-order optimization methods tend to inherently favor certain solutions over others when minimizing a given training objective with multiple local optima. This phenomenon, known as \emph{implicit bias}, plays a critical role in understanding the generalization capabilities of optimization algori…

Cited by 2SourcePDFScholar
2023

Towards Last-layer Retraining for Group Robustness with Fewer Annotations

NeurIPS 2023poster

Empirical risk minimization (ERM) of neural networks is prone to over-reliance on spurious correlations and poor generalization on minority groups. The recent deep feature reweighting (DFR) technique achieves state-of-the-art group robustness via simple last-layer retraining, but it requires held-ou…

2021

Benign Overfitting in Multiclass Classification: All Roads Lead to Interpolation

NeurIPS 2021poster

The growing literature on "benign overfitting" in overparameterized models has been mostly restricted to regression or binary classification settings; however, most success stories of modern machine learning have been recorded in multiclass settings. Motivated by this discrepancy, we study benign ov…

Cited by 64SourcePDFScholar
2021

Online Model Selection for Reinforcement Learning with Function Approximation

AISTATS 2021poster

Deep reinforcement learning has achieved impressive successes yet often requires a very large amount of interaction data. This result is perhaps unsurprising, as using complicated function approximation often requires more data to fit, and early theoretical results on linear Markov decision processe…

Cited by 46SourcePDFScholar
2020

OSOM: A simultaneously optimal algorithm for multi-armed and linear contextual bandits

AISTATS 2020poster

We consider the stochastic linear (multi-armed) contextual bandit problem with the possibility of hidden simple multi-armed bandit structure in which the rewards are independent of the contextual information. Algorithms that are designed solely for one of the regimes are known to be sub-optimal for…

Cited by 47SourcePDFScholar
2019

Best of many worlds: Robust model selection for online supervised learning

AISTATS 2019poster

We introduce algorithms for online, full-information prediction that are computationally efficient and competitive with contextual tree experts of unknown complexity, in both probabilistic and adversarial settings. We incorporate a novel probabilistic framework of structural risk minimization in…

Cited by 13SourcePDFScholar