I completed my Ph.D. in Computer Science from Tulane University in 2024, under the guidance of Prof. Jihun Hamm. The primary thrusts of my research are summarized as follows. The first aspect of my research focuses on rigorously analyzing how different performance metrics (such as accuracy, adversarial robustness, and transferability to downstream tasks) are affected when training and test distributions differ. The second aspect concentrates on developing algorithms inspired by these theoretical insights to improve the performance of the ML models under distribution shifts. Overall, my research analyzes the success of learning models in in-domain in-task and cross-domain cross-task settings and analyzes the impact of distribution shifts in these settings.

Research Interest

To deepen the understanding of machine learning models' robustness to different distribution shifts such as shifts induced by corrupted data, shifts induced by adversarial attacks, etc..

Publications

• “Understanding the Transferability of Representations via Task-Relatedness”.
  Code
  Akshay Mehra, Yunbei Zhang, and Jihun Hamm.
  Neural Information Processing Systems (NeurIPS) 2024.
• “OT-VP: Optimal Transport-guided Visual Prompting for Test-Time Adaptation”.
  Yunbei Zhang, Akshay Mehra, and Jihun Hamm.
  Winter Conference on Applications of Computer Vision (WACV) 2025.
• “Test-time Assessment of a Model’s Performance on Unseen Domains via Optimal Transport”.
  Code
  Akshay Mehra, Yunbei Zhang, and Jihun Hamm.
  Computer Vision and Pattern Recognition (CVPR) Workshop on Fair, Data-efficient, and Trusted Computer Vision 2024.
• “On the Fly Neural Style Smoothing for Risk-Averse Domain Generalization”.
  Code
  Akshay Mehra, Yunbei Zhang, Bhavya Kailkhura, and Jihun Hamm.
  Winter Conference on Applications of Computer Vision (WACV) 2024.
• “A Spectral View of Randomized Smoothing under Common Corruptions: Benchmarking and Improving Certified Robustness”.
  Jiachen Sun, Akshay Mehra, Bhavya Kailkhura, Pin-Yu Chen, Dan Hendrycks, Jihun Hamm and Z. Morley Mao.
  European Conference on Computer Vision (ECCV) 2022.
• “Understanding the Limits of Unsupervised Domain Adaptation via Data Poisoning”.
  Code, Poster
  Akshay Mehra, Bhavya Kailkhura, Pin-Yu Chen and Jihun Hamm.
  Neural Information Processing Systems (NeurIPS) 2021.
• “How Robust are Randomized Smoothing based Defenses to Data Poisoning?”.
  Code, Poster
  Akshay Mehra, Bhavya Kailkhura, Pin-Yu Chen and Jihun Hamm.
  Computer Vision and Pattern Recognition (CVPR) 2021.
• “Penalty Method for Inversion-Free Deep Bilevel Optimization”.
  Code, Poster
  Akshay Mehra, Jihun Hamm.
  Asian Conference on Machine Learning (ACML) 2021.

Preprints

• “Understanding the Robustness of Multi-Exit Models under Common Corruptions”.
  Akshay Mehra, Skyler Seto, Navdeep Jaitly and Barry-John Theobald
• “Do Domain Generalization Methods Generalize Well?”.
  Code, Poster
  Akshay Mehra, Bhavya Kailkhura, Pin-Yu Chen and Jihun Hamm.
  Workshop on Machine Learning Safety at Neural Information Processing Systems (NeurIPS) 2022.
• “Machine vs Machine: Minimax-Optimal Defense Against Adversarial Examples”.
  Jihun Hamm and Akshay Mehra.