Enhancing Token Filtering Efficiency in Large Language Model Training with Collider
Di Chai, Pengbo Li, Feiyuan Zhang, Yilun Jin, Han Tian, Junxue Zhang#, Kai Chen
arXiv:2502.00340 [cs.LG], 2025
paper | arXiv| Back to Index | Google Scholar | DBLP | Semantic Scholar | ORCID | arXiv Author Page |
The publications are in a chronological order.
* indicates equal contribution and # indicates corresponding author.
| Show All | Conference Papers | Journal Papers | Workshop Papers | Preprints |
CEIO: A Cache-Efficient Network I/O Architecture for NIC-CPU Data Paths
Bowen Liu, Xinyang Huang, Qijing Li, Zhuobin Huang, Yijun Sun, Wenxue Li, Junxue Zhang, Ping Yin, Kai Chen
In Proceedings of the ACM SIGCOMM 2025 Conference, 2025
(To appear)Revisiting RDMA Reliability for Lossy Fabrics
Wenxue Li, Xiangzhou Liu, Yunxuan Zhang, Zihao Wang, Wei Gu, Tao Qian, Gaoxiong Zeng, Shoushou Ren, Xinyang Huang, Zhenghang Ren, Bowen Liu, Junxue Zhang, Bingyang Liu, Kai Chen
In Proceedings of the ACM SIGCOMM 2025 Conference, 2025
(To appear)Enabling Efficient GPU Communication over Multiple NICs with FuseLink
Zhenghang Ren, Yuxuan Li, Zilong Wang, Xinyang Huang, Wenxue Li, Kaiqiang Xu, Xudong Liao, Yijun Sun, Bowen Liu, Han Tian, Junxue Zhang, Mingfei Wang, Zhizhen Zhong, Guyue Liu, Ying Zhang, Kai Chen
In Proceedings of the 19th USENIX Symposium on Operating Systems Design and Implementation, 2025
(To appear)Towards Optimal Rack-scale μs-level CPU Scheduling through In-Network Workload Shaping
Xudong Liao, Han Tian, Xinchen Wan, Chaoliang Zeng, Hao Wang, Junxue Zhang, Mengyu Ma, Guyue Liu, Kai Chen
In Proceedings of the USENIX Annual Technical Conference, 2025
(To appear)Accelerating Distributed Graph Learning by Using Collaborative In-Network Multicast and Aggregation
Zhaoyi Li, Jiawei Huang, Yijun Li, Jingling Liu, Junxue Zhang, Hui Li, Xiaojun Zhu, Shengwen Zhou, Jing Shao, Xiaojuan Lu, Qichen Su, Jianxin Wang, Chee Wei Tan, Yong Cui, Kai Chen
In Proceedings of the USENIX Annual Technical Conference, 2025
(To appear)GREEN: Carbon-efficient Resource Scheduling for Machine Learning Clusters
Kaiqiang Xu, Decang Sun, Han Tian, Junxue Zhang, Kai Chen
In Proceedings of the 22nd USENIX Symposium on Networked Systems Design and Implementation, 2025
paper | usenix pageDesign and Operation of Shared Machine Learning Clusters on Campus
Kaiqiang Xu, Decang Sun, Hao Wang, Zhenghang Ren, Xinchen Wan, Xudong Liao, Zilong Wang, Junxue Zhang, Kai Chen
In Proceedings of the International Conference on Architectural Support for Programming Languages and Operating Systems, 2025
paper | acm dleNetSTL: Towards an In-kernel Library for High-Performance eBPF-based Network Functions
Bin Yang, Dian Shen#, Junxue Zhang#, Hanlin Yang, Lunqi Zhao, Beilun Wang, Guyue Liu, Kai Chen
In Proceedings of the 20th European Conference on Computer Systems, 2025
paper | acm dl | supplementary materials | codeMerged into Coolbpf of OpenAnolis (龙蜥社区)
RhyR: Cache-Aware Rate Control for RDMA I/O Congestion
Qijing Li, Xinyang Huang, Bowen Liu, Pengbo Li, Junxue Zhang, Kai Chen
In Proceedings of the 9th Asia-Pacific Workshop on Networking, 2025
(To appear)Efficient Decentralized Federated Singular Vector Decomposition
Di Chai, Junxue Zhang#, Liu Yang, Yilun Jin, Leye Wang, Kai Chen#, Qiang Yang
In Proceedings of the USENIX Annual Technical Conference, 2024
paper | usenix page | code | dockerPass the Artfact Evaluation with all badges:
Accelerating Secure Collaborative Machine Learning with Protocol-Aware RDMA
Zhenghang Ren, Mingxuan Fan, Zilong Wang, Junxue Zhang, Chaoliang Zeng, Zhicong Huang, Cheng Hong, Kai Chen
In Proceedings of the 32nd USENIX Security Symposium, 2024
paper | usenix pageHigh-performance Hardware Acceleration Architecture for Cross-silo Federated Learning
Junxue Zhang, Xiaodian Cheng, Liu Yang, Jinbin Hu, Han Tian, Kai Chen
IEEE Transactions on Parallel and Distributed Systems, 2024, Volume: 35, Issue: 8
paper | ieee dl | supplementary materialsEnabling Load Balancing for Lossless Datacenters
Jinbin Hu, Chaoliang Zeng, Zilong Wang, Junxue Zhang, Kun Guo, Hong Xu, Jiawei Huang, Kai Chen
In Proceedings of the 31th IEEE International Conference on Network Protocols, 2023 (Best Paper Award)
paper | ieee dl | award certificate | extended versionFLASH: Towards a High-performance Hardware Acceleration Architecture for Cross-silo Federated Learning
Junxue Zhang, Xiaodian Cheng, Wei Wang, Liu Yang, Jinbin Hu, Kai Chen
In Proceedings of the 20th USENIX Symposium on Networked Systems Design and Implementation, 2023
paper | usenix page | slides | video | extended versionTowards the Full Extensibility of Multipath TCP with eMPTCP
Bin Yang, Dian Shen, Junxue Zhang, Fang Dong, Junzhou Luo, John. C.S. Lui
In Proceedings of the 30th IEEE International Conference on Network Protocols, 2022
paper | ieee dl | code | extended versionPractical and Secure Federated Recommendation with Personalized Masks
Liu Yang, Junxue Zhang, Di Chai, Leye Wang, Kuo Guo, Kai Chen, Qiang Yang
In International Workshop on Trustworthy Federated Learning in Conjunction with IJCAI, 2022
paper | lecture notesSecure Forward Aggregation for Vertical Federated Neural Networks
Shuowei Cai, Di Chai, Liu Yang, Junxue Zhang, Yilun Jin, Leye Wang, Kun Guo, Kai Chen
In International Workshop on Trustworthy Federated Learning in Conjunction with IJCAI, 2022
paper | lecture notesAegis: A Trusted, Automatic and Accurate Verification Framework for Vertical Federated Learning
Cengguang Zhang, Junxue Zhang, Di Chai, Kai Chen
In International Workshop on Federated and Transfer Learning for Data Sparsity and Confidentiality in Conjunction with IJCAI, 2021 (Best Application Paper Award)
paper | award certificateEnabling ECN for Datacenter Networks with RTT Variations
Junxue Zhang, Wei Bai, Kai Chen
In Proceedings of the 15th International Conference on emerging Networking EXperiments and Technologies, 2019
paper | acm dl | code | slides | extended versionQuantifying the Performance of Federated Transfer Learning
Qinghe Jing, Weiyang Wang, Junxue Zhang, Han Tian, Kai Chen
International Workshop on Federated Learning for User Privacy and Data Confidentiality in Conjunction with IJCAI, 2019 (Best Student Paper Award)
paper | award certificateBridging the Edge-Cloud Barrier for Real-time Advanced Vision Analytics
Yiding Wang, Weiyang Wang, Junxue Zhang, Junchen Jiang, Kai Chen
In Proceedings of the 11th USENIX Workshop on Hot Topics in Cloud Computing, 2019
paper | usenix page