Resiliency of automotive object detection networks on GPU architectures. - Atieh Lotfi; Saurabh Hukerikar; Keshav Balasubramanian; Paul Racunas; Nirmal R. Saxena; Richard Bramley; Yanxiang Huang (ITC, 2019)

Distributed Training of Graph Convolutional Networks using Subgraph Approximation. - Alexandra Angerd; Keshav Balasubramanian; Murali Annavaram (CoRR, 2020)

Origami Inference: Private Inference Using Hardware Enclaves. - Krishna Giri Narra; Zhifeng Lin; Yongqin Wang; Keshav Balasubramanian; Murali Annavaram (CLOUD, 2021)

Graph Traversal with Tensor Functionals: A Meta-Algorithm for Scalable Learning. - Elan Sopher Markowitz; Keshav Balasubramanian; Mehrnoosh Mirtaheri; Sami Abu-El-Haija; Bryan Perozzi; Greg Ver Steeg; Aram Galstyan (ICLR, 2021)

cDLRM: Look Ahead Caching for Scalable Training of Recommendation Models. - Keshav Balasubramanian; Abdulla Alshabanah; Joshua D. Choe; Murali Annavaram (RecSys, 2021)

FedGraphNN: A Federated Learning System and Benchmark for Graph Neural Networks. - Chaoyang He 001; Keshav Balasubramanian; Emir Ceyani; Yu Rong 001; Peilin Zhao; Junzhou Huang; Murali Annavaram; Salman Avestimehr (CoRR, 2021)

SpreadGNN: Serverless Multi-task Federated Learning for Graph Neural Networks. - Chaoyang He 001; Emir Ceyani; Keshav Balasubramanian; Murali Annavaram; Salman Avestimehr (CoRR, 2021)

SpreadGNN: Decentralized Multi-Task Federated Learning for Graph Neural Networks on Molecular Data. - Chaoyang He 001; Emir Ceyani; Keshav Balasubramanian; Murali Annavaram; Salman Avestimehr (AAAI, 2022)

StATIK: Structure and Text for Inductive Knowledge Graph Completion. - Elan Markowitz; Keshav Balasubramanian; Mehrnoosh Mirtaheri; Murali Annavaram; Aram Galstyan; Greg Ver Steeg (NAACL-HLT (Findings), 2022)

Biased User History Synthesis for Personalized Long-Tail Item Recommendation. - Keshav Balasubramanian; Abdulla Alshabanah; Elan Markowitz; Greg Ver Steeg; Murali Annavaram (RecSys, 2024)

Meta-Learn to Unlearn: Enhanced Exact Machine Unlearning in Recommendation Systems with Meta-Learning. - Abdulla Alshabanah; Keshav Balasubramanian; Murali Annavaram (Proc. Priv. Enhancing Technol., 2025)