Publications

Please find the complete list on Google Scholar.

2026

  1. Bit-Accurate Modeling of GPU Matrix Multiply-Accumulate Units: Demystifying Numerical Discrepancy and Accuracy
    Peichen Xie, and
    ArXiv 2026.
  2. STAR-PólyaMath: Multi-Agent Reasoning under Persistent Meta-Strategic Supervision
    Jiaao Wu, and
    ArXiv 2026.
  3. Unifying Sparse Attention with Hierarchical Memory for Scalable Long-Context LLM Serving
    Zihan Zhao, and
    ArXiv 2026.
  4. DualSpec: Accelerating Deep Research Agents via Dual-Process Action Speculation
    Shuzhang Zhong, and
    ArXiv 2026.
  5. MLSys
    Virtual Machine NUMA Placement at Scale: Learning the Norm, Shielding the Tail
    Yibo Zhao, and
    MLSys 2026.
  6. Euclean: Automated Geometry Problem Formalization with Unified Verification in Lean
    Linbin Tang, and
    ICML 2026.
  7. VLDB
    RetroInfer: A Vector Storage Engine for Scalable Long-Context LLM Inference
    Yaoqi Chen, and
    VLDB 2026.
  8. Sparse Attention Adaptation for Long Reasoning
    Yizhao Gao, and
    ICLR 2026.
  9. TileLang: Bridge Programmability and Performance in Modern Neural Kernels
    Lei Wang, and
    ICLR 2026.
  10. LoongRL: Reinforcement Learning for Advanced Reasoning over Long Contexts
    Siyuan Wang, and
    ICLR 2026.
  11. PPoPP
    MetaAttention: A Unified and Performant Attention Framework across Hardware Backends
    Feiyang Chen, and
    PPoPP 2026.

2025

  1. Vibe Reasoning: Eliciting Frontier AI Mathematical Capabilities – A Case Study on IMO 2025 Problem 6
    Jiaao Wu, and
    ArXiv 2025.
  2. Reviving DSP for Advanced Theorem Proving in the Era of Reasoning Models
    Chenrui Cao, and
    NeurIPS 2025.
  3. rStar-Coder: Scaling Competitive Code Reasoning with a Large-Scale Verified Dataset
    Yifei Liu, and
    NeurIPS 2025.
  4. RetrievalAttention: Accelerating Long-Context LLM Inference via Vector Retrieval
    Di Liu, and
    NeurIPS 2025.
  5. SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs
    Yizhao Gao, and
    NeurIPS 2025.
  6. TrainVerify: Equivalence-Based Verification for Distributed LLM Training
    Yunchi Lu, and
    SOSP 2025. ArXiv version.
  7. rStar2-Agent: Agentic Reasoning Technical Report
    Ning Shang, and
    ArXiv 2025.
  8. LUT Tensor Core: A Software-Hardware Co-Design for LUT-Based Low-Bit LLM Inference
    Zhiwen Mo, and
    ISCA 2025.
  9. WaferLLM: A Wafer-Scale LLM Inference System
    Congjie He, and
  10. PipeThreader: Software-Defined Pipelining for Efficient DNN Execution
    Yu Cheng, and
    OSDI 2025.
  11. TileLang: A Composable Tiled Programming Model for AI Systems
    Lei Wang, and
    ArXiv 2025. (A revised version appears in ICLR’26).
  12. RetroInfer: A Vector-Storage Approach for Scalable Long-Context LLM Inference
    Yaoqi Chen, and
    ArXiv 2025. (A revised version appears in VLDB’26).
  13. rStar-Math: Small LLMs Can Master Math Reasoning with Self-Evolved Deep Thinking
    Xinyu Guan, and
    ICML 2025.
  14. LongRoPE2: Near-Lossless LLM Context Window Scaling
    Ning Shang, and
    ICML 2025.
  15. Automated Proof Generation for Rust Code via Self-Evolution
    Tianyu Chen, and
    ICLR 2025.
  16. Mutual Reasoning Makes Smaller LLMs Stronger Problem-Solvers
    Zhenting Qi, and
    ICLR 2025.
  17. Proving Olympiad Inequalities by Synergizing LLMs and Symbolic Reasoning
    Zenan Li, and
    ICLR 2025.
  18. LUT-DLA: Lookup Table as Efficient Extreme Low-Bit Deep Learning Accelerator
    Guoyu Li, and
    HPCA 2025.
  19. OOPSLA
    AutoVerus: Automated Proof Generation for Rust Code
    Chenyuan Yang, and
    OOPSLA Oct. 2025. Site, the ArXiv version.

2024

  1. LUT Tensor Core: Lookup Table Enables Efficient Low-Bit LLM Inference Acceleration
    Zhiwen Mo, and
    ArXiv 2024. (A revised version will appear in ISCA’25).
  2. Autoformalize Mathematical Statements by Symbolic Equivalence and Semantic Consistency
    Zenan Li, and
    NeurIPS 2024.
  3. Neuro-Symbolic Data Generation for Math Reasoning
    Zenan Li, and
    NeurIPS 2024.
  4. EMNLP
    Fewer is More: Boosting LLM Reasoning with Reinforced Context Pruning
    Xijie Huang, and
    EMNLP 2024.
  5. Uncovering Nested Data Parallelism and Data Reuse in DNN Computation with FractalTensor
    Siran Liu, and
    SOSP 2024.
  6. ECCV
    IRGen: Generative Modeling for Image Retrieval
    Yidan Zhang, and
    ECCV 2024.
  7. WWW
    OneSparse: A Unified System for Multi-index Vector Search
    Yaoqi Chen, and
    WWW 2024, Singapore, Singapore.
  8. WWW
    MS MARCO Web Search: A Large-scale Information-rich Web Dataset with Millions of Real Click Labels
    Qi Chen, and
    WWW 2024, Singapore, Singapore.
  9. KDD
    Understanding the Weakness of Large Language Model Agents within a Complex Android Environment
    Mingzhe Xing, and
    KDD 2024, Barcelona, Spain.
  10. LongRoPE: Extending LLM Context Window Beyond 2 Million Tokens
    Yiran Ding, and
    ICML 2024.
  11. Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phone
    Marah Abdin, and
    ArXiv 2024. (Applying LongRoPE to Phi-3).
  12. nnScaler: Constraint-Guided Parallelization Plan Generation for Deep Learning Training
    Zhiqi Lin, and
    OSDI 2024.
  13. Parrot: Efficient Serving of LLM-based Applications with Semantic Variable
    Chaofan Lin, and
    OSDI 2024.
  14. Ladder: Enabling Efficient Low-Precision Deep Learning Computing through Hardware-aware Tensor Transformation
    Lei Wang, and
    OSDI 2024.
  15. Amanda: Unified Instrumentation Framework for Deep Neural Networks
    Yue Guan, and
    ASPLOS 2024.
  16. Aceso: Efficient Parallel DNN Training through Iterative Bottleneck Alleviation
    Guodong Liu, and
    EuroSys 2024.
  17. Tessel: Boosting Distributed DNN Execution with Flexible Schedule Search
    Zhiqi Lin, and
    HPCA 2024.
  18. ICME
    Integer or Floating Point? New Outlooks for Low-Bit Quantization on Large Language Models
    Yijia Zhang, and
    ICME 2024.

2023

  1. Adam Accumulation to Reduce Memory Footprints of both Activations and Gradients for Large-scale DNN Training
    Yijia Zhang, and
    ArXiv 2023.
  2. SuperScaler: Supporting Flexible DNN Parallelization via a Unified Abstraction
    Zhiqi Lin, and
    ArXiv 2023.
  3. VBASE: Unifying Online Vector Similarity Search and Relational Queries via Relaxed Monotonicity
    Qianxi Zhang, and
    OSDI 2023.
  4. Cocktailer: Analyzing and Optimizing Dynamic Control Flow in Deep Learning
    Chen Zhang, and
    OSDI 2023.
  5. Welder: Scheduling Deep Learning Memory Access via Tile-graph
    Yining Shi, and
    OSDI 2023.
  6. Optimizing Dynamic Neural Networks with Brainstorm
    Weihao Cui, and
    OSDI 2023.
  7. BitNet: Scaling 1-bit Transformers for Large Language Models
    Hongyu Wang, and
    ArXiv 2023.
  8. PIT: Optimization of Dynamic Sparse Deep Learning Models via Permutation Invariant Transformation
    Ningxin Zheng, and
    SOSP 2023.
  9. SPFresh: Incremental In-Place Update for Billion-Scale Vector Search
    Yuming Xu, and
    SOSP 2023.
  10. SiloD: A Co-design of Caching and Scheduling for Deep Learning Clusters
    Hanyu Zhao, and
    EuroSys 2023.
  11. ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning
    Diandian Gu, and
    ASPLOS 2023.
  12. OliVe: Accelerating Large Language Models via Hardware-friendly Outlier-Victim Pair Quantization
    Cong Guo, and
    ISCA 2023.
  13. On Modular Learning of Distributed Systems for Predicting End-to-End Latency
    Chieh-Jan Mike Liang, and
    NSDI 2023.
  14. Model-enhanced Vector Index
    Hailin Zhang, and
    NeurIPS 2023.
  15. MLSys
    Tutel: Adaptive Mixture-of-Experts at Scale
    Changho Hwang, and
    MLSys 2023.
  16. MLSys
    Efficient GPU Kernels for N:M-Sparse Weights in Deep Learning
    Bin Lin, and
    MLSys 2023.
  17. ACL
    NUWA-XL: Diffusion over Diffusion for eXtremely Long Video Generation
    Shengming Yin, and
    ACL Jul 2023.
  18. IJCAI
    Learning 3D photography videos via self-supervised diffusion on single images
    Xiaodong Wang, and
    IJCAI 2023, Macao, P.R.China.

2022

  1. SQuant: On-the-Fly Data-Free Quantization via Diagonal Hessian Approximation
    Cong Guo, and
    ICLR 2022.
  2. ICCD
    Nesting Forward Automatic Differentiation for Memory-Efficient Deep Neural Network Training
    Cong Guo, and
    ICCD 2022.
  3. ANT: Exploiting Adaptive Numerical Data Type for Low-bit Deep Neural Network Quantization
    Cong Guo, and
    MICRO 2022.
  4. ROLLER: Fast and Efficient Tensor Compilation for Deep Learning
    Hongyu Zhu, and
    OSDI 2022.
  5. PilotFish: Harvesting Free Cycles of Cloud Gaming with Deep Learning Training
    Wei Zhang, and
    USENIX ATC 2022.
  6. SparTA: Deep-Learning Model Sparsity via Tensor-with-Sparsity-Attribute
    Ningxin Zheng, and
    OSDI 2022.
  7. SIGIR
    Distill-VQ: Learning Retrieval Oriented Vector Quantization By Distilling Knowledge from Dense Embeddings
    Shitao Xiao, and
    SIGIR 2022, Madrid, Spain.
  8. ECCV
    NÜWA: Visual Synthesis Pre-training for Neural visUal World creAtion
    Chenfei Wu, and
    ECCV 2022.

2021

  1. GODIVA: Generating Open-DomaIn Videos from nAtural Descriptions
    Chenfei Wu, and
    ArXiv 2021.

2020

  1. EMNLP
    XGLUE: A New Benchmark Dataset for Cross-lingual Pre-training, Understanding and Generation
    Yaobo Liang, and
    EMNLP Nov. 2020.
  2. Capuchin: Tensor-based GPU Memory Management for Deep Learning
    Xuan Peng, and
    ASPLOS 2020.
  3. HiveD: Sharing a GPU Cluster for Deep Learning with Guarantees
    Hanyu Zhao, and
    OSDI 2020.
  4. Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks
    Lingxiao Ma, and
    OSDI 2020.
  5. Retiarii: A Deep Learning Exploratory-Training Framework
    Quanlu Zhang, and
    OSDI 2020.

2019

  1. Analysis of Large-Scale Multi-Tenant GPU Clusters for DNN Training Workloads
    Myeongjae Jeon, and
    USENIX ATC 2019.

2018

  1. Gandiva: Introspective Cluster Scheduling for Deep Learning
    Wencong Xiao, and
    OSDI 2018.
  2. Scheduling CPU for GPU-based Deep Learning Jobs
    Wencong Xiao, and
    In Proceedings of the ACM Symposium on Cloud Computing (SoCC) Poster, 2018, Carlsbad, CA, USA.

2015

  1. SoCC
    GraM: scaling graph computation to the trillions
    Ming Wu, and
    SoCC , 2015, Kohala Coast, Hawaii.
  2. ToS
    ImmortalGraph: A System for Storage and Analysis of Temporal Graphs
    Youshan Miao, and
    ToS , Jul 2015.

2014

  1. Chronos: a graph engine for temporal graph analysis
    Wentao Han, and
    EuroSys , 2014, Amsterdam, The Netherlands.
  2. TechReport
    Arming Cloud Services with Task Aspects
    Zhenyu Guo, and
    TechReport , Nov. 2014.

2012

  1. Kineograph: taking the pulse of a fast-changing and connected world
    Raymond Cheng, and
    EuroSys , 2012, Bern, Switzerland.
  2. TechReport
    Sonora: A Platform for Continuous Mobile-Cloud Computing
    Xiuwei Chen, and
    TechReport , Mar. 2012.

2007

  1. VTC
    Distributed Cooperative Rate Adaptation for Energy Efficiency in IEEE 802.11-Based Multihop Networks
    Kun Wang, and
    VTC , 2007.
  2. IEEENetwork
    Cooperative and opportunistic transmission for wireless ad hoc networks
    Qian Zhang, and
    IEEENetwork , 2007.
  3. TWC
    Modeling path capacity in multi-hop IEEE 802.11 networks for QoS services
    Kun Wang, and
    TWC , 2007.

2006

  1. QShine
    Distributed cooperative rate adaptation for energy efficiency in IEEE 802.11-based multi-hop networks
    Kun Wang, and
    QShine , 2006, Waterloo, Ontario, Canada.
  2. Globecom
    On Improving the Throughput of Media Delivery Applications in Heterogeneous Overlay Network
    Jin Zhao, and
    Globecom , 2006.
  3. JSAC
    Distributed Channel Assignment and Routing in Multiradio Multichannel Multihop Wireless Networks
    H. Wu, and
    JSAC , 2006.
  4. ChinaCom
    Next generation mobile multimedia communications: Media codec and media transport perspectives
    Feng Wu, and
    ChinaCom , 2006.
  5. ToM
    LION: Layered Overlay Multicast With Network Coding
    J. Zhao, and
    ToM , 2006.
  6. ICC
    Impact of Power and Rate Selection on the Throughput of Ad Hoc Networks
    Cong Peng, and
    ICC , 2006.

2005

  1. Cross-layer QoS support for multimedia delivery over wireless Internet
    Qian Zhang, and
    EURASIP Journal on Advances in Signal Processing, 2005.
  2. ICC
    AMTP: a multipath multimedia streaming protocol for mobile ad hoc networks
    K. Rojviboonchai, and
    ICC , 2005.

2004

  1. JSAC
    End-to-end TCP-friendly streaming protocol and bit allocation for scalable video over wireless Internet
    Fan Yang, and
    JSAC , 2004.
  2. INFOCOM
    Bit allocation for scalable video streaming over mobile wireless Internet
    Fan Yang, and
    INFOCOM , 2004.

2003

  1. ICME
    An end-to-end TCP-friendly streaming protocol for multimedia over wireless Internet
    Fan Yang, and
    ICME , 2003.

2001

  1. 3GWireless
    An efficient transport scheme for multimedia over wireless internet
    Fan Yang, and
    3GWireless , 2001.