activity
20182024
most citedPenalty Method for Inversion-Free Deep Bilevel Optimization

22 citations · 34 across the 12 of their papers we have counts for

collaborators
Showing cs.LGShow all

10 papers · 1 filter

cs.LG2024

DPCore: Dynamic Prompt Coreset for Continual Test-Time Adaptation

Yunbei Zhang, Akshay Mehra, Shuaicheng Niu +1

Continual Test-Time Adaptation (CTTA) seeks to adapt source pre-trained models to continually changing, unseen target domains. While existing CTTA methods assume structured domain…

cs.LG2024★ 1 cited

Test-time Assessment of a Model's Performance on Unseen Domains via Optimal Transport

Akshay Mehra, Yunbei Zhang, Jihun Hamm

Gauging the performance of ML models on data from unseen domains at test-time is essential yet a challenging problem due to the lack of labels in this setting. Moreover, the perfor…

cs.LG2023★ 1 cited

Understanding the Transferability of Representations via Task-Relatedness

Akshay Mehra, Yunbei Zhang, Jihun Hamm

The growing popularity of transfer learning, due to the availability of models pre-trained on vast amounts of data, makes it imperative to understand when the knowledge of these pr…

cs.LG2022★ 1 cited

Understanding the Robustness of Multi-Exit Models under Common Corruptions

Akshay Mehra, Skyler Seto, Navdeep Jaitly +1

Multi-Exit models (MEMs) use an early-exit strategy to improve the accuracy and efficiency of deep neural networks (DNNs) by allowing samples to exit the network before the last la…

cs.LG2022

On Certifying and Improving Generalization to Unseen Domains

Akshay Mehra, Bhavya Kailkhura, Pin-Yu Chen +1

Domain Generalization (DG) aims to learn models whose performance remains high on unseen domains encountered at test-time by using data from multiple related source domains. Many e…

cs.LG2021★ 5 cited

Understanding the Limits of Unsupervised Domain Adaptation via Data Poisoning

Akshay Mehra, Bhavya Kailkhura, Pin-Yu Chen +1

Unsupervised domain adaptation (UDA) enables cross-domain learning without target domain labels by transferring knowledge from a labeled source domain whose distribution differs fr…

cs.LG2021★ 3 cited

Machine Learning with Electronic Health Records is vulnerable to Backdoor Trigger Attacks

Byunggill Joe, Akshay Mehra, Insik Shin +1

Electronic Health Records (EHRs) provide a wealth of information for machine learning algorithms to predict the patient outcome from the data including diagnostic information, vita…

cs.LG2020

How Robust are Randomized Smoothing based Defenses to Data Poisoning?

Akshay Mehra, Bhavya Kailkhura, Pin-Yu Chen +1

Predictions of certifiably robust classifiers remain constant in a neighborhood of a point, making them resilient to test-time attacks with a guarantee. In this work, we present a…

cs.LG2019★ 22 cited

Penalty Method for Inversion-Free Deep Bilevel Optimization

Akshay Mehra, Jihun Hamm

Solving a bilevel optimization problem is at the core of several machine learning problems such as hyperparameter tuning, data denoising, meta- and few-shot learning, and training-…

cs.LG2018★ 1 cited

Fast Interactive Image Retrieval using large-scale unlabeled data

Akshay Mehra, Jihun Hamm, Mikhail Belkin

An interactive image retrieval system learns which images in the database belong to a user's query concept, by analyzing the example images and feedback provided by the user. The c…