Rai, Piyush
70 publications
TMLR
2026
Probing Layer-Wise Memorization and Generalization in Deep Neural Networks via Model Stitching
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