Alternate Model Growth and Pruning for Efficient Training of Recommendation Systems. - Xiaocong Du; Bhargav Bhushanam; Jiecao Yu; Dhruv Choudhary; Tianxiang Gao; Sherman Wong; Louis Feng; Jongsoo Park; Yu Cao 001; Arun Kejariwal (ICMLA, 2021)

Frequency-aware SGD for Efficient Embedding Learning with Provable Benefits. - Yan Li 074; Dhruv Choudhary; Xiaohan Wei; Baichuan Yuan; Bhargav Bhushanam; Tuo Zhao; Guanghui Lan (ICLR, 2022)

AutoShard: Automated Embedding Table Sharding for Recommender Systems. - Daochen Zha; Louis Feng; Bhargav Bhushanam; Dhruv Choudhary; Jade Nie; Yuandong Tian; Jay Chae; Yinbin Ma; Arun Kejariwal; Xia Hu 001 (KDD, 2022)

Harmless Transfer Learning for Item Embeddings. - Chengyue Gong; Xiaocong Du; Dhruv Choudhary; Bhargav Bhushanam; Qiang Liu 001; Arun Kejariwal (NAACL-HLT (Findings), 2022)

DreamShard: Generalizable Embedding Table Placement for Recommender Systems. - Daochen Zha; Louis Feng; Qiaoyu Tan; Zirui Liu 001; Kwei-Herng Lai; Bhargav Bhushanam; Yuandong Tian; Arun Kejariwal; Xia Hu 001 (NeurIPS, 2022)

Future gradient descent for adapting the temporal shifting data distribution in online recommendation systems. - Mao Ye 006; Ruichen Jiang; Haoxiang Wang 003; Dhruv Choudhary; Xiaocong Du; Bhargav Bhushanam; Aryan Mokhtari; Arun Kejariwal; Qiang Liu 001 (UAI, 2022)

Positive Unlabeled Contrastive Learning. - Anish Acharya; Sujay Sanghavi; Li Jing 001; Bhargav Bhushanam; Dhruv Choudhary; Michael G. Rabbat; Inderjit S. Dhillon (CoRR, 2022)

Understanding Scaling Laws for Recommendation Models. - Newsha Ardalani; Carole-Jean Wu; Zeliang Chen; Bhargav Bhushanam; Adnan Aziz (CoRR, 2022)

Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models. - Daochen Zha; Louis Feng; Liang Luo; Bhargav Bhushanam; Zirui Liu 001; Yusuo Hu; Jade Nie; Yuzhen Huang; Yuandong Tian; Arun Kejariwal; Xia Hu 001 (MLSys, 2023)

Layer Compression of Deep Networks with Straight Flows. - Chengyue Gong; Xiaocong Du; Bhargav Bhushanam; Lemeng Wu; Xingchao Liu; Dhruv Choudhary; Arun Kejariwal; Qiang Liu 001 (AAAI, 2024)

MQuAKE-Remastered: Multi-Hop Knowledge Editing Can Only Be Advanced with Reliable Evaluations. - Shaochen (Henry) Zhong; Yifan Lu; Lize Shao; Bhargav Bhushanam; Xiaocong Du; Yixin Wan; Yucheng Shi; Daochen Zha; Yiwei Wang; Ninghao Liu; Kaixiong Zhou; Shuai Xu; et al. (ICLR, 2025)