← Search

Mikhail Khodak

23 accepted papers

2025

Specialized Foundation Models Struggle to Beat Supervised Baselines

ICLR 2025poster

Following its success for vision and text, the "foundation model" (FM) paradigm—pretraining large models on massive data, then fine-tuning on target tasks—has rapidly expanded to domains in the sciences, engineering, healthcare, and beyond. Has this achieved what the original FMs accomplis…

Cited by 6SourcePDFScholar
2024

Learning to Relax: Setting Solver Parameters Across a Sequence of Linear System Instances

ICLR 2024spotlight

Solving a linear system ${\bf Ax}={\bf b}$ is a fundamental scientific computing primitive for which numerous solvers and preconditioners have been developed. These come with parameters whose optimal values depend on the system being solved and are often impossible or too expensive to identify; t…

Cited by 6SourcePDFScholar
2024

SureMap: Simultaneous mean estimation for single-task and multi-task disaggregated evaluation

NeurIPS 2024poster

Disaggregated evaluation—estimation of performance of a machine learning model on different subpopulations—is a core task when assessing performance and group-fairness of AI systems. A key challenge is that evaluation data is scarce, and subpopulations arising from intersections of attri…

2023

AANG : Automating Auxiliary Learning

ICLR 2023top-25%

Auxiliary objectives, supplementary learning signals that are introduced to help aid learning on data-starved or highly complex end-tasks, are commonplace in machine learning. Whilst much work has been done to formulate useful auxiliary objectives, their construction is still an art which proceeds b…

2023

Cross-Modal Fine-Tuning: Align then Refine

ICML 2023oral

Fine-tuning large-scale pretrained models has led to tremendous progress in well-studied modalities such as vision and NLP. However, similar gains have not been observed in many other modalities due to a lack of relevant pretrained models. In this work, we propose ORCA, a general cross-modal fine-tu…

2023

Learning-augmented private algorithms for multiple quantile release

ICML 2023poster

When applying differential privacy to sensitive data, we can often improve performance using external information such as other sensitive data, public data, or human priors. We propose to use the learning-augmented algorithms (or algorithms with predictions) framework---previously applied largely to…

2023

Meta-Learning Adversarial Bandit Algorithms

NeurIPS 2023poster

We study online meta-learning with bandit feedback, with the goal of improving performance across multiple tasks if they are similar according to some natural similarity measure. As the first to target the adversarial online-within-online partial-information setting, we design meta-algorithms that…

Cited by 4SourcePDFScholar
2023

Meta-Learning in Games

ICLR 2023poster

In the literature on game-theoretic equilibrium finding, focus has mainly been on solving a single game in isolation. In practice, however, strategic interactions—ranging from routing problems to online advertising auctions—evolve dynamically, thereby leading to many similar games to be solved. To a…

Cited by 22SourcePDFScholar
2022

Learning Predictions for Algorithms with Predictions

NeurIPS 2022accept

A burgeoning paradigm in algorithm design is the field of algorithms with predictions, in which algorithms can take advantage of a possibly-imperfect prediction of some aspect of the problem. While much work has focused on using predictions to improve competitive ratios, running times, or other perf…

Cited by 32SourcePDFScholar
2022

NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Tasks

NeurIPS 2022accept

Most existing neural architecture search (NAS) benchmarks and algorithms prioritize well-studied tasks, e.g. image classification on CIFAR or ImageNet. This makes the performance of NAS approaches in more diverse areas poorly understood. In this paper, we present NAS-Bench-360, a benchmark suite to…

2022

Provably tuning the ElasticNet across instances

NeurIPS 2022accept

An important unresolved challenge in the theory of regularization is to set the regularization coefficients of popular techniques like the ElasticNet with general provable guarantees. We consider the problem of tuning the regularization parameters of Ridge regression, LASSO, and the ElasticNet acros…

Cited by 19SourcePDFScholar
2021

Federated Hyperparameter Tuning: Challenges, Baselines, and Connections to Weight-Sharing

NeurIPS 2021poster

Tuning hyperparameters is a crucial but arduous part of the machine learning pipeline. Hyperparameter optimization is even more challenging in federated learning, where models are learned over a distributed network of heterogeneous devices; here, the need to keep data on device and perform local tra…

Cited by 98SourcePDFScholar
2021

Geometry-Aware Gradient Algorithms for Neural Architecture Search

ICLR 2021spotlight

Recent state-of-the-art methods for neural architecture search (NAS) exploit gradient-based optimization by relaxing the problem into continuous optimization over architectures and shared-weights, a noisy process that remains poorly understood. We argue for the study of single-level empirical risk m…

2021

Initialization and Regularization of Factorized Neural Layers

ICLR 2021poster

Factorized layers—operations parameterized by products of two or more matrices—occur in a variety of deep learning contexts, including compressed model training, certain types of knowledge distillation, and multi-head self-attention architectures. We study how to initialize and regularize deep nets…

2021

Learning-to-learn non-convex piecewise-Lipschitz functions

NeurIPS 2021poster

We analyze the meta-learning of the initialization and step-size of learning algorithms for piecewise-Lipschitz functions, a non-convex setting with applications to both machine learning and algorithms. Starting from recent regret bounds for the exponential forecaster on losses with dispersed discon…

Cited by 19SourcePDFScholar
2021

Rethinking Neural Operations for Diverse Tasks

NeurIPS 2021poster

An important goal of AutoML is to automate-away the design of neural networks on new tasks in under-explored domains. Motivated by this goal, we study the problem of enabling users to discover the right neural operations given data from their specific domain. We introduce a search space of operation…

Cited by 32SourcePDFScholar
2020

A Sample Complexity Separation between Non-Convex and Convex Meta-Learning

ICML 2020poster

One popular trend in meta-learning is to learn from many training tasks a common initialization that a gradient-based method can use to solve a new task with few samples. The theory of meta-learning is still in its early stages, with several recent learning-theoretic analyses of methods such as Rept…

Cited by 24SourcePDFScholar
2019

A Theoretical Analysis of Contrastive Unsupervised Representation Learning

ICML 2019oral

Recent empirical works have successfully used unlabeled data to learn feature representations that are broadly useful in downstream classification tasks. Several of these methods are reminiscent of the well-known word2vec embedding algorithm: leveraging availability of pairs of semantically “similar…

Cited by 933SourcePDFScholar
2019

Adaptive Gradient-Based Meta-Learning Methods

NeurIPS 2019poster

We build a theoretical framework for designing and understanding practical meta-learning methods that integrates sophisticated formalizations of task-similarity with the extensive literature on online convex optimization and sequential prediction algorithms. Our approach enables the task-similarity…

2019

Provable Guarantees for Gradient-Based Meta-Learning

ICML 2019oral

We study the problem of meta-learning through the lens of online convex optimization, developing a meta-algorithm bridging the gap between popular gradient-based meta-learning and classical regularization-based multi-task transfer methods. Our method is the first to simultaneously satisfy good sampl…

2018

A Compressed Sensing View of Unsupervised Text Embeddings, Bag-of-n-Grams, and LSTMs

ICLR 2018poster

Low-dimensional vector embeddings, computed using LSTMs or simpler techniques, are a popular approach for capturing the “meaning” of text and a form of unsupervised learning useful for downstream tasks. However, their power is not theoretically understood. The current paper derives formal understand…

Cited by 53SourcePDFScholar