Rai, Piyush
67 publications
TMLR
2025
Federated Learning with Uncertainty and Personalization via Efficient Second-Order Optimization
NeurIPSW
2024
Simple and Scalable Federated Learning with Uncertainty via Improved Variational Online Newton
NeurIPSW
2024
Understanding Memorization Using Representation Similarity Analysis and Model Stitching
TMLR
2022
DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents