| Sparse local embeddings for extreme multi-label classification K Bhatia, H Jain, P Kar, M Varma, P Jain Advances in neural information processing systems 28, 2015 | 565 | 2015 |
| Fastgrnn: A fast, accurate, stable and tiny kilobyte sized gated recurrent neural network A Kusupati, M Singh, K Bhatia, A Kumar, P Jain, M Varma Advances in neural information processing systems 31, 2018 | 239 | 2018 |
| The extreme classification repository: Multi-label datasets and code K Bhatia, K Dahiya, H Jain, P Kar, A Mittal, Y Prabhu, M Varma URL http://manikvarma. org/downloads/XC/XMLRepository. html, 2016 | 233* | 2016 |
| Derivative-free methods for policy optimization: Guarantees for linear quadratic systems D Malik, A Pananjady, K Bhatia, K Khamaru, PL Bartlett, MJ Wainwright Journal of Machine Learning Research 21 (21), 1-51, 2020 | 232 | 2020 |
| Ask me anything: A simple strategy for prompting language models S Arora, A Narayan, MF Chen, L Orr, N Guha, K Bhatia, I Chami, F Sala, ... arXiv preprint arXiv:2210.02441, 2022 | 222 | 2022 |
| Robust regression via hard thresholding K Bhatia, P Jain, P Kar Advances in neural information processing systems 28, 2015 | 199 | 2015 |
| The effects of reward misspecification: Mapping and mitigating misaligned models A Pan, K Bhatia, J Steinhardt arXiv preprint arXiv:2201.03544, 2022 | 170 | 2022 |
| Politex: Regret bounds for policy iteration using expert prediction Y Abbasi-Yadkori, P Bartlett, K Bhatia, N Lazic, C Szepesvari, G Weisz International Conference on Machine Learning, 3692-3702, 2019 | 157 | 2019 |
| Establishing appropriate trust via critical states SH Huang, K Bhatia, P Abbeel, AD Dragan RSJ International Conference on Intelligent Robots and Systems (IROS), 3929-3936, 2018 | 148 | 2018 |
| Consistent Robust Regression. K Bhatia, P Jain, P Kamalaruban, P Kar NIPS 2017, 2017 | 118 | 2017 |
| Skill-it! a data-driven skills framework for understanding and training language models M Chen, N Roberts, K Bhatia, J Wang, C Zhang, F Sala, C Ré Advances in Neural Information Processing Systems 36, 36000-36040, 2023 | 55 | 2023 |
| Adaptive hard thresholding for near-optimal consistent robust regression AS Suggala, K Bhatia, P Ravikumar, P Jain Conference on Learning Theory, 2892-2897, 2019 | 52 | 2019 |
| The hedgehog & the porcupine: Expressive linear attentions with softmax mimicry M Zhang, K Bhatia, H Kumbong, C Ré arXiv preprint arXiv:2402.04347, 2024 | 38 | 2024 |
| On the sensitivity of reward inference to misspecified human models J Hong, K Bhatia, A Dragan arXiv preprint arXiv:2212.04717, 2022 | 26 | 2022 |
| Gen-Oja: A Two-time-scale approach for Streaming CCA K Bhatia, A Pacchiano, N Flammarion, PL Bartlett, MI Jordan arXiv preprint arXiv:1811.08393, 2018 | 22* | 2018 |
| Online learning with dynamics: A minimax perspective K Bhatia, K Sridharan Advances in Neural Information Processing Systems 33, 15020-15030, 2020 | 18 | 2020 |
| Preference learning along multiple criteria: A game-theoretic perspective K Bhatia, A Pananjady, P Bartlett, A Dragan, MJ Wainwright Advances in neural information processing systems 33, 7413-7424, 2020 | 16 | 2020 |
| TART: A plug-and-play Transformer module for task-agnostic reasoning K Bhatia, A Narayan, CM De Sa, C Ré Advances in Neural Information Processing Systems 36, 9751-9788, 2023 | 15 | 2023 |
| Bayesian robustness: A nonasymptotic viewpoint K Bhatia, YA Ma, AD Dragan, PL Bartlett, MI Jordan Journal of the American Statistical Association 119 (546), 1112-1123, 2024 | 11* | 2024 |
| Lazy generic cuts D Khandelwal, K Bhatia, C Arora, P Singla Computer Vision and Image Understanding 143, 80-91, 2016 | 11 | 2016 |
