Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Picture for Prateek Jain

Streaming Linear System Identification with Reverse Experience Replay


Mar 10, 2021
Prateek Jain, Suhas S Kowshik, Dheeraj Nagaraj, Praneeth Netrapalli


  Access Paper or Ask Questions

Do Input Gradients Highlight Discriminative Features?


Feb 25, 2021
Harshay Shah, Prateek Jain, Praneeth Netrapalli

* Code: https://github.com/harshays/inputgradients 

  Access Paper or Ask Questions

Learning Accurate Decision Trees with Bandit Feedback via Quantized Gradient Descent


Feb 15, 2021
Ajaykrishna Karthikeyan, Naman Jain, Nagarajan Natarajan, Prateek Jain


  Access Paper or Ask Questions

Optimal Regret Algorithm for Pseudo-1d Bandit Convex Optimization


Feb 15, 2021
Aadirupa Saha, Nagarajan Natarajan, Praneeth Netrapalli, Prateek Jain


  Access Paper or Ask Questions

Projection Efficient Subgradient Method and Optimal Nonsmooth Frank-Wolfe Method


Oct 05, 2020
Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong Oh


  Access Paper or Ask Questions

Programming by Rewards


Jul 14, 2020
Nagarajan Natarajan, Ajaykrishna Karthikeyan, Prateek Jain, Ivan Radicek, Sriram Rajamani, Sumit Gulwani, Johannes Gehrke


  Access Paper or Ask Questions

Globally-convergent Iteratively Reweighted Least Squares for Robust Regression Problems


Jun 25, 2020
Bhaskar Mukhoty, Govind Gopakumar, Prateek Jain, Purushottam Kar

* Proceedings of Machine Learning Research (PMLR) 89:313-322, 2019 
* 30 pages, 5 figures, appeared as a publication in the 22nd International Conference on Artificial Intelligence and Statistics (AISTATS), 2019 

  Access Paper or Ask Questions

Least Squares Regression with Markovian Data: Fundamental Limits and Algorithms


Jun 16, 2020
Guy Bresler, Prateek Jain, Dheeraj Nagaraj, Praneeth Netrapalli, Xian Wu


  Access Paper or Ask Questions

The Pitfalls of Simplicity Bias in Neural Networks


Jun 13, 2020
Harshay Shah, Kaustav Tamuly, Aditi Raghunathan, Prateek Jain, Praneeth Netrapalli


  Access Paper or Ask Questions

COVID-19: Strategies for Allocation of Test Kits


Apr 03, 2020
Arpita Biswas, Shruthi Bannur, Prateek Jain, Srujana Merugu


  Access Paper or Ask Questions

Soft Threshold Weight Reparameterization for Learnable Sparsity


Mar 11, 2020
Aditya Kusupati, Vivek Ramanujan, Raghav Somani, Mitchell Wortsman, Prateek Jain, Sham Kakade, Ali Farhadi

* 17 pages, 9 figures 

  Access Paper or Ask Questions

DROCC: Deep Robust One-Class Classification


Feb 28, 2020
Sachin Goyal, Aditi Raghunathan, Moksh Jain, Harsha Vardhan Simhadri, Prateek Jain


  Access Paper or Ask Questions

RNNPool: Efficient Non-linear Pooling for RAM Constrained Inference


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

* 16 pages, 8 figures 

  Access Paper or Ask Questions

Rich-Item Recommendations for Rich-Users via GCNN: Exploiting Dynamic and Static Side Information


Jan 28, 2020
Amar Budhiraja, Gaurush Hiranandani, Navya Yarrabelly, Ayush Choure, Oluwasanmi Koyejo, Prateek Jain

* The first two authors contributed equally. 14 pages, 2 figures and 5 tables 

  Access Paper or Ask Questions

OASIS: ILP-Guided Synthesis of Loop Invariants


Nov 26, 2019
Sahil Bhatia, Saswat Padhi, Nagarajan Natarajan, Rahul Sharma, Prateek Jain


  Access Paper or Ask Questions

Learning Functions over Sets via Permutation Adversarial Networks


Jul 12, 2019
Chirag Pabbaraju, Prateek Jain


  Access Paper or Ask Questions

Efficient Algorithms for Smooth Minimax Optimization


Jul 02, 2019
Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong Oh


  Access Paper or Ask Questions

Making the Last Iterate of SGD Information Theoretically Optimal


May 29, 2019
Prateek Jain, Dheeraj Nagaraj, Praneeth Netrapalli

* 3 figures, Accepted for presentation at COLT 2019 

  Access Paper or Ask Questions

Adaptive Hard Thresholding for Near-optimal Consistent Robust Regression


Mar 19, 2019
Arun Sai Suggala, Kush Bhatia, Pradeep Ravikumar, Prateek Jain


  Access Paper or Ask Questions

SGD without Replacement: Sharper Rates for General Smooth Convex Functions


Mar 04, 2019
Prateek Jain, Dheeraj Nagaraj, Praneeth Netrapalli


  Access Paper or Ask Questions

FastGRNN: A Fast, Accurate, Stable and Tiny Kilobyte Sized Gated Recurrent Neural Network


Jan 08, 2019
Aditya Kusupati, Manish Singh, Kush Bhatia, Ashish Kumar, Prateek Jain, Manik Varma

* 23 pages, 10 figures, Published at Advances in Neural Information Processing Systems (NeurIPS) 2018 

  Access Paper or Ask Questions

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


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

* Published in ICLR 2018, International Conference on Learning Representations (2018) 

  Access Paper or Ask Questions

On the insufficiency of existing momentum schemes for Stochastic Optimization


Jul 31, 2018
Rahul Kidambi, Praneeth Netrapalli, Prateek Jain, Sham M. Kakade

* 28 pages, 10 figures. Updated acknowledgements. Appeared as an oral presentation at International Conference on Learning Representations (ICLR), 2018. Code implementing the ASGD method can be found at https://github.com/rahulkidambi/AccSGD 

  Access Paper or Ask Questions

Accelerating Stochastic Gradient Descent For Least Squares Regression


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

* 54 pages, 3 figures, 1 table; updated acknowledgements, minor title change. Paper appeared in the proceedings of the Conference on Learning Theory (COLT), 2018 

  Access Paper or Ask Questions

Parallelizing Stochastic Gradient Descent for Least Squares Regression: mini-batching, averaging, and model misspecification


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

* 39 pages. Published in the Journal of Machine Learning Research (JMLR) 

  Access Paper or Ask Questions

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


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

* Lemma 1 has been updated in v2 

  Access Paper or Ask Questions