Publications (141)
FlashProfile: A Framework for Synthesizing Data Profiles
Saswat Padhi, Prateek Jain, Daniel Perelman +3
Non-convex Optimization for Machine Learning
Prateek Jain, Purushottam Kar
Smoothed analysis for low-rank solutions to semidefinite programs in quadratic penalty form
Srinadh Bhojanapalli, Nicolas Boumal, Prateek Jain +1
Provable Tensor Factorization with Missing Data
Prateek Jain, Sewoong Oh
End-to-End Neural Network Compression via Regularized Latency Surrogates
Anshul Nasery, Hardik Shah, Arun Sai Suggala +1
Optimizing Non-decomposable Performance Measures: A Tale of Two Classes
Harikrishna Narasimhan, Purushottam Kar, Prateek Jain
Phase Retrieval using Alternating Minimization
Praneeth Netrapalli, Prateek Jain, Sujay Sanghavi
Programming by Rewards
Nagarajan Natarajan, Ajaykrishna Karthikeyan, Prateek Jain +4
Multi-Task Differential Privacy Under Distribution Skew
Walid Krichene, Prateek Jain, Shuang Song +3
Simplicity Bias in 1-Hidden Layer Neural Networks
Depen Morwani, Jatin Batra, Prateek Jain +1
LLM Augmented LLMs: Expanding Capabilities through Composition
Rachit Bansal, Bidisha Samanta, Siddharth Dalmia +6
MatFormer: Nested Transformer for Elastic Inference
Devvrit, Sneha Kudugunta, Aditya Kusupati +8
Parallelizing Stochastic Gradient Descent for Least Squares Regression: mini-batching, averaging, and model misspecification
Prateek Jain, Sham M. Kakade, Rahul Kidambi +2
Differentially Private Online Learning
Prateek Jain, Pravesh Kothari, Abhradeep Thakurta
Online and Stochastic Gradient Methods for Non-decomposable Loss Functions
Purushottam Kar, Harikrishna Narasimhan, Prateek Jain
On Iterative Hard Thresholding Methods for High-dimensional M-Estimation
Prateek Jain, Ambuj Tewari, Purushottam Kar
Optimal Regret Algorithm for Pseudo-1d Bandit Convex Optimization
Aadirupa Saha, Nagarajan Natarajan, Praneeth Netrapalli +1
Projection Efficient Subgradient Method and Optimal Nonsmooth Frank-Wolfe Method
Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli +1
XTile: An Error-Correction Package for DNA Self-Assembly
Anshul Chaurasia, Sudhanshu Dwivedi, Prateek Jain +1
One-Pass to Reason: Token Duplication and Block-Sparse Mask for Efficient Fine-Tuning on Multi-Turn Reasoning
Ritesh Goru, Shanay Mehta, Prateek Jain
RNNPool: Efficient Non-linear Pooling for RAM Constrained Inference
Oindrila Saha, Aditya Kusupati, Harsha Vardhan Simhadri +2
Locally Non-linear Embeddings for Extreme Multi-label Learning
Kush Bhatia, Himanshu Jain, Purushottam Kar +2
Optimal Algorithms for Latent Bandits with Cluster Structure
Soumyabrata Pal, Arun Sai Suggala, Karthikeyan Shanmugam +1
On the insufficiency of existing momentum schemes for Stochastic Optimization
Rahul Kidambi, Praneeth Netrapalli, Prateek Jain +1
Regret Bounds for Non-decomposable Metrics with Missing Labels
Prateek Jain, Nagarajan Natarajan
Fast Exact Matrix Completion with Finite Samples
Prateek Jain, Praneeth Netrapalli
Non-convex Robust PCA
Praneeth Netrapalli, U N Niranjan, Sujay Sanghavi +2
MET: Masked Encoding for Tabular Data
Kushal Majmundar, Sachin Goyal, Praneeth Netrapalli +1
ELT: Elastic Looped Transformers for Visual Generation
Sahil Goyal, Swayam Agrawal, Gautham Govind Anil +3
Online Target Q-learning with Reverse Experience Replay: Efficiently finding the Optimal Policy for Linear MDPs
Naman Agarwal, Syomantak Chaudhuri, Prateek Jain +2
iGLU 1.0: An Accurate Non-Invasive Near-Infrared Dual Short Wavelengths Spectroscopy based Glucometer for Smart Healthcare
Prateek Jain, Amit M. Joshi, Saraju P. Mohanty
SGD without Replacement: Sharper Rates for General Smooth Convex Functions
Prateek Jain, Dheeraj Nagaraj, Praneeth Netrapalli
Efficient and Consistent Robust Time Series Analysis
Kush Bhatia, Prateek Jain, Parameswaran Kamalaruban +1
DP-PCA: Statistically Optimal and Differentially Private PCA
Xiyang Liu, Weihao Kong, Prateek Jain +1
Nonlinear Inductive Matrix Completion based on One-layer Neural Networks
Kai Zhong, Zhao Song, Prateek Jain +1
Fast Second-order Cone Programming for Safe Mission Planning
Kai Zhong, Prateek Jain, Ashish Kapoor
Globally-convergent Iteratively Reweighted Least Squares for Robust Regression Problems
Bhaskar Mukhoty, Govind Gopakumar, Prateek Jain +1
EHI: End-to-end Learning of Hierarchical Index for Efficient Dense Retrieval
Ramnath Kumar, Anshul Mittal, Nilesh Gupta +3
Supervised Learning with Similarity Functions
Purushottam Kar, Prateek Jain
Accelerating Stochastic Gradient Descent For Least Squares Regression
Prateek Jain, Sham M. Kakade, Rahul Kidambi +2
Streaming PCA: Matching Matrix Bernstein and Near-Optimal Finite Sample Guarantees for Oja's Algorithm
Prateek Jain, Chi Jin, Sham M. Kakade +2
Mixture of Nested Experts: Adaptive Processing of Visual Tokens
Gagan Jain, Nidhi Hegde, Aditya Kusupati +5
Recovery Guarantees for One-hidden-layer Neural Networks
Kai Zhong, Zhao Song, Prateek Jain +2
Universal Matrix Completion
Srinadh Bhojanapalli, Prateek Jain
DROCC: Deep Robust One-Class Classification
Sachin Goyal, Aditi Raghunathan, Moksh Jain +2
Thresholding based Efficient Outlier Robust PCA
Yeshwanth Cherapanamjeri, Prateek Jain, Praneeth Netrapalli
On Scaling Data-Driven Loop Invariant Inference
Sahil Bhatia, Saswat Padhi, Nagarajan Natarajan +2
Multi-User Reinforcement Learning with Low Rank Rewards
Naman Agarwal, Prateek Jain, Suhas Kowshik +2
Leveraging Distributional Semantics for Multi-Label Learning
Rahul Wadbude, Vivek Gupta, Piyush Rai +3
Tensor vs Matrix Methods: Robust Tensor Decomposition under Block Sparse Perturbations
Animashree Anandkumar, Prateek Jain, Yang Shi +1
Learning an Invertible Output Mapping Can Mitigate Simplicity Bias in Neural Networks
Sravanti Addepalli, Anshul Nasery, R. Venkatesh Babu +2
Interleaved Gibbs Diffusion: Generating Discrete-Continuous Data with Implicit Constraints
Gautham Govind Anil, Sachin Yadav, Dheeraj Nagaraj +2
Dual-Encoders for Extreme Multi-Label Classification
Nilesh Gupta, Devvrit Khatri, Ankit S Rawat +3
Neural Network Compatible Off-Policy Natural Actor-Critic Algorithm
Raghuram Bharadwaj Diddigi, Prateek Jain, Prabuchandran K. J. +1
iGLU 2.0: A new non-invasive, accurate serum glucometer for smart healthcare
Prateek Jain, Amit M Joshi, Navneet Agrawal +1
Private Alternating Least Squares: Practical Private Matrix Completion with Tighter Rates
Steve Chien, Prateek Jain, Walid Krichene +4
Tighter Low-rank Approximation via Sampling the Leveraged Element
Srinadh Bhojanapalli, Prateek Jain, Sujay Sanghavi
(Nearly) Optimal Private Linear Regression via Adaptive Clipping
Prateek Varshney, Abhradeep Thakurta, Prateek Jain
Sample Efficient Linear Meta-Learning by Alternating Minimization
Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli +1
Succeeding at Scale: Enterprise Retrieval Benchmark Construction and Index-Preserving Query Adaptation for Multi-Tenant Search
Prateek Jain, Shabari S Nair, Ritesh Goru +4
Selective Inference for Group-Sparse Linear Models
Fan Yang, Rina Foygel Barber, Prateek Jain +1
Compressing Many-Shots in In-Context Learning
Devvrit Khatri, Pranamya Kulkarni, Nilesh Gupta +9
Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilities
Gheorghe Comanici, Eric Bieber, Mike Schaekermann +3431
Least Squares Regression with Markovian Data: Fundamental Limits and Algorithms
Guy Bresler, Prateek Jain, Dheeraj Nagaraj +2
Surrogate Functions for Maximizing Precision at the Top
Purushottam Kar, Harikrishna Narasimhan, Prateek Jain
Orthogonal Matching Pursuit with Replacement
Prateek Jain, Ambuj Tewari, Inderjit S. Dhillon
Nearly-optimal Robust Matrix Completion
Yeshwanth Cherapanamjeri, Kartik Gupta, Prateek Jain
Variational Quantum Algorithms for Chemical Simulation and Drug Discovery
Hasan Mustafa, Sai Nandan Morapakula, Prateek Jain +1
PennyLane: Automatic differentiation of hybrid quantum-classical computations
Ville Bergholm, Josh Izaac, Maria Schuld +65
IGLU: Efficient GCN Training via Lazy Updates
S Deepak Narayanan, Aditya Sinha, Prateek Jain +2
Neural-Guided Deductive Search for Real-Time Program Synthesis from Examples
Ashwin Kalyan, Abhishek Mohta, Oleksandr Polozov +3
To Drop or Not to Drop: Robustness, Consistency and Differential Privacy Properties of Dropout
Prateek Jain, Vivek Kulkarni, Abhradeep Thakurta +1
Memory Limited, Streaming PCA
Ioannis Mitliagkas, Constantine Caramanis, Prateek Jain
A Markov Chain Theory Approach to Characterizing the Minimax Optimality of Stochastic Gradient Descent (for Least Squares)
Prateek Jain, Sham M. Kakade, Rahul Kidambi +3
Provable Submodular Minimization using Wolfe's Algorithm
Deeparnab Chakrabarty, Prateek Jain, Pravesh Kothari
Rich-Item Recommendations for Rich-Users: Exploiting Dynamic and Static Side Information
Amar Budhiraja, Gaurush Hiranandani, Darshak Chhatbar +5
Node-Level Differentially Private Graph Neural Networks
Ameya Daigavane, Gagan Madan, Aditya Sinha +3
Estimating Causal Effects in Gaussian Linear SCMs with Finite Data
Aurghya Maiti, Prateek Jain
Does Safety Training of LLMs Generalize to Semantically Related Natural Prompts?
Sravanti Addepalli, Yerram Varun, Arun Suggala +2
A Faster Generalized Two-Stage Approximate Top-K
Yashas Samaga, Varun Yerram, Spandana Raj Babbula +2
Near Optimal Private and Robust Linear Regression
Xiyang Liu, Prateek Jain, Weihao Kong +2
Robust Training in High Dimensions via Block Coordinate Geometric Median Descent
Anish Acharya, Abolfazl Hashemi, Prateek Jain +3
Large-scale Multi-label Learning with Missing Labels
Hsiang-Fu Yu, Prateek Jain, Purushottam Kar +1
Robust Regression via Hard Thresholding
Kush Bhatia, Prateek Jain, Purushottam Kar
Everything You Wanted to Know About Noninvasive Glucose Measurement and Control
Prateek Jain, Amit M. Joshi, Saraju Mohanty
Structured Sparse Regression via Greedy Hard-Thresholding
Prateek Jain, Nikhil Rao, Inderjit Dhillon
Soft Threshold Weight Reparameterization for Learnable Sparsity
Aditya Kusupati, Vivek Ramanujan, Raghav Somani +4
Efficient Algorithms for Smooth Minimax Optimization
Kiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli +1
On the Generalization Ability of Online Learning Algorithms for Pairwise Loss Functions
Purushottam Kar, Bharath K Sriperumbudur, Prateek Jain +1
Guaranteed Rank Minimization via Singular Value Projection
Raghu Meka, Prateek Jain, Inderjit S. Dhillon
Near-optimal Offline and Streaming Algorithms for Learning Non-Linear Dynamical Systems
Prateek Jain, Suhas S Kowshik, Dheeraj Nagaraj +1
Do Input Gradients Highlight Discriminative Features?
Harshay Shah, Prateek Jain, Praneeth Netrapalli
The Interplay Between Stability and Regret in Online Learning
Ankan Saha, Prateek Jain, Ambuj Tewari
Gemini 1.5: Unlocking multimodal understanding across millions of tokens of context
Gemini Team, Petko Georgiev, Ving Ian Lei +1132
Gecko: Versatile Text Embeddings Distilled from Large Language Models
Jinhyuk Lee, Zhuyun Dai, Xiaoqi Ren +17
Towards Efficiently Solving Quantum Traveling Salesman Problem
Debabrata Goswami, Harish Karnick, Prateek Jain +1
Time-Reversal Provides Unsupervised Feedback to LLMs
Yerram Varun, Rahul Madhavan, Sravanti Addepalli +3
Learning Mixtures of Discrete Product Distributions using Spectral Decompositions
Prateek Jain, Sewoong Oh
The Pitfalls of Simplicity Bias in Neural Networks
Harshay Shah, Kaustav Tamuly, Aditi Raghunathan +2
iPAL: A Machine Learning Based Smart Healthcare Framework For Automatic Diagnosis Of Attention Deficit/Hyperactivity Disorder (ADHD)
Abhishek Sharma, Arpit Jain, Shubhangi Sharma +3