Publications

Preprint

  1. Distributed Graph Neural Network Training with Periodic Historical Embedding Synchronization. Zheng Chai, Guangji Bai, Liang Zhao, Yue Cheng. arXiv preprint arXiv:2206.00057 (2022)
  2. Towards Quantized Model Parallelism for Graph-Augmented MLPs Based on Gradient-Free ADMM framework. Junxiang Wang, Hongyi Li, Zheng Chai, Yongchao Wang, Yue Cheng, Liang Zhao. arXiv preprint arXiv:2105.09837 (2021)

Published

  1. FedAT: A Communication-Efficient Federated Learning Method with Asynchronous Tiers under Non-IID Data. Zheng Chai, Yujing Chen, Ali Anwar, Liang Zhao, Yue Cheng, Huzefa Rangwala. The International Conference for High Performance Computing, Networking, Storage, and Analysis (SC 2021).
  2. Toward Model Parallelism for Deep Neural Network basedon Gradient-free ADMM Framework.20th IEEE Internation. Junxiang Wang, Zheng Chai, Yue Cheng, Liang Zhao. 20th IEEE International Conference on Data Mining (ICDM). IEEE, 2020.
  3. Federated Multi-task Hierarchical Attention Model forSensor Analytics. Yujing Chen, Yue Ning, Zheng Chai, Huzefa Rangwala. In 2020 International Joint Conference on Neural Networks (IJCNN). IEEE, 2020.
  4. Tunable Subnetwork Splitting for Model-parallelism of Neural Network Training. Junxiang Wang, Zheng Chai, Yue Cheng, Liang Zhao. Workshop on ”Beyond first-order methods in ML systems” of the 37 th International Conference on Machine Learning(PMLR). 2020.
  5. TiFL: A Tier-Based Federated Learning System. Zheng Chai, Ahsan Ali, Syed Zawad, Stacey Truex, Ali Anwar, Nathalie Baracaldo, Yi Zhou, Heiko Ludwig, Feng Yan, Yue Cheng. In Proceedings of the 29th International Symposium on High-Performance Parallel and Distributed Computing(HPDC). ACM, 2020.
  6. Towards taming the resource and data heterogeneity in federated learning. Zheng Chai,Hannan Fayyaz, Zeshan Fayyaz, Ali Anwar, Yi Zhou, Nathalie Baracaldo, Heiko Ludwig, Yue Cheng. Conference on Operational Machine Learning (OpML 19). USENIX, 2019.
  7. Characterizing co-located datacenter workloads: An alibaba case study. Yue Cheng, Zheng Chai, Ali Anwar. Proceedings of the 9th Asia-Pacific Workshop on Systems (APSys). ACM, 2018.