activity
20092023
most citedLarge-scale Multi-label Learning with Missing Labels

357 citations · 2.8k across the 94 of their papers we have counts for

collaborators
Showing 2021Show all

13 papers · 1 filter

cs.LG2021★ 3 cited

Node-Level Differentially Private Graph Neural Networks

Ameya Daigavane, Gagan Madan, Aditya Sinha +3

Graph Neural Networks (GNNs) are a popular technique for modelling graph-structured data and computing node-level representations via aggregation of information from the neighborho…

cs.LG2021

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

Naman Agarwal, Syomantak Chaudhuri, Prateek Jain +2

Q-learning is a popular Reinforcement Learning (RL) algorithm which is widely used in practice with function approximation (Mnih et al., 2015). In contrast, existing theoretical re…

cs.LG2021

Neural Network Compatible Off-Policy Natural Actor-Critic Algorithm

Raghuram Bharadwaj Diddigi, Prateek Jain, Prabuchandran K. J. +1

Learning optimal behavior from existing data is one of the most important problems in Reinforcement Learning (RL). This is known as "off-policy control" in RL where an agent's obje…

cs.LG2021

IGLU: Efficient GCN Training via Lazy Updates

S Deepak Narayanan, Aditya Sinha, Prateek Jain +2

Training multi-layer Graph Convolution Networks (GCN) using standard SGD techniques scales poorly as each descent step ends up updating node embeddings for a large portion of the g…

cs.LG2021★ 1 cited

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

Steve Chien, Prateek Jain, Walid Krichene +4

We study the problem of differentially private (DP) matrix completion under user-level privacy. We design a joint differentially private variant of the popular Alternating-Least-Sq…

cs.LG2021★ 7 cited

Robust Training in High Dimensions via Block Coordinate Geometric Median Descent

Anish Acharya, Abolfazl Hashemi, Prateek Jain +3

Geometric median (\textsc{Gm}) is a classical method in statistics for achieving a robust estimation of the uncorrupted data; under gross corruption, it achieves the optimal breakd…