← Search

Xiaonan Nie

3 accepted papers

2025

NetMoE: Accelerating MoE Training through Dynamic Sample Placement

ICLR 2025spotlight

Mixture of Experts (MoE) is a widely used technique to expand model sizes for better model quality while maintaining the computation cost constant. In a nutshell, an MoE model consists of multiple experts in each model layer and routes the training tokens to only a fixed number of experts rather tha…

Cited by 1SourcePDFScholar
2024

LSH-MoE: Communication-efficient MoE Training via Locality-Sensitive Hashing

NeurIPS 2024poster

Larger transformer models perform better on various downstream tasks but require more cost to scale up the model size. To efficiently enlarge models, the Mixture-of-Expert (MoE) architecture is widely adopted, which consists of a gate network and a series of experts and keep the training cost consta…

Cited by 0SourcePDFScholar
2023

OSDP: Optimal Sharded Data Parallel for Distributed Deep Learning

IJCAI 2023poster

Large-scale deep learning models contribute to significant performance improvements on varieties of downstream tasks. Current data and model parallelism approaches utilize model replication and partition techniques to support the distributed training of ultra-large models. However, directly deployin…