← Search

Shaohuai Shi

11 accepted papers

2026

AGoQ: Activation and Gradient Quantization for Memory-Efficient Distributed Training of LLMs

ICML 2026poster

Quantization is a key method for reducing the GPU memory requirement of training large language models (LLMs). Yet, current approaches are ineffective for 4-bit activations and 8-bit gradients, which would easily cause slow convergence or accuracy loss. To address this, we introduce AGoQ, incorporat…

Cited by 0SourceScholar
2026

Identifying and Mitigating Errors in Gradient Aggregation of Distributed Data Parallel Training

ICML 2026poster

Hardware-related silent data corruptions during gradient aggregation pose significant challenges to fault-tolerant distributed training, often leading to slow or failed convergence. To address this, we first mathematically formulate these errors as gradient inconsistency and theoretically analyze ho…

Cited by 0SourceScholar
2026

PipeDiT: Accelerating Diffusion Transformers in Video Generation with Task Pipelining and Model Decoupling

AAAI 2026technical

Video generation has been advancing rapidly, and diffusion transformer (DiT) based models have demonstrated remarkable capabilities. However, their practical deployment is often hindered by slow inference speeds and high memory consumption. In this paper, we propose a novel pipelining framework name

Cited by 0SourcePDFScholar
2026

Reasoning Language Model Inference Serving Unveiled: An Empirical Study

ICLR 2026poster

The reasoning large language model (RLLM) has been proven competitive in solving complex reasoning tasks such as mathematics, coding, compared to traditional LLM. However, the serving performance and behavior of RLLM remains \textit{unexplored}, which may undermine the deployment and utilization of…

Cited by 0SourcecodeScholar
2026

SALR: Sparsity-Aware Low-Rank Representation for Efficient Fine-Tuning of Large Language Models

AAAI 2026technical

Adapting large pre-trained language models to downstream tasks often entails fine-tuning millions of parameters or deploying costly dense weight updates, which hinders their use in resource-constrained environments. Low-rank Adaptation (LoRA) reduces trainable parameters by factorizing weight update

Cited by 0SourcePDFScholar
2024

FedImpro: Measuring and Improving Client Update in Federated Learning

ICLR 2024poster

Federated Learning (FL) models often experience client drift caused by heterogeneous data, where the distribution of data differs across clients. To address this issue, advanced research primarily focuses on manipulating the existing gradients to achieve more consistent client models. In this paper,…

Cited by 21SourcePDFScholar
2022

EASNet: Searching Elastic and Accurate Network Architecture for Stereo Matching

ECCV 2022poster

"Recent advanced studies have spent considerable human efforts on optimizing network architectures for stereo matching but hardly achieved both high accuracy and fast inference speed. To ease the workload in network design, neural architecture search (NAS) has been applied with great success to vari…

2022

Virtual Homogeneity Learning: Defending against Data Heterogeneity in Federated Learning

ICML 2022spotlight

In federated learning (FL), model performance typically suffers from client drift induced by data heterogeneity, and mainstream works focus on correcting client drift. We propose a different approach named virtual homogeneity learning (VHL) to directly “rectify” the data heterogeneity. In particular…

2021

Automated Model Design and Benchmarking of Deep Learning Models for COVID-19 Detection with Chest CT Scans

AAAI 2021technical

The COVID-19 pandemic has spread globally for several months. Because its transmissibility and high pathogenicity seriously threaten people's lives, it is crucial to accurately and quickly detect COVID-19 infection. Many recent studies have shown that deep learning (DL) based solutions can help dete…

2020

FADNet: A Fast and Accurate Network for Disparity Estimation

ICRA 2020poster

Deep neural networks (DNNs) have achieved great success in the area of computer vision. The disparity estimation problem tends to be addressed by DNNs which achieve much better prediction accuracy in stereo matching than traditional hand-crafted feature based methods. On one hand, however, the desig…

Cited by 100SourcecodeScholar