Zhoujie Ding

Zhoujie (Jason) Ding

I am currently a first-year MSCS student at Stanford University. I received my B.A. in computer science and applied mathematics at UC Berkeley.

At Stanford, I am working on trustworthy machine learning with Professor Sanmi Koyejo at Stanford Trustworthy AI Research.

At Berkeley, I worked on the Skyplane project at the SkyLab with Professor Joseph Gonzalez and Professor Ion Stoica. I also researched deep learning for code vulnerability detection under the supervision of Professor Yizheng Chen and Professor David Wagner.

Email / Resume / Google Scholar / LinkedIn / Github

Research

I am passionate about building machine learning systems and studying responsible foundation models (especially large language models). In the past, I worked on machine learning systems and deep learning for security issues.

DiverseVul: A New Vulnerable Source Code Dataset for Deep Learning Based Vulnerability Detection
Yizheng Chen, Zhoujie Ding, Lamya Alowain, Xinyun Chen, David Wagner
In the 26th International Symposium on Research in Attacks, Intrusions and Defenses (RAID 2023)
arXiv, dataset

We propose and release a new vulnerable source code dataset. Our new dataset contains 18,945 vulnerable functions spanning 150 CWEs and 330,492 non-vulnerable functions extracted from 7,514 commits. It covers 295 more projects than all previous datasets combined.

Continuous Learning for Android Malware Detection
Yizheng Chen, Zhoujie Ding, David Wagner
In the 32th USENIX Security Symposium (USENIX Security 2023)
arXiv
Top 10 Finalist of the CSAW'23 Applied Research Competition

We propose a new hierarchical contrastive learning with active learning to combat the concept drift problem of Android malware classifiers. Our approach reduces the false negative rate from 14% (for the best baseline) to 9%, while also reducing the false positive rate (from 0.86% to 0.48%).

Kernel-as-a-Service: A Serverless Interface to GPUs
Nathan Pemberton, Anton Zabreyko, Zhoujie Ding, Randy Katz, Joseph Gonzalez
Arxiv, 2022
arXiv

We present Kernel-as-a-Service (KaaS), a serverless interface to GPUs. KaaS runs graphs of GPU-only code while host code is run on traditional functions. Our results show that KaaS is able to drive up to 50x higher throughput and 16x lower latency when GPU resources are contended.

Teaching