Ziheng Jiang

According to our database, Ziheng Jiang authored at least 46 papers between 2010 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2026

DisagMoE: Computation-Communication overlapped MoE Training via Disaggregated AF-Pipe Parallelism.

Zhichen Zeng, Chi-Chih Chang, Jiayi Wang, Zezhou Wang, Ningxin Zheng, Zheng Zhong, Cesar A. Stuardo, Dongyang Wang, Mohamed S. Abdelfattah, Haibin Lin, Banghua Zhu, Ang Li, Ziheng Jiang

CoRR, May, 2026

SRT: Accelerating Reinforcement Learning via Speculative Rollout with Tree-Structured Cache.

Chi-Chih Chang, Siqi Zhu, Zhichen Zeng, Haibin Lin, Jiaxuan You, Mohamed S. Abdelfattah, Ziheng Jiang, Xuehai Qian

CoRR, January, 2026

MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production.

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

Proceedings of the 21st European Conference on Computer Systems, 2026

SwiftSpec: Disaggregated Speculative Decoding and Fused Kernels for Low-Latency LLM Inference.

Ziyi Zhang, Ziheng Jiang, Chengquan Jiang, Menghan Yu, Size Zheng, Haibin Lin, Xin Liu, Henry Hoffmann

Proceedings of the 31st ACM International Conference on Architectural Support for Programming Languages and Operating Systems, 2026

2025

Mesh-Attention: A New Communication-Efficient Distributed Attention with Improved Data Locality.

Sirui Chen, Jingji Chen, Siqi Zhu, Ziheng Jiang, Yanghua Peng, Xuehai Qian

CoRR, December, 2025

SwiftSpec: Ultra-Low Latency LLM Decoding by Scaling Asynchronous Speculative Decoding.

Ziyi Zhang, Ziheng Jiang, Chengquan Jiang, Menghan Yu, Size Zheng, Haibin Lin, Henry Hoffmann, Xin Liu

CoRR, June, 2025

MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production.

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

CoRR, May, 2025

Triton-distributed: Programming Overlapping Kernels on Distributed AI Systems with the Triton Compiler.

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

CoRR, April, 2025

MegaScale-Infer: Serving Mixture-of-Experts at Scale with Disaggregated Expert Parallelism.

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

CoRR, April, 2025

Batch Informed Vines (BIV*): Heuristically Guided Exploration of Narrow Passages by Batch Vine Expansion.

Ziheng Jiang, Chang Huang, Hongtu Zhou, Linlin Zhou, Chen Ye, Junqiao Zhao

IEEE Robotics Autom. Lett., February, 2025

MegaScale-Infer: Efficient Mixture-of-Experts Model Serving with Disaggregated Expert Parallelism.

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

Proceedings of the ACM SIGCOMM 2025 Conference, 2025

Understanding Stragglers in Large Model Training Using What-if Analysis.

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

Proceedings of the 19th USENIX Symposium on Operating Systems Design and Implementation, 2025

COMET: Fine-grained Computation-communication Overlapping for Mixture-of-Experts.

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

Proceedings of the Eighth Conference on Machine Learning and Systems, 2025

TileLink: Generating Efficient Compute-Communication Overlapping Kernels using Tile-Centric Primitives.

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

Proceedings of the Eighth Conference on Machine Learning and Systems, 2025

Relax: Composable Abstractions for End-to-End Dynamic Machine Learning.

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

Proceedings of the 30th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, 2025

2024

FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusion.

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

CoRR, 2024

MegaScale: Scaling Large Language Model Training to More Than 10, 000 GPUs.

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

Proceedings of the 21st USENIX Symposium on Networked Systems Design and Implementation, 2024

Towards Automated Chinese Ancient Character Restoration: A Diffusion-Based Method with a New Dataset.

Haolong Li, Chenghao Du, Ziheng Jiang, Yifan Zhang, Jiawei Ma, Chen Ye

Proceedings of the Thirty-Eighth AAAI Conference on Artificial Intelligence, 2024

2023

High Accuracy Near-Field Electromagnetic Emission Identification System Using Characteristics Recognition and Image Localization.

Yanhua Peng, Hui Xu, Hangyi Zhou, Ziheng Jiang, Dongdong Shen, Shuo Cui, Donglin Su

IEEE Trans. Instrum. Meas., 2023

EfficientPhys: Enabling Simple, Fast and Accurate Camera-Based Cardiac Measurement.

Xin Liu, Brian L. Hill, Ziheng Jiang, Shwetak N. Patel, Daniel McDuff

Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision, 2023

2022

DietCode: Automatic Optimization for Dynamic Tensor Programs.

Bojian Zheng, Ziheng Jiang, Cody Hao Yu, Haichen Shen, Joshua Fromm, Yizhi Liu, Yida Wang, Luis Ceze, Tianqi Chen, Gennady Pekhimenko

Proceedings of the Fifth Conference on Machine Learning and Systems, 2022

Federated Remote Physiological Measurement with Imperfect Data.

Xin Liu, Mingchuan Zhang, Ziheng Jiang, Shwetak N. Patel, Daniel McDuff

Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, 2022

2021

SplitSR: An End-to-End Approach to Super-Resolution on Mobile Devices.

Xin Liu, Yuang Li, Josh Fromm, Yuntao Wang, Ziheng Jiang, Alex Mariakakis, Shwetak N. Patel

Proc. ACM Interact. Mob. Wearable Ubiquitous Technol., 2021

EfficientPhys: Enabling Simple, Fast and Accurate Camera-Based Vitals Measurement.

Xin Liu, Brian L. Hill, Ziheng Jiang, Shwetak N. Patel, Daniel McDuff

CoRR, 2021

Automated Backend-Aware Post-Training Quantization.

Ziheng Jiang, Animesh Jain, Andrew Liu, Josh Fromm, Chengqian Ma, Tianqi Chen, Luis Ceze

CoRR, 2021

Characterizing Structural Regularities of Labeled Data in Overparameterized Models.

Ziheng Jiang, Chiyuan Zhang, Kunal Talwar, Michael C. Mozer

Proceedings of the 38th International Conference on Machine Learning, 2021

MetaPhys: few-shot adaptation for non-contact physiological measurement.

Xin Liu, Ziheng Jiang, Josh Fromm, Xuhai Xu, Shwetak N. Patel, Daniel McDuff

Proceedings of the ACM CHIL '21: ACM Conference on Health, 2021

2020

MetaPhys: Unsupervised Few-Shot Adaptation for Non-Contact Physiological Measurement.

Xin Liu, Ziheng Jiang, Josh Fromm, Xuhai Xu, Shwetak N. Patel, Daniel McDuff

CoRR, 2020

Exploring the Memorization-Generalization Continuum in Deep Learning.

Ziheng Jiang, Chiyuan Zhang, Kunal Talwar, Michael C. Mozer

CoRR, 2020

2019

A Hardware-Software Blueprint for Flexible Deep Learning Specialization.

Thierry Moreau, Tianqi Chen, Luis Vega, Jared Roesch, Eddie Q. Yan, Lianmin Zheng, Josh Fromm, Ziheng Jiang, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy

IEEE Micro, 2019

Relay: A High-Level IR for Deep Learning.

Jared Roesch, Steven Lyubomirsky, Marisa Kirisame, Josh Pollock, Logan Weber, Ziheng Jiang, Tianqi Chen, Thierry Moreau, Zachary Tatlock

CoRR, 2019

Just-in-Time Dynamic-Batching.

Sheng Zha, Ziheng Jiang, Haibin Lin, Zhi Zhang

CoRR, 2019

2018

VTA: An Open Hardware-Software Stack for Deep Learning.

Thierry Moreau, Tianqi Chen, Ziheng Jiang, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy

CoRR, 2018

TVM: End-to-End Optimization Stack for Deep Learning.

Tianqi Chen, Thierry Moreau, Ziheng Jiang, Haichen Shen, Eddie Q. Yan, Leyuan Wang, Yuwei Hu, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy

CoRR, 2018

A Path-constrained Framework for Discriminating Substitutable and Complementary Products in E-commerce.

Zihan Wang, Ziheng Jiang, Zhaochun Ren, Jiliang Tang, Dawei Yin

Proceedings of the Eleventh ACM International Conference on Web Search and Data Mining, 2018

Multi-Dimensional Network Embedding with Hierarchical Structure.

Yao Ma, Zhaochun Ren, Ziheng Jiang, Jiliang Tang, Dawei Yin

Proceedings of the Eleventh ACM International Conference on Web Search and Data Mining, 2018

TVM: An Automated End-to-End Optimizing Compiler for Deep Learning.

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

Proceedings of the 13th USENIX Symposium on Operating Systems Design and Implementation, 2018

Learning to Optimize Tensor Programs.

Tianqi Chen, Lianmin Zheng, Eddie Q. Yan, Ziheng Jiang, Thierry Moreau, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy

Proceedings of the Advances in Neural Information Processing Systems 31: Annual Conference on Neural Information Processing Systems 2018, 2018

2015

Erratum to: Image Fusion by Hierarchical Joint Sparse Representation.

Yao Yao, Ping Guo, Xin Xin, Ziheng Jiang

Cogn. Comput., 2015

2014

Image Fusion by Hierarchical Joint Sparse Representation.

Yao Yao, Ping Guo, Xin Xin, Ziheng Jiang

Cogn. Comput., 2014

Locality-Constrained Low-Rank Coding for Image Classification.

Ziheng Jiang, Ping Guo, Lihong Peng

Proceedings of the Twenty-Eighth AAAI Conference on Artificial Intelligence, 2014

2013

Learning open-domain comparable entity graphs from user search queries.

Ziheng Jiang, Lei Ji, Jianwen Zhang, Jun Yan, Ping Guo, Ning Liu

Proceedings of the 22nd ACM International Conference on Information and Knowledge Management, 2013

2012

Combining LVQ with SVM technique for image semantic annotation.

Ping Guo, Ziheng Jiang, Song Lin, Yao Yao

Neural Comput. Appl., 2012

Endless and Scalable Knowledge Table Extraction from Semi-structured Websites.

Yingqin Gu, Lei Ji, Ziheng Jiang, Jun He

Proceedings of the 12th IEEE International Conference on Data Mining Workshops, 2012

2011

A study of block-global feature based supervised image annotation.

Jing He, Ziheng Jiang, Ping Guo, Lixiong Liu

Proceedings of the IEEE International Conference on Systems, 2011

2010

Feature data optimization with LVQ technique in semantic image annotation.

Ziheng Jiang, Jing He, Ping Guo

Proceedings of the 10th International Conference on Intelligent Systems Design and Applications, 2010