activity
20182026
most citedTensorOpt: Exploring the Tradeoffs in Distributed DNN Training with Auto-Parallelism

45 citations · 46 across the 8 of their papers we have counts for

collaborators

8 papers

cs.DC2026

CRAFT: Fine-Grained Cost-Aware Expert Replication For Efficient Mixture-of-Experts Serving

Adrian Zhao, Zhenkun Cai, Zhenyu Song +5

Mixture-of-Experts (MoE) has recently emerged as the mainstream architecture for efficiently scaling large language models while maintaining near-constant computational cost. Exper…

cs.DC2025

DCP: Addressing Input Dynamism In Long-Context Training via Dynamic Context Parallelism

Chenyu Jiang, Zhenkun Cai, Ye Tian +3

Context parallelism has emerged as a key technique to support long-context training, a growing trend in generative AI for modern large models. However, existing context parallel me…

cs.LG2024★ 1 cited

DiskGNN: Bridging I/O Efficiency and Model Accuracy for Out-of-Core GNN Training

Renjie Liu, Yichuan Wang, Xiao Yan +5

Graph neural networks (GNNs) are machine learning models specialized for graph data and widely used in many applications. To train GNNs on large graphs that exceed CPU memory, seve…

cs.LG2024

4DBInfer: A 4D Benchmarking Toolbox for Graph-Centric Predictive Modeling on Relational DBs

Minjie Wang, Quan Gan, David Wipf +17

Although RDBs store vast amounts of rich, informative data spread across interconnected tables, the progress of predictive machine learning models as applied to such tasks arguably…

cs.LG2023

MuseGNN: Forming Scalable, Convergent GNN Layers that Minimize a Sampling-Based Energy

Haitian Jiang, Renjie Liu, Zengfeng Huang +5

Among the many variants of graph neural network (GNN) architectures capable of modeling data with cross-instance relations, an important subclass involves layers designed such that…

cs.LG2022

DGI: Easy and Efficient Inference for GNNs

Peiqi Yin, Xiao Yan, Jinjing Zhou +5

While many systems have been developed to train Graph Neural Networks (GNNs), efficient model inference and evaluation remain to be addressed. For instance, using the widely adopte…

cs.DC2020★ 45 cited

TensorOpt: Exploring the Tradeoffs in Distributed DNN Training with Auto-Parallelism

Zhenkun Cai, Kaihao Ma, Xiao Yan +5

A good parallelization strategy can significantly improve the efficiency or reduce the cost for the distributed training of deep neural networks (DNNs). Recently, several methods h…

cs.DC2018

Scalable De Novo Genome Assembly Using Pregel

Da Yan, Hongzhi Chen, James Cheng +2

De novo genome assembly is the process of stitching short DNA sequences to generate longer DNA sequences, without using any reference sequence for alignment. It enables high-throug…