Uncertainty-aware Attention Graph Neural Network for Defending Adversarial Attacks. - Boyuan Feng; Yuke Wang; Zheng Wang 075; Yufei Ding 001 (CoRR, 2020)

EL-Rec: Efficient Large-Scale Recommendation Model Training via Tensor-Train Embedding Table. - Zheng Wang 075; Yuke Wang; Boyuan Feng; Dheevatsa Mudigere; Bharath Muthiah; Yufei Ding 001 (SC, 2022)

Faith: An Efficient Framework for Transformer Verification on GPUs. - Boyuan Feng; Tianqi Tang 001; Yuke Wang; Zhaodong Chen; Zheng Wang 075; Shu Yang; Yuan Xie 001; Yufei Ding 001 (USENIX ATC, 2022)

GMI-DRL: Empowering Multi-GPU Deep Reinforcement Learning with GPU Spatial Multiplexing. - Yuke Wang; Boyuan Feng; Zheng Wang 075; Tong Geng; Ang Li 006; Yufei Ding 001 (CoRR, 2022)

Empowering GNNs with Fine-grained Communication-Computation Pipelining on Multi-GPU Platforms. - Yuke Wang; Boyuan Feng; Zheng Wang 075; Tong Geng; Kevin J. Barker; Ang Li 006; Yufei Ding 001 (CoRR, 2022)

ECSSD: Hardware/Data Layout Co-Designed In-Storage-Computing Architecture for Extreme Classification. - Siqi Li; Fengbin Tu; Liu Liu 017; Jilan Lin; Zheng Wang 075; Yangwook Kang; Yufei Ding 001; Yuan Xie 001 (ISCA, 2023)

MGG: Accelerating Graph Neural Networks with Fine-Grained Intra-Kernel Communication-Computation Pipelining on Multi-GPU Platforms. - Yuke Wang; Boyuan Feng; Zheng Wang 075; Tong Geng; Kevin J. Barker; Ang Li 006; Yufei Ding 001 (OSDI, 2023)

TC-GNN: Bridging Sparse GNN Computation and Dense Tensor Cores on GPUs. - Yuke Wang; Boyuan Feng; Zheng Wang 075; Guyue Huang; Yufei Ding 001 (USENIX ATC, 2023)

ZENO: A Type-based Optimization Framework for Zero Knowledge Neural Network Inference. - Boyuan Feng; Zheng Wang 075; Yuke Wang; Shu Yang; Yufei Ding 001 (ASPLOS (1), 2024)

RAP: Resource-aware Automated GPU Sharing for Multi-GPU Recommendation Model Training and Input Preprocessing. - Zheng Wang 075; Yuke Wang; Jiaqi Deng; Da Zheng; Ang Li 006; Yufei Ding 001 (ASPLOS (2), 2024)

OPER: Optimality-Guided Embedding Table Parallelization for Large-scale Recommendation Model. - Zheng Wang 075; Yuke Wang; Boyuan Feng; Guyue Huang; Dheevatsa Mudigere; Bharath Muthiah; Ang Li 006; Yufei Ding 001 (USENIX ATC, 2024)

WLB-LLM: Workload-Balanced 4D Parallelism for Large Language Model Training. - Zheng Wang 075; Anna Cai; Xinfeng Xie; Zaifeng Pan; Yue Guan; Weiwei Chu; Jie Wang 022; Shikai Li; Jianyu Huang; Chris Cai; Yuchen Hao; Yufei Ding 001 (OSDI, 2025)

GMI-DRL: Empowering Multi-GPU DRL with Adaptive-Grained Parallelism. - Yuke Wang; Boyuan Feng; Zheng Wang 075; Guyue Huang; Tony Tong Geng; Ang Li 006; Yufei Ding 001 (USENIX ATC, 2025)