Aashish Kolluri

Distributed Machine Learning | Graph Learning | Security & Privacy

Bio

👋 Hi! I’m a PhD student at the National University of Singapore 🦁, navigating the crossroads of Distributed Machine Learning and Computer Security under the guidance of Prateek Saxena. My passion lies in designing efficient distributed machine learning systems and fortifying them against security threats, focusing on distributed graph learning.

🚀 Research Focus: My research focus is to build Efficient and Trustworthy Distributed Machine Learning (ML) Systems. Here’s a glimpse of three recent projects:

  • HiDRA: An untargeted model poisoning attack that destroys the model performance while training with strong robust aggregators. We prove that our attack is optimal irrespective of the application, and that designing strong robust aggregators is computationally as hard as finding the largest eigenvector in high dimensions. Our paper (arXiv) is published at IEEE S&P’24.

  • Retexo: A communication-efficient system for training Graph Neural Networks (GNNs) on expansive distributed graphs. This project stands out by provably enhancing the end-to-end training process efficiency over state-of-the-art systems. It can be used to train GNNs in data centers, collaborative setups, and on mobile/edge networks.

  • LPGNet: A novel Graph Neural Network architecture with strong differential privacy guarantees for graph edges, offering state-of-the-art privacy-utility tradeoffs. Thus, they prevent all attacks that steal the training graph data from the model. This work is published at CCS’22.

🔍 Other Contributions: My thesis also delves into crafting other differentially private queries on distributed data with notable publications in top-tier security conferences.

🌐 Broader Interests: Beyond my thesis pursuits, I actively engage in solving algorithmic problems and enjoy building systems. My contributions extend to diverse areas such as program synthesis, translation, debugging, and the security of decentralized applications, with publications in top-tier venues.

Academic History & Internships
drawing
B.Tech (2013-17)
drawing
Internship (May-Jul'16)
drawing
Internship (Aug'17-May'18)
drawing
PhD (Since 2018)
drawing
Internship (Jun-Jul'20)
drawing
Internship (Jun-Aug'23)

news

Mar 9, 2024 [New] Our paper on attacking byzantine robust aggregation protocols is published at IEEE S&P’24 (see arXiv).
Feb 1, 2024 [New] Please find our updated work on scalable neural network training on distributed graphs on arXiv.
Oct 1, 2023 Implemented several popular federated learning (FL) protocols for Flower framework as part of Summer of Reproducibility (see LinkedIn post).
Jun 1, 2023 Starting my research internship with Nokia Bell Labs.
Jan 1, 2023 Our paper on user-customizable automatic transpilation, DuoGlot, from Python to Javascript has been accepted to OOPSLA’23.