Ziheng Jiang

53 papers A* 7A 5B 1C 1Misc 1Journal 30Unranked 8
YearRankTypeTitle / Venue / Authors
2026 A conf
MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production.
EuroSys
Chao Jin, Ziheng Jiang, Zhihao Bai, Zheng Zhong, Juncai Liu, Xiang Li, Ningxin Zheng, Xi Wang, Cong Xie, Qi Huang, Wen Heng, Yiyuan Ma, Wenlei Bao, Size Zheng, Xuegui Zheng, Yanghua Peng, Haibin Lin, Xuanzhe Liu, Xin Jin, Xin Liu
2026 J jnl
SRT: Accelerating Reinforcement Learning via Speculative Rollout with Tree-Structured Cache.
CoRR
Chi-Chih Chang, Siqi Zhu, Zhichen Zeng, Haibin Lin, Jiaxuan You, Mohamed S. Abdelfattah, Ziheng Jiang, Xuehai Qian
2026 — conf
SwiftSpec: Disaggregated Speculative Decoding and Fused Kernels for Low-Latency LLM Inference.
ASPLOS (2)
Ziyi Zhang, Ziheng Jiang, Chengquan Jiang, Menghan Yu, Size Zheng, Haibin Lin, Xin Liu, Henry Hoffmann
2025 J jnl
Batch Informed Vines (BIV*): Heuristically Guided Exploration of Narrow Passages by Batch Vine Expansion.
IEEE Robotics Autom. Lett.
Ziheng Jiang, Chang Huang, Hongtu Zhou, Linlin Zhou, Chen Ye, Junqiao Zhao
2025 — conf
COMET: Fine-grained Computation-communication Overlapping for Mixture-of-Experts.
MLSys
Shulai Zhang, Ningxin Zheng, Haibin Lin, Ziheng Jiang, Wenlei Bao, Chengquan Jiang, Qi Hou, Weihao Cui, Size Zheng, Li-Wen Chang, Quan Chen, Xin Liu
2025 J jnl
Comet: Fine-grained Computation-communication Overlapping for Mixture-of-Experts.
CoRR
Shulai Zhang, Ningxin Zheng, Haibin Lin, Ziheng Jiang, Wenlei Bao, Chengquan Jiang, Qi Hou, Weihao Cui, Size Zheng, Li-Wen Chang, Quan Chen, Xin Liu
2025 A* conf
MegaScale-Infer: Efficient Mixture-of-Experts Model Serving with Disaggregated Expert Parallelism.
SIGCOMM
Ruidong Zhu, Ziheng Jiang, Chao Jin, Peng Wu, Cesar A. Stuardo, Dongyang Wang, Xinlei Zhang, Huaping Zhou, Haoran Wei, Yang Cheng, Jianzhe Xiao, Xinyi Zhang, Lingjun Liu, Haibin Lin, Li-Wen Chang, Jianxi Ye, Xiao Yu, Xuanzhe Liu, Xin Jin, Xin Liu
2025 J jnl
MegaScale-Infer: Serving Mixture-of-Experts at Scale with Disaggregated Expert Parallelism.
CoRR
Ruidong Zhu, Ziheng Jiang, Chao Jin, Peng Wu, Cesar A. Stuardo, Dongyang Wang, Xinlei Zhang, Huaping Zhou, Haoran Wei, Yang Cheng, Jianzhe Xiao, Xinyi Zhang, Lingjun Liu, Haibin Lin, Li-Wen Chang, Jianxi Ye, Xiao Yu, Xuanzhe Liu, Xin Jin, Xin Liu
2025 J jnl
MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production.
CoRR
Chao Jin, Ziheng Jiang, Zhihao Bai, Zheng Zhong, Juncai Liu, Xiang Li, Ningxin Zheng, Xi Wang, Cong Xie, Qi Huang, Wen Heng, Yiyuan Ma, Wenlei Bao, Size Zheng, Yanghua Peng, Haibin Lin, Xuanzhe Liu, Xin Jin, Xin Liu
2025 J jnl
Mesh-Attention: A New Communication-Efficient Distributed Attention with Improved Data Locality.
CoRR
Sirui Chen, Jingji Chen, Siqi Zhu, Ziheng Jiang, Yanghua Peng, Xuehai Qian
2025 — conf
Relax: Composable Abstractions for End-to-End Dynamic Machine Learning.
ASPLOS (2)
Ruihang Lai, Junru Shao, Siyuan Feng, Steven Lyubomirsky, Bohan Hou, Wuwei Lin, Zihao Ye, Hongyi Jin, Yuchen Jin, Jiawei Liu, Lesheng Jin, Yaxing Cai, Ziheng Jiang, Yong Wu, Sunghyun Park, Prakalp Srivastava, Jared Roesch, Todd C. Mowry, Tianqi Chen
2025 J jnl
SwiftSpec: Ultra-Low Latency LLM Decoding by Scaling Asynchronous Speculative Decoding.
CoRR
Ziyi Zhang, Ziheng Jiang, Chengquan Jiang, Menghan Yu, Size Zheng, Haibin Lin, Henry Hoffmann, Xin Liu
2025 — conf
TileLink: Generating Efficient Compute-Communication Overlapping Kernels using Tile-Centric Primitives.
MLSys
Size Zheng, Jin Fang, Xuegui Zheng, Qi Hou, Wenlei Bao, Ningxin Zheng, Ziheng Jiang, Dongyang Wang, Jianxi Ye, Haibin Lin, Li-Wen Chang, Xin Liu
2025 J jnl
TileLink: Generating Efficient Compute-Communication Overlapping Kernels using Tile-Centric Primitives.
CoRR
Size Zheng, Jin Fang, Xuegui Zheng, Qi Hou, Wenlei Bao, Ningxin Zheng, Ziheng Jiang, Dongyang Wang, Jianxi Ye, Haibin Lin, Li-Wen Chang, Xin Liu
2025 J jnl
Triton-distributed: Programming Overlapping Kernels on Distributed AI Systems with the Triton Compiler.
CoRR
Size Zheng, Wenlei Bao, Qi Hou, Xuegui Zheng, Jin Fang, Chenhui Huang, Tianqi Li, Haojie Duanmu, Renze Chen, Ruifan Xu, Yifan Guo, Ningxin Zheng, Ziheng Jiang, Xinyi Di, Dongyang Wang, Jianxi Ye, Haibin Lin, Li-Wen Chang, Liqiang Lu, Yun Liang, Jidong Zhai, Xin Liu
2025 A* conf
Understanding Stragglers in Large Model Training Using What-if Analysis.
OSDI
Jinkun Lin, Ziheng Jiang, Zuquan Song, Sida Zhao, Menghan Yu, Zhanghan Wang, Chenyuan Wang, Zuocheng Shi, Xiang Shi, Wei Jia, Zherui Liu, Shuguang Wang, Haibin Lin, Xin Liu, Aurojit Panda, Jinyang Li
2025 J jnl
Understanding Stragglers in Large Model Training Using What-if Analysis.
CoRR
Jinkun Lin, Ziheng Jiang, Zuquan Song, Sida Zhao, Menghan Yu, Zhanghan Wang, Chenyuan Wang, Zuocheng Shi, Xiang Shi, Wei Jia, Zherui Liu, Shuguang Wang, Haibin Lin, Xin Liu, Aurojit Panda, Jinyang Li
2024 J jnl
FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusion.
CoRR
Li-Wen Chang, Wenlei Bao, Qi Hou, Chengquan Jiang, Ningxin Zheng, Yinmin Zhong, Xuanrun Zhang, Zuquan Song, Ziheng Jiang, Haibin Lin, Xin Jin, Xin Liu
2024 Misc conf
MegaScale: Scaling Large Language Model Training to More Than 10, 000 GPUs.
NSDI
Ziheng Jiang, Haibin Lin, Yinmin Zhong, Qi Huang, Yangrui Chen, Zhi Zhang, Yanghua Peng, Xiang Li, Cong Xie, Shibiao Nong, Yulu Jia, Sun He, Hongmin Chen, Zhihao Bai, Qi Hou, Shipeng Yan, Ding Zhou, Yiyao Sheng, Zhuo Jiang, Haohan Xu, Haoran Wei, Zhang Zhang, Pengfei Nie, Leqi Zou, Sida Zhao, Liang Xiang, Zherui Liu, Zhe Li, Xiaoying Jia, Jianxi Ye, Xin Jin, Xin Liu
2024 J jnl
MegaScale: Scaling Large Language Model Training to More Than 10, 000 GPUs.
CoRR
Ziheng Jiang, Haibin Lin, Yinmin Zhong, Qi Huang, Yangrui Chen, Zhi Zhang, Yanghua Peng, Xiang Li, Cong Xie, Shibiao Nong, Yulu Jia, Sun He, Hongmin Chen, Zhihao Bai, Qi Hou, Shipeng Yan, Ding Zhou, Yiyao Sheng, Zhuo Jiang, Haohan Xu, Haoran Wei, Zhang Zhang, Pengfei Nie, Leqi Zou, Sida Zhao, Liang Xiang, Zherui Liu, Zhe Li, Xiaoying Jia, Jianxi Ye, Xin Jin, Xin Liu
2024 A* conf
Towards Automated Chinese Ancient Character Restoration: A Diffusion-Based Method with a New Dataset.
AAAI
Haolong Li, Chenghao Du, Ziheng Jiang, Yifan Zhang, Jiawei Ma, Chen Ye
2023 A conf
EfficientPhys: Enabling Simple, Fast and Accurate Camera-Based Cardiac Measurement.
WACV
Xin Liu, Brian L. Hill, Ziheng Jiang, Shwetak N. Patel, Daniel McDuff
2023 J jnl
High Accuracy Near-Field Electromagnetic Emission Identification System Using Characteristics Recognition and Image Localization.
IEEE Trans. Instrum. Meas.
Yanhua Peng, Hui Xu, Hangyi Zhou, Ziheng Jiang, Dongdong Shen, Shuo Cui, Donglin Su
2023 J jnl
Relax: Composable Abstractions for End-to-End Dynamic Machine Learning.
CoRR
Ruihang Lai, Junru Shao, Siyuan Feng, Steven S. Lyubomirsky, Bohan Hou, Wuwei Lin, Zihao Ye, Hongyi Jin, Yuchen Jin, Jiawei Liu, Lesheng Jin, Yaxing Cai, Ziheng Jiang, Yong Wu, Sunghyun Park, Prakalp Srivastava, Jared G. Roesch, Todd C. Mowry, Tianqi Chen
2022 — conf
DietCode: Automatic Optimization for Dynamic Tensor Programs.
MLSys
Bojian Zheng, Ziheng Jiang, Cody Hao Yu, Haichen Shen, Joshua Fromm, Yizhi Liu, Yida Wang, Luis Ceze, Tianqi Chen, Gennady Pekhimenko
2022 — conf
Federated Remote Physiological Measurement with Imperfect Data.
CVPR Workshops
Xin Liu, Mingchuan Zhang, Ziheng Jiang, Shwetak N. Patel, Daniel McDuff
2022 J jnl
Federated Remote Physiological Measurement with Imperfect Data.
CoRR
Xin Liu, Mingchuan Zhang, Ziheng Jiang, Shwetak N. Patel, Daniel McDuff
2021 J jnl
Automated Backend-Aware Post-Training Quantization.
CoRR
Ziheng Jiang, Animesh Jain, Andrew Liu, Josh Fromm, Chengqian Ma, Tianqi Chen, Luis Ceze
2021 A* conf
Characterizing Structural Regularities of Labeled Data in Overparameterized Models.
ICML
Ziheng Jiang, Chiyuan Zhang, Kunal Talwar, Michael C. Mozer
2021 J jnl
EfficientPhys: Enabling Simple, Fast and Accurate Camera-Based Vitals Measurement.
CoRR
Xin Liu, Brian L. Hill, Ziheng Jiang, Shwetak N. Patel, Daniel McDuff
2021 — conf
MetaPhys: few-shot adaptation for non-contact physiological measurement.
CHIL
Xin Liu, Ziheng Jiang, Josh Fromm, Xuhai Xu, Shwetak N. Patel, Daniel McDuff
2021 J jnl
SplitSR: An End-to-End Approach to Super-Resolution on Mobile Devices.
CoRR
Xin Liu, Yuang Li, Josh Fromm, Yuntao Wang, Ziheng Jiang, Alex Mariakakis, Shwetak N. Patel
2021 J jnl
SplitSR: An End-to-End Approach to Super-Resolution on Mobile Devices.
Proc. ACM Interact. Mob. Wearable Ubiquitous Technol.
Xin Liu, Yuang Li, Josh Fromm, Yuntao Wang, Ziheng Jiang, Alex Mariakakis, Shwetak N. Patel
2020 J jnl
Exploring the Memorization-Generalization Continuum in Deep Learning.
CoRR
Ziheng Jiang, Chiyuan Zhang, Kunal Talwar, Michael C. Mozer
2020 J jnl
MetaPhys: Unsupervised Few-Shot Adaptation for Non-Contact Physiological Measurement.
CoRR
Xin Liu, Ziheng Jiang, Josh Fromm, Xuhai Xu, Shwetak N. Patel, Daniel McDuff
2019 J jnl
A Hardware-Software Blueprint for Flexible Deep Learning Specialization.
IEEE Micro
Thierry Moreau, Tianqi Chen, Luis Vega, Jared Roesch, Eddie Q. Yan, Lianmin Zheng, Josh Fromm, Ziheng Jiang, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy
2019 J jnl
Just-in-Time Dynamic-Batching.
CoRR
Sheng Zha, Ziheng Jiang, Haibin Lin, Zhi Zhang
2019 J jnl
Relay: A High-Level IR for Deep Learning.
CoRR
Jared Roesch, Steven Lyubomirsky, Marisa Kirisame, Josh Pollock, Logan Weber, Ziheng Jiang, Tianqi Chen, Thierry Moreau, Zachary Tatlock
2018 A conf
A Path-constrained Framework for Discriminating Substitutable and Complementary Products in E-commerce.
WSDM
Zihan Wang, Ziheng Jiang, Zhaochun Ren, Jiliang Tang, Dawei Yin
2018 A* conf
Learning to Optimize Tensor Programs.
NeurIPS
Tianqi Chen, Lianmin Zheng, Eddie Q. Yan, Ziheng Jiang, Thierry Moreau, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy
2018 J jnl
Learning to Optimize Tensor Programs.
CoRR
Tianqi Chen, Lianmin Zheng, Eddie Q. Yan, Ziheng Jiang, Thierry Moreau, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy
2018 A conf
Multi-Dimensional Network Embedding with Hierarchical Structure.
WSDM
Yao Ma, Zhaochun Ren, Ziheng Jiang, Jiliang Tang, Dawei Yin
2018 A* conf
TVM: An Automated End-to-End Optimizing Compiler for Deep Learning.
OSDI
Tianqi Chen, Thierry Moreau, Ziheng Jiang, Lianmin Zheng, Eddie Q. Yan, Haichen Shen, Meghan Cowan, Leyuan Wang, Yuwei Hu, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy
2018 J jnl
TVM: End-to-End Optimization Stack for Deep Learning.
CoRR
Tianqi Chen, Thierry Moreau, Ziheng Jiang, Haichen Shen, Eddie Q. Yan, Leyuan Wang, Yuwei Hu, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy
2018 J jnl
VTA: An Open Hardware-Software Stack for Deep Learning.
CoRR
Thierry Moreau, Tianqi Chen, Ziheng Jiang, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy
2015 J jnl
Erratum to: Image Fusion by Hierarchical Joint Sparse Representation.
Cogn. Comput.
Yao Yao, Ping Guo, Xin Xin, Ziheng Jiang
2014 J jnl
Image Fusion by Hierarchical Joint Sparse Representation.
Cogn. Comput.
Yao Yao, Ping Guo, Xin Xin, Ziheng Jiang
2014 A* conf
Locality-Constrained Low-Rank Coding for Image Classification.
AAAI
Ziheng Jiang, Ping Guo, Lihong Peng
2013 A conf
Learning open-domain comparable entity graphs from user search queries.
CIKM
Ziheng Jiang, Lei Ji, Jianwen Zhang, Jun Yan, Ping Guo, Ning Liu
2012 J jnl
Combining LVQ with SVM technique for image semantic annotation.
Neural Comput. Appl.
Ping Guo, Ziheng Jiang, Song Lin, Yao Yao
2012 — conf
Endless and Scalable Knowledge Table Extraction from Semi-structured Websites.
ICDM Workshops
Yingqin Gu, Lei Ji, Ziheng Jiang, Jun He
2011 B conf
A study of block-global feature based supervised image annotation.
SMC
Jing He, Ziheng Jiang, Ping Guo, Lixiong Liu
2010 C conf
Feature data optimization with LVQ technique in semantic image annotation.
ISDA
Ziheng Jiang, Jing He, Ping Guo