HAFLO: GPU-Based Acceleration for Federated Logistic Regression. - Xiaodian Cheng; Wanhang Lu; Xinyang Huang; Shuihai Hu; Kai Chen 005 (CoRR, 2021)

Herald: An Embedding Scheduler for Distributed Embedding Model Training. - Chaoliang Zeng; Xiaodian Cheng; Han Tian; Hao Wang 116; Kai Chen 005 (APNet, 2022)

Communication Efficient Secret Sharing with Dynamic Communication-Computation Conversion. - Zhenghang Ren; Xiaodian Cheng; Mingxuan Fan; Junxue Zhang 001; Cheng Hong 001 (INFOCOM, 2023)

FLASH: Towards a High-performance Hardware Acceleration Architecture for Cross-silo Federated Learning. - Junxue Zhang 001; Xiaodian Cheng; Wei Wang; Liu Yang 008; Jinbin Hu; Kai Chen 005 (NSDI, 2023)

Accelerating Privacy-Preserving Machine Learning With GeniBatch. - Xinyang Huang; Junxue Zhang 001; Xiaodian Cheng; Hong Zhang 025; Yilun Jin; Shuihai Hu; Han Tian; Kai Chen 005 (EuroSys, 2024)

Accelerating Neural Recommendation Training with Embedding Scheduling. - Chaoliang Zeng; Xudong Liao; Xiaodian Cheng; Han Tian; Xinchen Wan; Hao Wang 116; Kai Chen 005 (NSDI, 2024)

High-Performance Hardware Acceleration Architecture for Cross-Silo Federated Learning. - Junxue Zhang 001; Xiaodian Cheng; Liu Yang 008; Jinbin Hu; Han Tian; Kai Chen 005 (IEEE Trans. Parallel Distributed Syst., 2024)

SoK: Fully Homomorphic Encryption Accelerators. - Junxue Zhang 001; Xiaodian Cheng; Liu Yang 008; Jinbin Hu; Ximeng Liu; Kai Chen 005 (ACM Comput. Surv., 2024)

FLASH-FHE: A Heterogeneous Architecture for Fully Homomorphic Encryption Acceleration. - Junxue Zhang 001; Xiaodian Cheng; Gang Cao; Meng Dai; Yijun Sun; Han Tian; Dian Shen; Yong Wang 046; Kai Chen 005 (CoRR, 2025)