Recent Publications

SourceBench: Can AI Answers Reference Quality Web Sources?
Hexi Jin, Stephen Liu, Yuheng Li, Simran Malik, Yiying Zhang
to appear at The Fortieth Annual Conference on Neural Information Processing Systems (NeurIPS '26)

TClone: Low-Latency Forking of Live GUI Environments for Computer-Use Agents
Yutong Huang, Vikranth Srivatsa, Alex Asch, Hansin Tushar Patwa, Yiying Zhang
arxiv preprint arXiv:2605.17320

Nitsum: Serving Tiered LLM Requests with Adaptive Tensor Parallelism
Vikranth Srivatsa, Zijian He, Pu Guo, Dongming Li, Yiying Zhang
arxiv preprint arXiv:2605.05467

VDCores: Resource Decoupled Programming and Execution for Asynchronous GPU
Zijian He, Adrian Sampson, Yiying Zhang, Zhiyuan Guo
arxiv preprint arXiv:2605.03190

SymGPT: Auditing Smart Contracts via Combining Symbolic Execution with Large Language Model
Shihao Xia, Mengting He, Shuai Shao, Tingting Yu, Yiying Zhang, Linhai Song
to appear at the the International Conference on Object-Oriented Programming Systems, Languages, and Applications (OOPSLA '26)

OSWorld-Human: Benchmarking the Efficiency of Computer-Use Agents
Reyna Abhyankar*, Qi Qi*, Yiying Zhang (* equal contribution)
Proceedings of the 9th Annual Conference on Machine Learning and Systems (MLSys '26)
the 1st Workshop on Computer-Use Agents Co-Located with ICML 2025 (WUCA '25)

Beat the long tail: Distribution-Aware Speculative Decoding for RL Training
Zelei Shao*, Vikranth Srivatsa*, Sanjana Srivastava, Qingyang Wu, Alpay Ariyak, Xiaoxia Wu, Ameen Patel, Jue Wang, Percy Liang, Tri Dao, Ce Zhang, Yiying Zhang, Ben Athiwaratkun, Chenfeng Xu, Junxiong Wang (* equal contribution)
Proceedings of the 9th Annual Conference on Machine Learning and Systems (MLSys '26)

How to Save My Gas Fees: Understanding and Detecting Real-world Gas Issues in Solidity Programs
Mengting He, Shihao Xia, Boqin Qin, Nobuko Yoshida, Tingting Yu, Yiying Zhang, Linhai Song
to appear at the IEEE/ACM International Conference on Software Engineering (ICSE '26)

Demystifying Delays in Reasoning: A Pilot Temporal and Token Analysis of Reasoning Systems
Qi Qi*, Reyna Abhyankar*, Yiying Zhang (* equal contribution)
the 1st Workshop on Efficient Reasoning Co-Located with NeurIPS 2025 (ER '25)

An Early Exploration of Deep-Learning-Driven Prefetching for Far Memory
Yutong Huang, Zhiyuan Guo, Yiying Zhang
the 2025 Workshop on Machine Learning for Systems Co-Located with NeurIPS 2025 (MLForSys '25)

Learning Semantics, Not Addresses: Runtime Neural Prefetching for Far Memory
Yutong Huang, Zhiyuan Guo, Yiying Zhang
arxiv preprint arXiv:2506.00384

Cognify: Supercharging Gen-AI Workflows With Hierarchical Autotuning
Zijian He*, Reyna Abhyankar*, Vikranth Srivatsa, Yiying Zhang (* equal contribution)
Proceedings of the 31st ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD '25)

Preble: Efficient Distributed Prompt Scheduling for LLM Serving
Vikranth Srivatsa*, Zijian He*, Reyna Abhyankar, Dongming Li, Yiying Zhang (* equal contribution)
the Thirteenth International Conference on Learning Representations (ICLR '25)

SC-Bench: A Large-Scale Dataset for Smart Contract Auditing
Shihao Xia, Mengting He, Linhai Song, Yiying Zhang
Second International Workshop on Large Language Models for Code (LLM4Code '25)

Portable and High-Performance SmartNIC Programs with Alkali
Jiaxin Lin*, Zhiyuan Guo*, Mihir Shah, Tao Ji, Yiying Zhang, Daehyeok Kim, Aditya Akella (* equal contribution)
Proceedings of the 22nd USENIX Symposium on Networked Systems Design and Implementation (NSDI '25)

Teaching