Prateek Jain

According to our database, Prateek Jain authored at least 168 papers between 2004 and 2024.

Collaborative distances:

• Dijkstra number of three.
• Erdős number of three.

2024

Mixture of Nested Experts: Adaptive Processing of Visual Tokens.

Gagan Jain, Nidhi Hegde, Aditya Kusupati, Arsha Nagrani, Shyamal Buch, Prateek Jain, Anurag Arnab, Sujoy Paul

CoRR, 2024

Gecko: Versatile Text Embeddings Distilled from Large Language Models.

Jinhyuk Lee, Zhuyun Dai, Xiaoqi Ren, Blair Chen, Daniel Cer, Jeremy R. Cole, Kai Hui, Michael Boratko, Rajvi Kapadia, Wen Ding, Yi Luan, Sai Meher Karthik Duddu, Gustavo Hernández Ábrego, Weiqiang Shi, Nithi Gupta, Aditya Kusupati, Prateek Jain, Siddhartha Reddy Jonnalagadda, Ming-Wei Chang, Iftekhar Naim

CoRR, 2024

HiRE: High Recall Approximate Top-k Estimation for Efficient LLM Inference.

Yashas Samaga, Varun Yerram, Chong You, Srinadh Bhojanapalli, Sanjiv Kumar, Prateek Jain, Praneeth Netrapalli

CoRR, 2024

Tandem Transformers for Inference Efficient LLMs.

Aishwarya P. S., Pranav Ajit Nair, Yashas Samaga, Toby Boyd, Sanjiv Kumar, Prateek Jain, Praneeth Netrapalli

Proceedings of the Forty-first International Conference on Machine Learning, 2024

Dual-Encoders for Extreme Multi-label Classification.

Nilesh Gupta, Devvrit, Ankit Singh Rawat, Srinadh Bhojanapalli, Prateek Jain, Inderjit S. Dhillon

Proceedings of the Twelfth International Conference on Learning Representations, 2024

LookupViT: Compressing Visual Information to a Limited Number of Tokens.

Rajat Koner, Gagan Jain, Prateek Jain, Volker Tresp, Sujoy Paul

Proceedings of the Computer Vision - ECCV 2024, 2024

End-to-End Neural Network Compression via l1/l2 Regularized Latency Surrogates.

Anshul Nasery, Hardik Shah, Arun Sai Suggala, Prateek Jain

Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2024

Sample-Efficient Personalization: Modeling User Parameters as Low Rank Plus Sparse Components.

Soumyabrata Pal, Prateek Varshney, Gagan Madan, Prateek Jain, Abhradeep Thakurta, Gaurav Aggarwal, Pradeep Shenoy, Gaurav Srivastava

Proceedings of the International Conference on Artificial Intelligence and Statistics, 2024

2023

Efficacy of Dual-Encoders for Extreme Multi-Label Classification.

Nilesh Gupta, Devvrit Khatri, Ankit Singh Rawat, Srinadh Bhojanapalli, Prateek Jain, Inderjit S. Dhillon

CoRR, 2023

EHI: End-to-end Learning of Hierarchical Index for Efficient Dense Retrieval.

Ramnath Kumar, Anshul Mittal, Nilesh Gupta, Aditya Kusupati, Inderjit S. Dhillon, Prateek Jain

CoRR, 2023

MatFormer: Nested Transformer for Elastic Inference.

Devvrit, Sneha Kudugunta, Aditya Kusupati, Tim Dettmers, Kaifeng Chen, Inderjit S. Dhillon, Yulia Tsvetkov, Hannaneh Hajishirzi, Sham M. Kakade, Ali Farhadi, Prateek Jain

CoRR, 2023

End-to-End Neural Network Compression via 𝓁₁/𝓁₂ Regularized Latency Surrogates.

Anshul Nasery, Hardik Shah, Arun Sai Suggala, Prateek Jain

CoRR, 2023

Near Optimal Private and Robust Linear Regression.

Xiyang Liu, Prateek Jain, Weihao Kong, Sewoong Oh, Arun Sai Suggala

CoRR, 2023

AdANNS: A Framework for Adaptive Semantic Search.

Aniket Rege, Aditya Kusupati, Sharan Ranjit S, Alan Fan, Qingqing Cao, Sham M. Kakade, Prateek Jain, Ali Farhadi

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Blocked Collaborative Bandits: Online Collaborative Filtering with Per-Item Budget Constraints.

Soumyabrata Pal, Arun Sai Suggala, Karthikeyan Shanmugam, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Simplicity Bias in 1-Hidden Layer Neural Networks.

Depen Morwani, Jatin Batra, Prateek Jain, Praneeth Netrapalli

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Label Robust and Differentially Private Linear Regression: Computational and Statistical Efficiency.

Xiyang Liu, Prateek Jain, Weihao Kong, Sewoong Oh, Arun Sai Suggala

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Multi-User Reinforcement Learning with Low Rank Rewards.

Dheeraj Mysore Nagaraj, Suhas S. Kowshik, Naman Agarwal, Praneeth Netrapalli, Prateek Jain

Proceedings of the International Conference on Machine Learning, 2023

Multi-Task Differential Privacy Under Distribution Skew.

Walid Krichene, Prateek Jain, Shuang Song, Mukund Sundararajan, Abhradeep Guha Thakurta, Li Zhang

Proceedings of the International Conference on Machine Learning, 2023

Online Low Rank Matrix Completion.

Soumyabrata Pal, Prateek Jain

Proceedings of the Eleventh International Conference on Learning Representations, 2023

Treeformer: Dense Gradient Trees for Efficient Attention Computation.

Lovish Madaan, Srinadh Bhojanapalli, Himanshu Jain, Prateek Jain

Proceedings of the Eleventh International Conference on Learning Representations, 2023

Feature Reconstruction From Outputs Can Mitigate Simplicity Bias in Neural Networks.

Sravanti Addepalli, Anshul Nasery, Venkatesh Babu Radhakrishnan, Praneeth Netrapalli, Prateek Jain

Proceedings of the Eleventh International Conference on Learning Representations, 2023

Optimal Algorithms for Latent Bandits with Cluster Structure.

Soumyabrata Pal, Arun Sai Suggala, Karthikeyan Shanmugam, Prateek Jain

Proceedings of the International Conference on Artificial Intelligence and Statistics, 2023

2022

Learning Accurate Decision Trees with Bandit Feedback via Quantized Gradient Descent.

Ajaykrishna Karthikeyan, Naman Jain, Nagarajan Natarajan, Prateek Jain

Trans. Mach. Learn. Res., 2022

Private and Efficient Meta-Learning with Low Rank and Sparse Decomposition.

Soumyabrata Pal, Prateek Varshney, Prateek Jain, Abhradeep Guha Thakurta, Gagan Madan, Gaurav Aggarwal, Pradeep Shenoy, Gaurav Srivastava

CoRR, 2022

Learning an Invertible Output Mapping Can Mitigate Simplicity Bias in Neural Networks.

Sravanti Addepalli, Anshul Nasery, R. Venkatesh Babu, Praneeth Netrapalli, Prateek Jain

CoRR, 2022

DAFT: Distilling Adversarially Fine-tuned Models for Better OOD Generalization.

Anshul Nasery, Sravanti Addepalli, Praneeth Netrapalli, Prateek Jain

CoRR, 2022

(Nearly) Optimal Private Linear Regression via Adaptive Clipping.

Prateek Varshney, Abhradeep Thakurta, Prateek Jain

CoRR, 2022

MET: Masked Encoding for Tabular Data.

Kushal Majmundar, Sachin Goyal, Praneeth Netrapalli, Prateek Jain

CoRR, 2022

Matryoshka Representations for Adaptive Deployment.

Aditya Kusupati, Gantavya Bhatt, Aniket Rege, Matthew Wallingford, Aditya Sinha, Vivek Ramanujan, William Howard-Snyder, Kaifeng Chen, Sham M. Kakade, Prateek Jain, Ali Farhadi

CoRR, 2022

DP-PCA: Statistically Optimal and Differentially Private PCA.

Xiyang Liu, Weihao Kong, Prateek Jain, Sewoong Oh

Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022

Matryoshka Representation Learning.

Aditya Kusupati, Gantavya Bhatt, Aniket Rege, Matthew Wallingford, Aditya Sinha, Vivek Ramanujan, William Howard-Snyder, Kaifeng Chen, Sham M. Kakade, Prateek Jain, Ali Farhadi

Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022

S3GC: Scalable Self-Supervised Graph Clustering.

Devvrit, Aditya Sinha, Inderjit S. Dhillon, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022

Reproducibility in Optimization: Theoretical Framework and Limits.

Kwangjun Ahn, Prateek Jain, Ziwei Ji, Satyen Kale, Praneeth Netrapalli, Gil I. Shamir

Proceedings of the Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, 2022

IGLU: Efficient GCN Training via Lazy Updates.

S. Deepak Narayanan, Aditya Sinha, Prateek Jain, Purushottam Kar, Sundararajan Sellamanickam

Proceedings of the Tenth International Conference on Learning Representations, 2022

Online Target Q-learning with Reverse Experience Replay: Efficiently finding the Optimal Policy for Linear MDPs.

Naman Agarwal, Syomantak Chaudhuri, Prateek Jain, Dheeraj Mysore Nagaraj, Praneeth Netrapalli

Proceedings of the Tenth International Conference on Learning Representations, 2022

(Nearly) Optimal Private Linear Regression for Sub-Gaussian Data via Adaptive Clipping.

Prateek Varshney, Abhradeep Thakurta, Prateek Jain

Proceedings of the Conference on Learning Theory, 2-5 July 2022, London, UK., 2022

Robust Training in High Dimensions via Block Coordinate Geometric Median Descent.

Anish Acharya, Abolfazl Hashemi, Prateek Jain, Sujay Sanghavi, Inderjit S. Dhillon, Ufuk Topcu

Proceedings of the International Conference on Artificial Intelligence and Statistics, 2022

2021

Making the Last Iterate of SGD Information Theoretically Optimal.

Prateek Jain, Dheeraj M. Nagaraj, Praneeth Netrapalli

SIAM J. Optim., 2021

Node-Level Differentially Private Graph Neural Networks.

Ameya Daigavane, Gagan Madan, Aditya Sinha, Abhradeep Guha Thakurta, Gaurav Aggarwal, Prateek Jain

CoRR, 2021

Sample Efficient Linear Meta-Learning by Alternating Minimization.

Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong Oh

CoRR, 2021

Statistically and Computationally Efficient Linear Meta-representation Learning.

Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong Oh

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

Do Input Gradients Highlight Discriminative Features?

Harshay Shah, Prateek Jain, Praneeth Netrapalli

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

LLC: Accurate, Multi-purpose Learnt Low-dimensional Binary Codes.

Aditya Kusupati, Matthew Wallingford, Vivek Ramanujan, Raghav Somani, Jae Sung Park, Krishna Pillutla, Prateek Jain, Sham M. Kakade, Ali Farhadi

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

Near-optimal Offline and Streaming Algorithms for Learning Non-Linear Dynamical Systems.

Suhas S. Kowshik, Dheeraj Nagaraj, Prateek Jain, Praneeth Netrapalli

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

Differentially Private Model Personalization.

Prateek Jain, John Rush, Adam D. Smith, Shuang Song, Abhradeep Guha Thakurta

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

Streaming Linear System Identification with Reverse Experience Replay.

Prateek Jain, Suhas S. Kowshik, Dheeraj Nagaraj, Praneeth Netrapalli

Proceedings of the Advances in Neural Information Processing Systems 34: Annual Conference on Neural Information Processing Systems 2021, 2021

Optimal regret algorithm for Pseudo-1d Bandit Convex Optimization.

Aadirupa Saha, Nagarajan Natarajan, Praneeth Netrapalli, Prateek Jain

Proceedings of the 38th International Conference on Machine Learning, 2021

Private Alternating Least Squares: Practical Private Matrix Completion with Tighter Rates.

Steve Chien, Prateek Jain, Walid Krichene, Steffen Rendle, Shuang Song, Abhradeep Thakurta, Li Zhang

Proceedings of the 38th International Conference on Machine Learning, 2021

2020

Nonconvex Optimization for Signal Processing and Machine Learning [From the Guest Editors].

Anthony Man-Cho So, Prateek Jain, Wing-Kin Ma, Gesualdo Scutari

IEEE Signal Process. Mag., 2020

Programming by Rewards.

Nagarajan Natarajan, Ajaykrishna Karthikeyan, Prateek Jain, Ivan Radicek, Sriram K. Rajamani, Sumit Gulwani, Johannes Gehrke

CoRR, 2020

Projection Efficient Subgradient Method and Optimal Nonsmooth Frank-Wolfe Method.

Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong Oh

Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020

The Pitfalls of Simplicity Bias in Neural Networks.

Harshay Shah, Kaustav Tamuly, Aditi Raghunathan, Prateek Jain, Praneeth Netrapalli

Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020

RNNPool: Efficient Non-linear Pooling for RAM Constrained Inference.

Oindrila Saha, Aditya Kusupati, Harsha Vardhan Simhadri, Manik Varma, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020

Least Squares Regression with Markovian Data: Fundamental Limits and Algorithms.

Dheeraj Nagaraj, Xian Wu, Guy Bresler, Prateek Jain, Praneeth Netrapalli

Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020

Soft Threshold Weight Reparameterization for Learnable Sparsity.

Aditya Kusupati, Vivek Ramanujan, Raghav Somani, Mitchell Wortsman, Prateek Jain, Sham M. Kakade, Ali Farhadi

Proceedings of the 37th International Conference on Machine Learning, 2020

DROCC: Deep Robust One-Class Classification.

Sachin Goyal, Aditi Raghunathan, Moksh Jain, Harsha Vardhan Simhadri, Prateek Jain

Proceedings of the 37th International Conference on Machine Learning, 2020

2019

OASIS: ILP-Guided Synthesis of Loop Invariants.

Sahil Bhatia, Saswat Padhi, Nagarajan Natarajan, Rahul Sharma, Prateek Jain

CoRR, 2019

Learning Functions over Sets via Permutation Adversarial Networks.

Chirag Pabbaraju, Prateek Jain

CoRR, 2019

GesturePod: Enabling On-device Gesture-based Interaction for White Cane Users.

Shishir G. Patil, Don Kurian Dennis, Chirag Pabbaraju, Nadeem Shaheer, Harsha Vardhan Simhadri, Vivek Seshadri, Manik Varma, Prateek Jain

Proceedings of the 32nd Annual ACM Symposium on User Interface Software and Technology, 2019

Provable Non-linear Inductive Matrix Completion.

Kai Zhong, Zhao Song, Prateek Jain, Inderjit S. Dhillon

Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019

Efficient Algorithms for Smooth Minimax Optimization.

Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong Oh

Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019

Shallow RNN: Accurate Time-series Classification on Resource Constrained Devices.

Don Kurian Dennis, Durmus Alp Emre Acar, Vikram Mandikal, Vinu Sankar Sadasivan, Venkatesh Saligrama, Harsha Vardhan Simhadri, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019

SGD without Replacement: Sharper Rates for General Smooth Convex Functions.

Dheeraj Nagaraj, Prateek Jain, Praneeth Netrapalli

Proceedings of the 36th International Conference on Machine Learning, 2019

Adaptive Hard Thresholding for Near-optimal Consistent Robust Regression.

Arun Sai Suggala, Kush Bhatia, Pradeep Ravikumar, Prateek Jain

Proceedings of the Conference on Learning Theory, 2019

Open Problem: Do Good Algorithms Necessarily Query Bad Points?

Rong Ge, Prateek Jain, Sham M. Kakade, Rahul Kidambi, Dheeraj M. Nagaraj, Praneeth Netrapalli

Proceedings of the Conference on Learning Theory, 2019

Cost aware Inference for IoT Devices.

Pengkai Zhu, Durmus Alp Emre Acar, Nan Feng, Prateek Jain, Venkatesh Saligrama

Proceedings of the 22nd International Conference on Artificial Intelligence and Statistics, 2019

Learning Natural Programs from a Few Examples in Real-Time.

Nagarajan Natarajan, Danny Simmons, Naren Datha, Prateek Jain, Sumit Gulwani

Proceedings of the 22nd International Conference on Artificial Intelligence and Statistics, 2019

Globally-convergent Iteratively Reweighted Least Squares for Robust Regression Problems.

Bhaskar Mukhoty, Govind Gopakumar, Prateek Jain, Purushottam Kar

Proceedings of the 22nd International Conference on Artificial Intelligence and Statistics, 2019

Distributional Semantics Meets Multi-Label Learning.

Vivek Gupta, Rahul Wadbude, Nagarajan Natarajan, Harish Karnick, Prateek Jain, Piyush Rai

Proceedings of the Thirty-Third AAAI Conference on Artificial Intelligence, 2019

2018

FlashProfile: a framework for synthesizing data profiles.

Saswat Padhi, Prateek Jain, Daniel Perelman, Oleksandr Polozov, Sumit Gulwani, Todd D. Millstein

Proc. ACM Program. Lang., 2018

Nonlinear Inductive Matrix Completion based on One-layer Neural Networks.

Kai Zhong, Zhao Song, Prateek Jain, Inderjit S. Dhillon

CoRR, 2018

Neural-Guided Deductive Search for Real-Time Program Synthesis from Examples.

Ashwin J. Vijayakumar, Abhishek Mohta, Oleksandr Polozov, Dhruv Batra, Prateek Jain, Sumit Gulwani

CoRR, 2018

Support Recovery for Orthogonal Matching Pursuit: Upper and Lower bounds.

Raghav Somani, Chirag Gupta, Prateek Jain, Praneeth Netrapalli

Proceedings of the Advances in Neural Information Processing Systems 31: Annual Conference on Neural Information Processing Systems 2018, 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

Proceedings of the Advances in Neural Information Processing Systems 31: Annual Conference on Neural Information Processing Systems 2018, 2018

Multiple Instance Learning for Efficient Sequential Data Classification on Resource-constrained Devices.

Don Kurian Dennis, Chirag Pabbaraju, Harsha Vardhan Simhadri, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 31: Annual Conference on Neural Information Processing Systems 2018, 2018

Differentially Private Matrix Completion Revisited.

Prateek Jain, Om Dipakbhai Thakkar, Abhradeep Thakurta

Proceedings of the 35th International Conference on Machine Learning, 2018

On the insufficiency of existing momentum schemes for Stochastic Optimization.

Rahul Kidambi, Praneeth Netrapalli, Prateek Jain, Sham M. Kakade

Proceedings of the 6th International Conference on Learning Representations, 2018

Neural-Guided Deductive Search for Real-Time Program Synthesis from Examples.

Ashwin Kalyan, Abhishek Mohta, Oleksandr Polozov, Dhruv Batra, Prateek Jain, Sumit Gulwani

Proceedings of the 6th International Conference on Learning Representations, 2018

Smoothed analysis for low-rank solutions to semidefinite programs in quadratic penalty form.

Srinadh Bhojanapalli, Nicolas Boumal, Prateek Jain, Praneeth Netrapalli

Proceedings of the Conference On Learning Theory, 2018

Accelerating Stochastic Gradient Descent for Least Squares Regression.

Prateek Jain, Sham M. Kakade, Rahul Kidambi, Praneeth Netrapalli, Aaron Sidford

Proceedings of the Conference On Learning Theory, 2018

2017

Partial Hard Thresholding.

Prateek Jain, Ambuj Tewari, Inderjit S. Dhillon

IEEE Trans. Inf. Theory, 2017

Parallelizing Stochastic Gradient Descent for Least Squares Regression: Mini-batching, Averaging, and Model Misspecification.

Prateek Jain, Sham M. Kakade, Rahul Kidambi, Praneeth Netrapalli, Aaron Sidford

J. Mach. Learn. Res., 2017

Non-convex Optimization for Machine Learning.

Prateek Jain, Purushottam Kar

Found. Trends Mach. Learn., 2017

FlashProfile: Interactive Synthesis of Syntactic Profiles.

Saswat Padhi, Prateek Jain, Daniel Perelman, Oleksandr Polozov, Sumit Gulwani, Todd D. Millstein

CoRR, 2017

Thresholding based Efficient Outlier Robust PCA.

Yeshwanth Cherapanamjeri, Prateek Jain, Praneeth Netrapalli

CoRR, 2017

Accelerating Stochastic Gradient Descent.

Prateek Jain, Sham M. Kakade, Rahul Kidambi, Praneeth Netrapalli, Aaron Sidford

CoRR, 2017

Learning Mixture of Gaussians with Streaming Data.

Aditi Raghunathan, Prateek Jain, Ravishankar Krishnaswamy

Proceedings of the Advances in Neural Information Processing Systems 30: Annual Conference on Neural Information Processing Systems 2017, 2017

Consistent Robust Regression.

Kush Bhatia, Prateek Jain, Parameswaran Kamalaruban, Purushottam Kar

Proceedings of the Advances in Neural Information Processing Systems 30: Annual Conference on Neural Information Processing Systems 2017, 2017

Fast second-order cone programming for safe mission planning.

Kai Zhong, Prateek Jain, Ashish Kapoor

Proceedings of the 2017 IEEE International Conference on Robotics and Automation, 2017

Recovery Guarantees for One-hidden-layer Neural Networks.

Kai Zhong, Zhao Song, Prateek Jain, Peter L. Bartlett, Inderjit S. Dhillon

Proceedings of the 34th International Conference on Machine Learning, 2017

ProtoNN: Compressed and Accurate kNN for Resource-scarce Devices.

Chirag Gupta, Arun Sai Suggala, Ankit Goyal, Harsha Vardhan Simhadri, Bhargavi Paranjape, Ashish Kumar, Saurabh Goyal, Raghavendra Udupa, Manik Varma, Prateek Jain

Proceedings of the 34th International Conference on Machine Learning, 2017

Nearly Optimal Robust Matrix Completion.

Yeshwanth Cherapanamjeri, Kartik Gupta, Prateek Jain

Proceedings of the 34th International Conference on Machine Learning, 2017

Active Heteroscedastic Regression.

Kamalika Chaudhuri, Prateek Jain, Nagarajan Natarajan

Proceedings of the 34th International Conference on Machine Learning, 2017

A Markov Chain Theory Approach to Characterizing the Minimax Optimality of Stochastic Gradient Descent (for Least Squares).

Prateek Jain, Sham M. Kakade, Rahul Kidambi, Praneeth Netrapalli, Venkata Krishna Pillutla, Aaron Sidford

Proceedings of the 37th IARCS Annual Conference on Foundations of Software Technology and Theoretical Computer Science, 2017

Thresholding Based Outlier Robust PCA.

Yeshwanth Cherapanamjeri, Prateek Jain, Praneeth Netrapalli

Proceedings of the 30th Conference on Learning Theory, 2017

Programming by Examples: PL Meets ML.

Sumit Gulwani, Prateek Jain

Proceedings of the Programming Languages and Systems - 15th Asian Symposium, 2017

Global Convergence of Non-Convex Gradient Descent for Computing Matrix Squareroot.

Prateek Jain, Chi Jin, Sham M. Kakade, Praneeth Netrapalli

Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, 2017

Scalable Optimization of Multivariate Performance Measures in Multi-instance Multi-label Learning.

Apoorv Aggarwal, Sandip Ghoshal, Ankith M. S. Shetty, Suhit Sinha, Ganesh Ramakrishnan, Purushottam Kar, Prateek Jain

Proceedings of the Thirty-First AAAI Conference on Artificial Intelligence, 2017

2016

Learning Sparsely Used Overcomplete Dictionaries via Alternating Minimization.

Alekh Agarwal, Animashree Anandkumar, Prateek Jain, Praneeth Netrapalli

SIAM J. Optim., 2016

Efficient and Consistent Robust Time Series Analysis.

Kush Bhatia, Prateek Jain, Parameswaran Kamalaruban, Purushottam Kar

CoRR, 2016

Parallelizing Stochastic Approximation Through Mini-Batching and Tail-Averaging.

Prateek Jain, Sham M. Kakade, Rahul Kidambi, Praneeth Netrapalli, Aaron Sidford

CoRR, 2016

Matching Matrix Bernstein with Little Memory: Near-Optimal Finite Sample Guarantees for Oja's Algorithm.

Prateek Jain, Chi Jin, Sham M. Kakade, Praneeth Netrapalli, Aaron Sidford

CoRR, 2016

Mixed Linear Regression with Multiple Components.

Kai Zhong, Prateek Jain, Inderjit S. Dhillon

Proceedings of the Advances in Neural Information Processing Systems 29: Annual Conference on Neural Information Processing Systems 2016, 2016

Selective inference for group-sparse linear models.

Fan Yang, Rina Foygel Barber, Prateek Jain, John D. Lafferty

Proceedings of the Advances in Neural Information Processing Systems 29: Annual Conference on Neural Information Processing Systems 2016, 2016

Regret Bounds for Non-decomposable Metrics with Missing Labels.

Nagarajan Natarajan, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 29: Annual Conference on Neural Information Processing Systems 2016, 2016

Structured Sparse Regression via Greedy Hard Thresholding.

Prateek Jain, Nikhil Rao, Inderjit S. Dhillon

Proceedings of the Advances in Neural Information Processing Systems 29: Annual Conference on Neural Information Processing Systems 2016, 2016

Diverse Yet Efficient Retrieval using Locality Sensitive Hashing.

Vidyadhar Rao, Prateek Jain, C. V. Jawahar

Proceedings of the 2016 ACM on International Conference on Multimedia Retrieval, 2016

Streaming PCA: Matching Matrix Bernstein and Near-Optimal Finite Sample Guarantees for Oja's Algorithm.

Prateek Jain, Chi Jin, Sham M. Kakade, Praneeth Netrapalli, Aaron Sidford

Proceedings of the 29th Conference on Learning Theory, 2016

Tensor vs. Matrix Methods: Robust Tensor Decomposition under Block Sparse Perturbations.

Anima Anandkumar, Prateek Jain, Yang Shi, U. N. Niranjan

Proceedings of the 19th International Conference on Artificial Intelligence and Statistics, 2016

2015

Phase Retrieval Using Alternating Minimization.

Praneeth Netrapalli, Prateek Jain, Sujay Sanghavi

IEEE Trans. Signal Process., 2015

Diverse Yet Efficient Retrieval using Hash Functions.

Vidyadhar Rao, Prateek Jain, C. V. Jawahar

CoRR, 2015

Locally Non-linear Embeddings for Extreme Multi-label Learning.

Kush Bhatia, Himanshu Jain, Purushottam Kar, Prateek Jain, Manik Varma

CoRR, 2015

To Drop or Not to Drop: Robustness, Consistency and Differential Privacy Properties of Dropout.

Prateek Jain, Vivek Kulkarni, Abhradeep Thakurta, Oliver Williams

CoRR, 2015

Computing Matrix Squareroot via Non Convex Local Search.

Prateek Jain, Chi Jin, Sham M. Kakade, Praneeth Netrapalli

CoRR, 2015

Tighter Low-rank Approximation via Sampling the Leveraged Element.

Srinadh Bhojanapalli, Prateek Jain, Sujay Sanghavi

Proceedings of the Twenty-Sixth Annual ACM-SIAM Symposium on Discrete Algorithms, 2015

Alternating Minimization for Regression Problems with Vector-valued Outputs.

Prateek Jain, Ambuj Tewari

Proceedings of the Advances in Neural Information Processing Systems 28: Annual Conference on Neural Information Processing Systems 2015, 2015

Predtron: A Family of Online Algorithms for General Prediction Problems.

Prateek Jain, Nagarajan Natarajan, Ambuj Tewari

Proceedings of the Advances in Neural Information Processing Systems 28: Annual Conference on Neural Information Processing Systems 2015, 2015

Sparse Local Embeddings for Extreme Multi-label Classification.

Kush Bhatia, Himanshu Jain, Purushottam Kar, Manik Varma, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 28: Annual Conference on Neural Information Processing Systems 2015, 2015

Robust Regression via Hard Thresholding.

Kush Bhatia, Prateek Jain, Purushottam Kar

Proceedings of the Advances in Neural Information Processing Systems 28: Annual Conference on Neural Information Processing Systems 2015, 2015

Optimizing Non-decomposable Performance Measures: A Tale of Two Classes.

Harikrishna Narasimhan, Purushottam Kar, Prateek Jain

Proceedings of the 32nd International Conference on Machine Learning, 2015

Surrogate Functions for Maximizing Precision at the Top.

Purushottam Kar, Harikrishna Narasimhan, Prateek Jain

Proceedings of the 32nd International Conference on Machine Learning, 2015

Fast Exact Matrix Completion with Finite Samples.

Prateek Jain, Praneeth Netrapalli

Proceedings of The 28th Conference on Learning Theory, 2015

Efficient Matrix Sensing Using Rank-1 Gaussian Measurements.

Kai Zhong, Prateek Jain, Inderjit S. Dhillon

Proceedings of the Algorithmic Learning Theory - 26th International Conference, 2015

2014

Hashing Hyperplane Queries to Near Points with Applications to Large-Scale Active Learning.

Sudheendra Vijayanarasimhan, Prateek Jain, Kristen Grauman

IEEE Trans. Pattern Anal. Mach. Intell., 2014

Non-convex Robust PCA.

Praneeth Netrapalli, U. N. Niranjan, Sujay Sanghavi, Animashree Anandkumar, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 27: Annual Conference on Neural Information Processing Systems 2014, 2014

Online and Stochastic Gradient Methods for Non-decomposable Loss Functions.

Purushottam Kar, Harikrishna Narasimhan, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 27: Annual Conference on Neural Information Processing Systems 2014, 2014

Provable Submodular Minimization using Wolfe's Algorithm.

Deeparnab Chakrabarty, Prateek Jain, Pravesh Kothari

Proceedings of the Advances in Neural Information Processing Systems 27: Annual Conference on Neural Information Processing Systems 2014, 2014

On Iterative Hard Thresholding Methods for High-dimensional M-Estimation.

Prateek Jain, Ambuj Tewari, Purushottam Kar

Proceedings of the Advances in Neural Information Processing Systems 27: Annual Conference on Neural Information Processing Systems 2014, 2014

Provable Tensor Factorization with Missing Data.

Prateek Jain, Sewoong Oh

Proceedings of the Advances in Neural Information Processing Systems 27: Annual Conference on Neural Information Processing Systems 2014, 2014

Large-scale Multi-label Learning with Missing Labels.

Hsiang-Fu Yu, Prateek Jain, Purushottam Kar, Inderjit S. Dhillon

Proceedings of the 31th International Conference on Machine Learning, 2014

Universal Matrix Completion.

Srinadh Bhojanapalli, Prateek Jain

Proceedings of the 31th International Conference on Machine Learning, 2014

(Near) Dimension Independent Risk Bounds for Differentially Private Learning.

Prateek Jain, Abhradeep Guha Thakurta

Proceedings of the 31th International Conference on Machine Learning, 2014

Learning Sparsely Used Overcomplete Dictionaries.

Alekh Agarwal, Animashree Anandkumar, Prateek Jain, Praneeth Netrapalli, Rashish Tandon

Proceedings of The 27th Conference on Learning Theory, 2014

Learning Mixtures of Discrete Product Distributions using Spectral Decompositions.

Prateek Jain, Sewoong Oh

Proceedings of The 27th Conference on Learning Theory, 2014

2013

Improved Multiple Sequence Alignments Using Coupled Pattern Mining.

K. S. M. Tozammel Hossain, Debprakash Patnaik, Srivatsan Laxman, Prateek Jain, Chris Bailey-Kellogg, Naren Ramakrishnan

IEEE ACM Trans. Comput. Biol. Bioinform., 2013

Large-scale Multi-label Learning with Missing Labels.

Hsiang-Fu Yu, Prateek Jain, Inderjit S. Dhillon

CoRR, 2013

Learning Sparsely Used Overcomplete Dictionaries via Alternating Minimization.

Alekh Agarwal, Animashree Anandkumar, Prateek Jain, Praneeth Netrapalli, Rashish Tandon

CoRR, 2013

Provable Inductive Matrix Completion.

Prateek Jain, Inderjit S. Dhillon

CoRR, 2013

Ad impression forecasting for sponsored search.

Abhirup Nath, Shibnath Mukherjee, Prateek Jain, Navin Goyal, Srivatsan Laxman

Proceedings of the 22nd International World Wide Web Conference, 2013

Low-rank matrix completion using alternating minimization.

Prateek Jain, Praneeth Netrapalli, Sujay Sanghavi

Proceedings of the Symposium on Theory of Computing Conference, 2013

Memory Limited, Streaming PCA.

Ioannis Mitliagkas, Constantine Caramanis, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 26: 27th Annual Conference on Neural Information Processing Systems 2013. Proceedings of a meeting held December 5-8, 2013

On the Generalization Ability of Online Learning Algorithms for Pairwise Loss Functions.

Purushottam Kar, Bharath K. Sriperumbudur, Prateek Jain, Harish Karnick

Proceedings of the 30th International Conference on Machine Learning, 2013

One-Bit Compressed Sensing: Provable Support and Vector Recovery.

Sivakanth Gopi, Praneeth Netrapalli, Prateek Jain, Aditya V. Nori

Proceedings of the 30th International Conference on Machine Learning, 2013

Differentially Private Learning with Kernels.

Prateek Jain, Abhradeep Thakurta

Proceedings of the 30th International Conference on Machine Learning, 2013

2012

Differentially Private Online Learning.

Prateek Jain, Pravesh Kothari, Abhradeep Thakurta

Proceedings of the COLT 2012, 2012

Metric and Kernel Learning Using a Linear Transformation.

Prateek Jain, Brian Kulis, Jason V. Davis, Inderjit S. Dhillon

J. Mach. Learn. Res., 2012

The Interplay Between Stability and Regret in Online Learning

Ankan Saha, Prateek Jain, Ambuj Tewari

CoRR, 2012

Supervised Learning with Similarity Functions.

Purushottam Kar, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 25: 26th Annual Conference on Neural Information Processing Systems 2012. Proceedings of a meeting held December 3-6, 2012

Multilabel Classification using Bayesian Compressed Sensing.

Ashish Kapoor, Raajay Viswanathan, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 25: 26th Annual Conference on Neural Information Processing Systems 2012. Proceedings of a meeting held December 3-6, 2012

Mirror Descent Based Database Privacy.

Prateek Jain, Abhradeep Thakurta

Proceedings of the Approximation, Randomization, and Combinatorial Optimization. Algorithms and Techniques, 2012

2011

A Learning Framework for Self-Tuning Histograms

Raajay Viswanathan, Prateek Jain, Srivatsan Laxman, Arvind Arasu

CoRR, 2011

Similarity-based Learning via Data Driven Embeddings.

Purushottam Kar, Prateek Jain

Proceedings of the Advances in Neural Information Processing Systems 24: 25th Annual Conference on Neural Information Processing Systems 2011. Proceedings of a meeting held 12-14 December 2011, 2011

Orthogonal Matching Pursuit with Replacement.

Prateek Jain, Ambuj Tewari, Inderjit S. Dhillon

Proceedings of the Advances in Neural Information Processing Systems 24: 25th Annual Conference on Neural Information Processing Systems 2011. Proceedings of a meeting held 12-14 December 2011, 2011

2010

Guaranteed Rank Minimization via Singular Value Projection.

Prateek Jain, Raghu Meka, Inderjit S. Dhillon

Proceedings of the Advances in Neural Information Processing Systems 23: 24th Annual Conference on Neural Information Processing Systems 2010. Proceedings of a meeting held 6-9 December 2010, 2010

Inductive Regularized Learning of Kernel Functions.

Prateek Jain, Brian Kulis, Inderjit S. Dhillon

Proceedings of the Advances in Neural Information Processing Systems 23: 24th Annual Conference on Neural Information Processing Systems 2010. Proceedings of a meeting held 6-9 December 2010, 2010

Far-sighted active learning on a budget for image and video recognition.

Sudheendra Vijayanarasimhan, Prateek Jain, Kristen Grauman

Proceedings of the Twenty-Third IEEE Conference on Computer Vision and Pattern Recognition, 2010

2009

Fast Similarity Search for Learned Metrics.

Brian Kulis, Prateek Jain, Kristen Grauman

IEEE Trans. Pattern Anal. Mach. Intell., 2009

Matrix Completion from Power-Law Distributed Samples.

Raghu Meka, Prateek Jain, Inderjit S. Dhillon

Proceedings of the Advances in Neural Information Processing Systems 22: 23rd Annual Conference on Neural Information Processing Systems 2009. Proceedings of a meeting held 7-10 December 2009, 2009

Geometry-aware metric learning.

Zhengdong Lu, Prateek Jain, Inderjit S. Dhillon

Proceedings of the 26th Annual International Conference on Machine Learning, 2009

Active learning for large multi-class problems.

Prateek Jain, Ashish Kapoor

Proceedings of the 2009 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR 2009), 2009

2008

Simultaneous Unsupervised Learning of Disparate Clusterings.

Prateek Jain, Raghu Meka, Inderjit S. Dhillon

Stat. Anal. Data Min., 2008

Online Metric Learning and Fast Similarity Search.

Prateek Jain, Brian Kulis, Inderjit S. Dhillon, Kristen Grauman

Proceedings of the Advances in Neural Information Processing Systems 21, 2008

Rank minimization via online learning.

Raghu Meka, Prateek Jain, Constantine Caramanis, Inderjit S. Dhillon

Proceedings of the Machine Learning, 2008

Fast image search for learned metrics.

Prateek Jain, Brian Kulis, Kristen Grauman

Proceedings of the 2008 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR 2008), 2008

2007

Information-theoretic metric learning.

Jason V. Davis, Brian Kulis, Prateek Jain, Suvrit Sra, Inderjit S. Dhillon

Proceedings of the Machine Learning, 2007

2005

Multi-objective Optimization for Adaptive Web Site Generation.

Prateek Jain, Pabitra Mitra

Proceedings of the Pattern Recognition and Machine Intelligence, 2005

2004

Anaphora Resolution in Multi-Person Dialogues.

Prateek Jain, Manav Ratan Mital, Sumit Kumar, Amitabha Mukerjee, Achla M. Raina

Proceedings of the SIGDIAL 2004 Workshop, The 5th Annual Meeting of the Special Interest Group on Discourse and Dialogue, April 30, 2004