Publications

Wiley StatsRef: Statistics Reference2026

Bayesian Coresets

Rajiv Khanna

ICML Workshop (Demo)2026

Curvature-Aware Active Statistical Inference : Reducing Labeling via Data Coherence

Pinaki Mohanty, Rajiv Khanna

ICML Workshop (Demo)2026

Adaptive Stratified Active Statistical Inference

Pinaki Mohanty, Rajiv Khanna

COLT2026

Spectral Valleys and Sharp Failures in Greedy Determinant Maximization

Rajiv Khanna

ICML2026

Consistent Diffusion Language Models

Syed Hasan Amin Mahmood, Yuan Gao, Yaser Souri, Subhojit Som, Ming Yin, Rajiv Khanna, Xia Song

ACL (Findings)2026

From Fallback to Frontline: When Can LLMs be Superior Annotators of Human Perspectives?

Syed Hasan Amin Mahmood, Harry Yizhou Tian, Xiaoni Duan, Chien-Ju Ho, Rajiv Khanna, Ming Yin

ICLR2026

Sharpness-Aware Machine Unlearning

Haoran Tang, Rajiv Khanna

ICLR2026

Membership Privacy Risks of Sharpness Aware Minimization

Young In Kim, Andrea Agiollo, Pratiksha Agrawal, Johannes O. Royset, Rajiv Khanna

AAAI2026

Align When They Want, Complement When They Need! Human-Centered Ensembles for Adaptive Human-AI Collaboration

Syed Hasan Amin Mahmood, Ming Yin, Rajiv Khanna

Neurips2025

Structure-Aware Spectral Sparsification via Uniform Edge Sampling

Kaiwen He, Petros Drineas, Rajiv Khanna

Neurips2025

A Unified Stability Analysis of SAM vs SGD: Role of Data Coherence and Emergence of Simplicity Bias

Wei-kai Chang, Rajiv Khanna

Neurips2025

Stable Coresets via Posterior Sampling: Aligning Induced and Full Loss Landscapes

Wei-kai Chang, Rajiv Khanna

KDD2025

On the Support Vector Effect in DNNs: Rethinking Data Selection and Attribution

Syed Hasan Amin Mahmood, Ming Yin, Rajiv Khanna

Neurips2024

The Space Complexity of Approximating Logistic Loss

Gregory Dexter, Petros Drineas, Rajiv Khanna

KDD2024

Approximating Memorization Using Loss Surface Geometry for Dataset Pruning and Summarization

Andrea Agiollo*, Young In Kim*, Rajiv Khanna

ICLR2024

A Precise Characterization of SGD Stability Using Loss Surface Geometry

Gregory Dexter, Borja Ocejo, Sathiya Keerthi, Aman Gupta, Ayan Acharya, Rajiv Khanna

ICML2023

On Memorization and Privacy risks of Sharpness Aware Minimization

Young In Kim, Pratiksha Agrawal, Johannes O Royset, Rajiv Khanna

COLT2023

Generalization Guarantees via Algorithm-dependent Rademacher Complexity

Sarah Sachs, Tim van Erven, Liam Hodgkinson, Rajiv Khanna, Umut Simsekli

AISTATS2023

Fast Feature Selection with Fairness Constraints

Francesco Quinzan, Rajiv Khanna, Moshik Hershcovitch, Sarel Cohen, Daniel G. Waddington, Tobias Friedrich, Michael W. Mahoney

ICML2022

Generalization Bounds using Lower Tail Exponents in Stochastic Optimizers

Liam Hodgkinson, Umut Simsekli, Rajiv Khanna, Michael Mahoney

UAI (Oral)2021

Geometric Rates of Convergence for Kernel-based Sampling Algorithms

Rajiv Khanna, Liam Hodgkinson, Michael Mahoney

UAI2021

LocalNewton: Reducing Communication Bottleneck for Distributed Learning

Vipul Gupta, Avishek Ghosh, Michal Derezinski, Rajiv Khanna, Ramchandran Kannan, Michael Mahoney

AISTATS (Oral)2021

Bayesian Coresets: An Optimization Perspective

Yibo Zhang, Rajiv Khanna, Anastasios Kyrillidis, Oluwasanmi Koyejo

ICLR2021

Adversarially-trained deep nets transfer better

Francisco Utrera, Evan Kravitz, N Benjamin Erichson, Rajiv Khanna, Michael W Mahoney

Neurips (Best paper award. Top 3 papers out of over 9400 submissions)2020

Improved guarantees and a multiple-descent curve for Column Subset Selection and the Nystrom method

Michal Derezinski, Rajiv Khanna, Michael W Mahoney

Neurips2020

Boundary thickness and robustness in learning models

Yaoqing Yang, Rajiv Khanna, Yaodong Yu, Amir Gholami, Kurt Keutzer, Joseph E Gonzalez, Kannan Ramchandran, Michael W Mahoney

Neurips2019

Learning Sparse Distributions using Iterative Hard Thresholding

Yibo Zhang, Rajiv Khanna, Anastasios Kyrillidis, Oluwasanmi Koyejo

AISTATS2019

Interpreting black box predictions using fisher kernels

Rajiv Khanna, Been Kim, Joydeep Ghosh, Oluwasanmi Koyejo

NIPS (Spotlight)2018

Boosting Black Box Variational Inference

Francesco Locatello, Gideon Dresdner, Rajiv Khanna, Isabel Valera, Gunnar Rätsch

AISTATS2018

Provable Accelerated Iterative Hard Thresholding

Rajiv Khanna, Anastasios Kyrillidis

SDM2018

Co-regularized Monotone Retargeting for Semi-supervised LeTOR

Shalmali Joshi, Rajiv Khanna, Joydeep Ghosh

AISTATS2017

Scalable Greedy Feature Selection via Weak Submodularity

Rajiv Khanna, Ethan Elenberg, Alex Dimakis, Sahand Neghaban, Joydeep Ghosh

AISTATS2017

Information Projection and Approximate Inference for Structured Sparse Variables

Rajiv Khanna, Joydeep Ghosh, Russell A. Poldrack, Oluwasanmi Koyejo

AISTATS2017

A Unified Analysis of Frank Wolfe and Matching Pursuit

Francesco Locatello, Rajiv Khanna, Michael Tschannen, Martin Jaggi

SDM2017

A Deflation Method for Structured Probabilistic PCA

Rajiv Khanna, Joydeep Ghosh, Russell A. Poldrack, Oluwasanmi Koyejo

NIPS (Oral)2016

Examples are not Enough, Learn to Criticize! Criticism for Interpretability

Been Kim*, Rajiv Khanna*, Oluwasanmi Koyejo*

AISTATS (Oral)2015

Sparse Submodular Probabilistic PCA

Rajiv Khanna, Joydeep Ghosh, Russell A. Poldrack, Oluwasanmi O. Koyejo

NIPS2014

On Prior Distributions and Approximate Inference for Structured Variables

Oluwasanmi O. Koyejo, Rajiv Khanna, Joydeep Ghosh, Russell A. Poldrack

Preprint2015

DPM: A State Space Model for Large-Scale Direct Marketing

Yubin Park, Rajiv Khanna, Joydeep Ghosh, Daniel Mihalko. CoRR abs/1507.01135

IEEE BigData Conference2013

Parallel matrix factorization for binary response

Rajiv Khanna, Liang Zhang, Deepak Agarwal, Bee-Chung Chen

KDD2010

Estimating rates of rare events with multiple hierarchies through scalable log-linear models

Deepak Agarwal, Rahul Agrawal, Rajiv Khanna, Nagaraj Kota

CIKM2009

Translating relevance scores to probabilities for contextual advertising

Deepak Agarwal, Evgeniy Gabrilovich, Robert Hall, Vanja Josifovski, Rajiv Khanna

KDD2008

Structured learning for non-smooth ranking losses

Soumen Chakrabarti, Rajiv Khanna, Uma Sawant, Chiru Bhattacharyya