Publications

Conference Papers

  • OSWorld-Human: Benchmarking the Efficiency of Computer-Use Agents
    Reyna Abhyankar*, Qi Qi*, Yiying Zhang
    Workshop on Computer Use Agents at ICML, 2025
    9th Annual Conference on Machine Learning and Systems 2026
    MLSys ‘2026 [BibTeX]

  • 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
    9th Annual Conference on Machine Learning and Systems 2026
    MLSys ‘2026 [BibTeX]

  • FarSight: Deep-Learning-Driven Prefetching for Far Memory
    Yutong Huang, Zhiyuan Guo, Yiying Zhang
    Workshop on Machine Learning for Systems at NeurIPS, 2025
    NeurIPS 2025 [BibTeX]

  • Demystifying Delays in Reasoning
    Qi Qi, Reyna Abhyankar,Yiying Zhang
    Workshop on Efficient Reasoning at NeurIPS, 2025
    NeurIPS 2025 [BibTeX]

  • Cognify: The Automated Optimizer for Generative AI Workflows
    Zijian He*, Reyna Abhyankar*, Vikranth Srivatsa, Yiying Zhang
    31st ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD), 2025
    KDD ‘25 [BibTeX]

  • 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

  • Preble: Efficient Distributed Prompt Scheduling for LLM Serving
    Vikranth Srivatsa*, Zijian He*, Reyna Abhyankar, Dongming Li, Yiying Zhang
    International Conference on Learning Representations (ICLR), 2025
    ICLR 2025 [BibTeX]

  • InferCept: Efficient Intercept Support for Augmented Large-Language Model Inferencing
    Reyna Abhyankar*, Zijian He*, Vikranth Srivatsa, Hao Zhang, Yiying Zhang
    International Conference on Machine Learning (ICML), 2024
    ICML 2024 [BibTeX]

Preprints

  • Deep-Learning-Driven Prefetching for Far Memory
    Yutong Huang, Zhiyuan Guo, Yiying Zhang
    Preprint, April 2025
    arXiv:2506.00384

  • AuditGPT: Auditing Smart Contracts with ChatGPT
    Shihao Xia, Shuai Shao, Mengting He, Tingting Yu, Linhai Song, Yiying Zhang
    Preprint, April 2024
    arXiv:2404.04306