collaborators

6 papers

cs.LG202424 cited

MegaScale: Scaling Large Language Model Training to More Than 10,000 GPUs

Ziheng Jiang, Haibin Lin, Yinmin Zhong +29

We present the design, implementation and engineering experience in building and deploying MegaScale, a production system for training large language models (LLMs) at the scale of…

eess.SP2023

Knowledge-driven Meta-learning for CSI Feedback

Han Xiao, Wenqiang Tian, Wendong Liu +6

Accurate and effective channel state information (CSI) feedback is a key technology for massive multiple-input and multiple-output systems. Recently, deep learning (DL) has been in…

cs.CR20232 cited

DeepTheft: Stealing DNN Model Architectures through Power Side Channel

Yansong Gao, Huming Qiu, Zhi Zhang +6

Deep Neural Network (DNN) models are often deployed in resource-sharing clouds as Machine Learning as a Service (MLaaS) to provide inference services.To steal model architectures t…

cs.CR2023

SemDiff: Binary Similarity Detection by Diffing Key-Semantics Graphs

Zian Liu, Zhi Zhang, Siqi Ma +6

Binary similarity detection is a critical technique that has been applied in many real-world scenarios where source code is not available, e.g., bug search, malware analysis, and c…

eess.SP2023

A Knowledge-Driven Meta-Learning Method for CSI Feedback

Han Xiao, Wenqiang Tian, Wendong Liu +4

Accurate and effective channel state information (CSI) feedback is a key technology for massive multiple-input and multiple-output (MIMO) systems. Recently, deep learning (DL) has…

cs.CR2022

MUD-PQFed: Towards Malicious User Detection in Privacy-Preserving Quantized Federated Learning

Hua Ma, Qun Li, Yifeng Zheng +5

Federated Learning (FL), a distributed machine learning paradigm, has been adapted to mitigate privacy concerns for customers. Despite their appeal, there are various inference att…