Bhatia, Kush

26 publications

ICLRW 2024 Cookbook: A Framework for Improving LLM Generative Abilities via Programmatic Data Generating Templates Avanika Narayan, Mayee F Chen, Kush Bhatia, Christopher Re

ICLR 2024 The Hedgehog & the Porcupine: Expressive Linear Attentions with SoftMax Mimicry Michael Zhang, Kush Bhatia, Hermann Kumbong, Christopher Re

NeurIPS 2023 A Case for Reframing Automated Medical Image Classification as Segmentation Sarah Hooper, Mayee Chen, Khaled Saab, Kush Bhatia, Curtis Langlotz, Christopher Ré

ICLR 2023 Ask Me Anything: A Simple Strategy for Prompting Language Models Simran Arora, Avanika Narayan, Mayee F Chen, Laurel Orr, Neel Guha, Kush Bhatia, Ines Chami, Christopher Re

NeurIPSW 2023 Does In-Context Operator Learning Generalize to Domain-Shifted Settings? Jerry Weihong Liu, N. Benjamin Erichson, Kush Bhatia, Michael W. Mahoney, Christopher Re

NeurIPS 2023 Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification Neel Guha, Mayee Chen, Kush Bhatia, Azalia Mirhoseini, Frederic Sala, Christopher Ré

ICLR 2023 On the Sensitivity of Reward Inference to Misspecified Human Models Joey Hong, Kush Bhatia, Anca Dragan

AISTATS 2023 Reward Learning as Doubly Nonparametric Bandits: Optimal Design and Scaling Laws Kush Bhatia, Wenshuo Guo, Jacob Steinhardt

NeurIPS 2023 Skill-It! a Data-Driven Skills Framework for Understanding and Training Language Models Mayee Chen, Nicholas Roberts, Kush Bhatia, Jue Wang, Ce Zhang, Frederic Sala, Christopher Ré

NeurIPSW 2023 SuperHF: Supervised Iterative Learning from Human Feedback Gabriel Mukobi, Peter Chatain, Su Fong, Robert Windesheim, Gitta Kutyniok, Kush Bhatia, Silas Alberti

NeurIPS 2023 TART: A Plug-and-Play Transformer Module for Task-Agnostic Reasoning Kush Bhatia, Avanika Narayan, Christopher M De Sa, Christopher Ré

NeurIPSW 2023 TART: A Plug-and-Play Transformer Module for Task-Agnostic Reasoning Kush Bhatia, Avanika Narayan, Christopher De Sa, Christopher Re

ICML 2022 Congested Bandits: Optimal Routing via Short-Term Resets Pranjal Awasthi, Kush Bhatia, Sreenivas Gollapudi, Kostas Kollias

ICLR 2022 The Effects of Reward Misspecification: Mapping and Mitigating Misaligned Models Alexander Pan, Kush Bhatia, Jacob Steinhardt

NeurIPSW 2021 The Effects of Reward Misspecification: Mapping and Mitigating Misaligned Models Alexander Pan, Kush Bhatia, Jacob Steinhardt

JMLR 2020 Derivative-Free Methods for Policy Optimization: Guarantees for Linear Quadratic Systems Dhruv Malik, Ashwin Pananjady, Kush Bhatia, Koulik Khamaru, Peter L. Bartlett, Martin J. Wainwright

NeurIPS 2020 Online Learning with Dynamics: A Minimax Perspective Kush Bhatia, Karthik Sridharan

NeurIPS 2020 Preference Learning Along Multiple Criteria: A Game-Theoretic Perspective Kush Bhatia, Ashwin Pananjady, Peter L. Bartlett, Anca Dragan, Martin J. Wainwright

COLT 2019 Adaptive Hard Thresholding for Near-Optimal Consistent Robust Regression Arun Sai Suggala, Kush Bhatia, Pradeep Ravikumar, Prateek Jain

AISTATS 2019 Derivative-Free Methods for Policy Optimization: Guarantees for Linear Quadratic Systems Dhruv Malik, Ashwin Pananjady, Kush Bhatia, Koulik Khamaru, Peter Bartlett, Martin Wainwright

ICML 2019 POLITEX: Regret Bounds for Policy Iteration Using Expert Prediction Yasin Abbasi-Yadkori, Peter Bartlett, Kush Bhatia, Nevena Lazic, Csaba Szepesvari, Gellert Weisz

NeurIPS 2018 FastGRNN: A Fast, Accurate, Stable and Tiny Kilobyte Sized Gated Recurrent Neural Network Aditya Kusupati, Manish Singh, Kush Bhatia, Ashish Kumar, Prateek Jain, Manik Varma

NeurIPS 2018 Gen-Oja: Simple & Efficient Algorithm for Streaming Generalized Eigenvector Computation Kush Bhatia, Aldo Pacchiano, Nicolas Flammarion, Peter L Bartlett, Michael I Jordan

NeurIPS 2017 Consistent Robust Regression Kush Bhatia, Prateek Jain, Parameswaran Kamalaruban, Purushottam Kar

NeurIPS 2015 Robust Regression via Hard Thresholding Kush Bhatia, Prateek Jain, Purushottam Kar

NeurIPS 2015 Sparse Local Embeddings for Extreme Multi-Label Classification Kush Bhatia, Himanshu Jain, Purushottam Kar, Manik Varma, Prateek Jain