Menon, Aditya Krishna

42 publications

ICLR 2025 Better Autoregressive Regression with LLMs via Regression-Aware Fine-Tuning Michal Lukasik, Zhao Meng, Harikrishna Narasimhan, Yin-Wen Chang, Aditya Krishna Menon, Felix Yu, Sanjiv Kumar

ICML 2025 Bipartite Ranking from Multiple Labels: On Loss Versus Label Aggregation Michal Lukasik, Lin Chen, Harikrishna Narasimhan, Aditya Krishna Menon, Wittawat Jitkrittum, Felix X. Yu, Sashank J. Reddi, Gang Fu, Mohammadhossein Bateni, Sanjiv Kumar

ICLR 2025 Faster Cascades via Speculative Decoding Harikrishna Narasimhan, Wittawat Jitkrittum, Ankit Singh Rawat, Seungyeon Kim, Neha Gupta, Aditya Krishna Menon, Sanjiv Kumar

ICLRW 2025 Universal LLM Routing with Correctness-Based Representation Wittawat Jitkrittum, Harikrishna Narasimhan, Ankit Singh Rawat, Jeevesh Juneja, Zifeng Wang, Chen-Yu Lee, Pradeep Shenoy, Rina Panigrahy, Aditya Krishna Menon, Sanjiv Kumar

ICLR 2024 DistillSpec: Improving Speculative Decoding via Knowledge Distillation Yongchao Zhou, Kaifeng Lyu, Ankit Singh Rawat, Aditya Krishna Menon, Afshin Rostamizadeh, Sanjiv Kumar, Jean-François Kagy, Rishabh Agarwal

ICMLW 2024 Efficient Document Ranking with Learnable Late Interactions Himanshu Jain, Ziwei Ji, Sashank J. Reddi, Ankit Singh Rawat, Felix Yu, Aditya Krishna Menon, Sadeep Jayasumana

ICMLW 2024 Efficient Document Ranking with Learnable Late Interactions Himanshu Jain, Ziwei Ji, Ankit Singh Rawat, Andreas Veit, Sadeep Jayasumana, Sashank J. Reddi, Aditya Krishna Menon, Felix Yu

ICLR 2024 Language Model Cascades: Token-Level Uncertainty and Beyond Neha Gupta, Harikrishna Narasimhan, Wittawat Jitkrittum, Ankit Singh Rawat, Aditya Krishna Menon, Sanjiv Kumar

ICLR 2024 Learning to Reject Meets Long-Tail Learning Harikrishna Narasimhan, Aditya Krishna Menon, Wittawat Jitkrittum, Neha Gupta, Sanjiv Kumar

ICLR 2024 Plugin Estimators for Selective Classification with Out-of-Distribution Detection Harikrishna Narasimhan, Aditya Krishna Menon, Wittawat Jitkrittum, Sanjiv Kumar

ICLR 2024 The Importance of Feature Preprocessing for Differentially Private Linear Optimization Ziteng Sun, Ananda Theertha Suresh, Aditya Krishna Menon

ICLR 2024 Think Before You Speak: Training Language Models with Pause Tokens Sachin Goyal, Ziwei Ji, Ankit Singh Rawat, Aditya Krishna Menon, Sanjiv Kumar, Vaishnavh Nagarajan

ICML 2024 USTAD: Unified Single-Model Training Achieving Diverse Scores for Information Retrieval Seungyeon Kim, Ankit Singh Rawat, Manzil Zaheer, Wittawat Jitkrittum, Veeranjaneyulu Sadhanala, Sadeep Jayasumana, Aditya Krishna Menon, Rob Fergus, Sanjiv Kumar

TMLR 2024 What Do Larger Image Classifiers Memorise? Michal Lukasik, Vaishnavh Nagarajan, Ankit Singh Rawat, Aditya Krishna Menon, Sanjiv Kumar

UAI 2023 Robust Distillation for Worst-Class Performance: On the Interplay Between Teacher and Student Objectives Serena Wang, Harikrishna Narasimhan, Yichen Zhou, Sara Hooker, Michal Lukasik, Aditya Krishna Menon

ICLR 2023 Supervision Complexity and Its Role in Knowledge Distillation Hrayr Harutyunyan, Ankit Singh Rawat, Aditya Krishna Menon, Seungyeon Kim, Sanjiv Kumar

NeurIPSW 2023 Think Before You Speak: Training Language Models with Pause Tokens Sachin Goyal, Ziwei Ji, Ankit Singh Rawat, Aditya Krishna Menon, Sanjiv Kumar, Vaishnavh Nagarajan

JMLR 2022 Interval-Censored Hawkes Processes Marian-Andrei Rizoiu, Alexander Soen, Shidi Li, Pio Calderon, Leanne J. Dong, Aditya Krishna Menon, Lexing Xie

TMLR 2022 Teacher’s Pet: Understanding and Mitigating Biases in Distillation Michal Lukasik, Srinadh Bhojanapalli, Aditya Krishna Menon, Sanjiv Kumar

NeurIPSW 2022 When Does Mixup Promote Local Linearity in Learned Representations? Arslan Chaudhry, Aditya Krishna Menon, Andreas Veit, Sadeep Jayasumana, Srikumar Ramalingam, Sanjiv Kumar

ICLR 2021 Coping with Label Shift via Distributionally Robust Optimisation Jingzhao Zhang, Aditya Krishna Menon, Andreas Veit, Srinadh Bhojanapalli, Sanjiv Kumar, Suvrit Sra

ICLR 2021 Long-Tail Learning via Logit Adjustment Aditya Krishna Menon, Sadeep Jayasumana, Ankit Singh Rawat, Himanshu Jain, Andreas Veit, Sanjiv Kumar

ICLR 2021 Overparameterisation and Worst-Case Generalisation: Friend or Foe? Aditya Krishna Menon, Ankit Singh Rawat, Sanjiv Kumar

ICLR 2020 Can Gradient Clipping Mitigate Label Noise? Aditya Krishna Menon, Ankit Singh Rawat, Sashank J. Reddi, Sanjiv Kumar

AAAI 2019 Comparative Document Summarisation via Classification Umanga Bista, Alexander Patrick Mathews, Minjeong Shin, Aditya Krishna Menon, Lexing Xie

ICLR 2019 On the Minimal Supervision for Training Any Binary Classifier from Only Unlabeled Data Nan Lu, Gang Niu, Aditya Krishna Menon, Masashi Sugiyama

MLJ 2019 The Risk of Trivial Solutions in Bipartite Top Ranking Aditya Krishna Menon

NeurIPS 2018 A Loss Framework for Calibrated Anomaly Detection Aditya Krishna Menon, Robert C. Williamson

MLJ 2018 Learning from Binary Labels with Instance-Dependent Noise Aditya Krishna Menon, Brendan van Rooyen, Nagarajan Natarajan

AAAI 2018 Proper Loss Functions for Nonlinear Hawkes Processes Aditya Krishna Menon, Young Lee

AAAI 2017 Low-Rank Linear Cold-Start Recommendation from Social Data Suvash Sedhain, Aditya Krishna Menon, Scott Sanner, Lexing Xie, Darius Braziunas

CVPR 2017 Making Deep Neural Networks Robust to Label Noise: A Loss Correction Approach Giorgio Patrini, Alessandro Rozza, Aditya Krishna Menon, Richard Nock, Lizhen Qu

ECML-PKDD 2017 Robust, Deep and Inductive Anomaly Detection Raghavendra Chalapathy, Aditya Krishna Menon, Sanjay Chawla

JMLR 2016 Bipartite Ranking: A Risk-Theoretic Perspective Aditya Krishna Menon, Robert C. Williamson

AAAI 2016 On the Effectiveness of Linear Models for One-Class Collaborative Filtering Suvash Sedhain, Aditya Krishna Menon, Scott Sanner, Darius Braziunas

IJCAI 2016 Practical Linear Models for Large-Scale One-Class Collaborative Filtering Suvash Sedhain, Hung Bui, Jaya Kawale, Nikos Vlassis, Branislav Kveton, Aditya Krishna Menon, Trung Bui, Scott Sanner

COLT 2014 Bayes-Optimal Scorers for Bipartite Ranking Aditya Krishna Menon, Robert C. Williamson

MLJ 2014 Detecting Inappropriate Access to Electronic Health Records Using Collaborative Filtering Aditya Krishna Menon, Xiaoqian Jiang, Jihoon Kim, Jaideep Vaidya, Lucila Ohno-Machado

MLJ 2013 Beam Search Algorithms for Multilabel Learning Abhishek Kumar, Shankar Vembu, Aditya Krishna Menon, Charles Elkan

ECML-PKDD 2012 Learning and Inference in Probabilistic Classifier Chains with Beam Search Abhishek Kumar, Shankar Vembu, Aditya Krishna Menon, Charles Elkan

ICML 2012 Predicting Accurate Probabilities with a Ranking Loss Aditya Krishna Menon, Xiaoqian Jiang, Shankar Vembu, Charles Elkan, Lucila Ohno-Machado

ECML-PKDD 2011 Link Prediction via Matrix Factorization Aditya Krishna Menon, Charles Elkan