Shakkottai, Sanjay

45 publications

NeurIPS 2025 Anchored Diffusion Language Model Litu Rout, Constantine Caramanis, Sanjay Shakkottai

NeurIPS 2025 Constrained Posterior Sampling: Time Series Generation with Hard Constraints Sai Shankar Narasimhan, Shubhankar Agarwal, Litu Rout, Sanjay Shakkottai, Sandeep P. Chinchali

ICLR 2025 Infilling Score: A Pretraining Data Detection Algorithm for Large Language Models Negin Raoof, Litu Rout, Giannis Daras, Sujay Sanghavi, Constantine Caramanis, Sanjay Shakkottai, Alex Dimakis

NeurIPS 2025 Machine Unlearning Under Overparameterization Jacob L. Block, Aryan Mokhtari, Sanjay Shakkottai

NeurIPS 2025 Provable Meta-Learning with Low-Rank Adaptations Jacob L. Block, Sundararajan Srinivasan, Liam Collins, Aryan Mokhtari, Sanjay Shakkottai

ICLR 2025 RB-Modulation: Training-Free Stylization Using Reference-Based Modulation Litu Rout, Yujia Chen, Nataniel Ruiz, Abhishek Kumar, Constantine Caramanis, Sanjay Shakkottai, Wen-Sheng Chu

ICLR 2025 Semantic Image Inversion and Editing Using Rectified Stochastic Differential Equations Litu Rout, Yujia Chen, Nataniel Ruiz, Constantine Caramanis, Sanjay Shakkottai, Wen-Sheng Chu

TMLR 2024 Bandits with Mean Bounds Nihal Sharma, Soumya Basu, Karthikeyan Shanmugam, Sanjay Shakkottai

CVPR 2024 Beyond First-Order Tweedie: Solving Inverse Problems Using Latent Diffusion Litu Rout, Yujia Chen, Abhishek Kumar, Constantine Caramanis, Sanjay Shakkottai, Wen-Sheng Chu

NeurIPS 2024 In-Context Learning with Transformers: SoftMax Attention Adapts to Function Lipschitzness Liam Collins, Advait Parulekar, Aryan Mokhtari, Sujay Sanghavi, Sanjay Shakkottai

ICML 2024 Provable Multi-Task Representation Learning by Two-Layer ReLU Neural Networks Liam Collins, Hamed Hassani, Mahdi Soltanolkotabi, Aryan Mokhtari, Sanjay Shakkottai

COLT 2023 Beyond Uniform Smoothness: A Stopped Analysis of Adaptive SGD Matthew Faw, Litu Rout, Constantine Caramanis, Sanjay Shakkottai

ICML 2023 Collaborative Multi-Agent Heterogeneous Multi-Armed Bandits Ronshee Chawla, Daniel Vial, Sanjay Shakkottai, R. Srikant

COLT 2023 InfoNCE Loss Provably Learns Cluster-Preserving Representations Advait Parulekar, Liam Collins, Karthikeyan Shanmugam, Aryan Mokhtari, Sanjay Shakkottai

ICML 2023 PAC Generalization via Invariant Representations Advait U Parulekar, Karthikeyan Shanmugam, Sanjay Shakkottai

NeurIPS 2023 Solving Linear Inverse Problems Provably via Posterior Sampling with Latent Diffusion Models Litu Rout, Negin Raoof, Giannis Daras, Constantine Caramanis, Alex Dimakis, Sanjay Shakkottai

AISTATS 2022 Improved Algorithms for Misspecified Linear Markov Decision Processes Daniel Vial, Advait Parulekar, Sanjay Shakkottai, R Srikant

ICML 2022 Asymptotically-Optimal Gaussian Bandits with Side Observations Alexia Atsidakou, Orestis Papadigenopoulos, Constantine Caramanis, Sujay Sanghavi, Sanjay Shakkottai

NeurIPS 2022 FedAvg with Fine Tuning: Local Updates Lead to Representation Learning Liam Collins, Hamed Hassani, Aryan Mokhtari, Sanjay Shakkottai

CoLLAs 2022 How Does the Task Landscape Affect MAML Performance? Liam Collins, Aryan Mokhtari, Sanjay Shakkottai

ICML 2022 Linear Bandit Algorithms with Sublinear Time Complexity Shuo Yang, Tongzheng Ren, Sanjay Shakkottai, Eric Price, Inderjit S. Dhillon, Sujay Sanghavi

ICML 2022 MAML and ANIL Provably Learn Representations Liam Collins, Aryan Mokhtari, Sewoong Oh, Sanjay Shakkottai

NeurIPS 2022 Minimax Regret for Cascading Bandits Daniel Vial, Sujay Sanghavi, Sanjay Shakkottai, R. Srikant

NeurIPS 2022 Non-Stationary Bandits Under Recharging Payoffs: Improved Planning with Sublinear Regret Orestis Papadigenopoulos, Constantine Caramanis, Sanjay Shakkottai

ICML 2022 Regret Bounds for Stochastic Shortest Path Problems with Linear Function Approximation Daniel Vial, Advait Parulekar, Sanjay Shakkottai, R Srikant

COLT 2022 The Power of Adaptivity in SGD: Self-Tuning Step Sizes with Unbounded Gradients and Affine Variance Matthew Faw, Isidoros Tziotis, Constantine Caramanis, Aryan Mokhtari, Sanjay Shakkottai, Rachel Ward

AISTATS 2021 Contextual Blocking Bandits Soumya Basu, Orestis Papadigenopoulos, Constantine Caramanis, Sanjay Shakkottai

ICML 2021 Combinatorial Blocking Bandits with Stochastic Delays Alexia Atsidakou, Orestis Papadigenopoulos, Soumya Basu, Constantine Caramanis, Sanjay Shakkottai

ICML 2021 Exploiting Shared Representations for Personalized Federated Learning Liam Collins, Hamed Hassani, Aryan Mokhtari, Sanjay Shakkottai

NeurIPS 2021 Finite-Sample Analysis of Off-Policy TD-Learning via Generalized Bellman Operators Zaiwei Chen, Siva Theja Maguluri, Sanjay Shakkottai, Karthikeyan Shanmugam

NeurIPS 2020 Applications of Common Entropy for Causal Inference Murat Kocaoglu, Sanjay Shakkottai, Alexandros G Dimakis, Constantine Caramanis, Sriram Vishwanath

NeurIPS 2020 Finite-Sample Analysis of Contractive Stochastic Approximation Using Smooth Convex Envelopes Zaiwei Chen, Siva Theja Maguluri, Sanjay Shakkottai, Karthikeyan Shanmugam

NeurIPS 2020 Mix and Match: An Optimistic Tree-Search Approach for Learning Models from Mixture Distributions Matthew Faw, Rajat Sen, Karthikeyan Shanmugam, Constantine Caramanis, Sanjay Shakkottai

NeurIPS 2020 Task-Robust Model-Agnostic Meta-Learning Liam Collins, Aryan Mokhtari, Sanjay Shakkottai

AISTATS 2020 The Gossiping Insert-Eliminate Algorithm for Multi-Agent Bandits Ronshee Chawla, Abishek Sankararaman, Ayalvadi Ganesh, Sanjay Shakkottai

NeurIPS 2019 Blocking Bandits Soumya Basu, Rajat Sen, Sujay Sanghavi, Sanjay Shakkottai

ECML-PKDD 2019 Importance Weighted Generative Networks Maurice Diesendruck, Ethan R. Elenberg, Rajat Sen, Guy W. Cole, Sanjay Shakkottai, Sinead A. Williamson

AISTATS 2019 Noisy Blackbox Optimization Using Multi-Fidelity Queries: A Tree Search Approach Rajat Sen, Kirthevasan Kandasamy, Sanjay Shakkottai

ICML 2019 Pareto Optimal Streaming Unsupervised Classification Soumya Basu, Steven Gutstein, Brent Lance, Sanjay Shakkottai

AISTATS 2018 Contextual Bandits with Stochastic Experts Rajat Sen, Karthikeyan Shanmugam, Sanjay Shakkottai

ICML 2018 Multi-Fidelity Black-Box Optimization with Hierarchical Partitions Rajat Sen, Kirthevasan Kandasamy, Sanjay Shakkottai

AISTATS 2017 Contextual Bandits with Latent Confounders: An NMF Approach Rajat Sen, Karthikeyan Shanmugam, Murat Kocaoglu, Alexandros G. Dimakis, Sanjay Shakkottai

ICML 2017 Identifying Best Interventions Through Online Importance Sampling Rajat Sen, Karthikeyan Shanmugam, Alexandros G. Dimakis, Sanjay Shakkottai

NeurIPS 2017 Model-Powered Conditional Independence Test Rajat Sen, Ananda Theertha Suresh, Karthikeyan Shanmugam, Alexandros G Dimakis, Sanjay Shakkottai

NeurIPS 2016 Regret of Queueing Bandits Subhashini Krishnasamy, Rajat Sen, Ramesh Johari, Sanjay Shakkottai