Speculate, then Collaborate: Fusing Knowledge of Language Models during Decoding
Ziyao Wang, Muneeza Azmat, Ang Li, Raya Horesh, Mikhail Yurochkin.
ICML 25

FLoRA: Federated Fine-Tuning Large Language Models with Heterogeneous Low-Rank Adaptations
Ziyao Wang, Zheyu Shen, Yexiao He, Guoheng Sun, Hongyi Wang, Lingjuan Lyu, Ang Li.
NeurIPS 24

FedHyper: A Universal and Robust Learning Rate Scheduler for Federated Learning with Hypergradient Descent
Ziyao Wang, Jianyu Wang, Ang Li.
ICLR 24

UPTON: Unattributable Authorship Text via Data Poisoning
Ziyao Wang, Dongwon Lee, Thai Le.
EMNLP 23

Prada: Black-Box LLM Adaptation with Private Data on Resource-Constrained Devices
Ziyao Wang, Yexiao He, Zheyu Shen, Yu Li, Guoheng Sun, Myungjin Lee, Ang Li.
arXiv: 2503.14932

One Communication Round is All It Needs for Federated Fine-Tuning Foundation Models
Ziyao Wang, Bowei Tian, Yexiao He, Zheyu Shen, Luyang Liu, Ang Li.
arXiv: 2412.04650

Predictive Auditing of Hidden Tokens in LLM APIs via Reasoning Length Estimation
Ziyao Wang, Guoheng Sun, Yexiao He, Zheyu Shen, Bowei Tian, Ang Li.
arXiv: 2508.00912

Invisible Tokens, Visible Bills: The Urgent Need to Audit Hidden Operations in Opaque LLM Services
Guoheng Sun*, Ziyao Wang*, Xuandong Zhao, Bowei Tian, Zheyu Shen, Yexiao He, Jinming Xing, Ang Li.
arXiv: 2505.18471

CoIn: Counting the Invisible Reasoning Tokens in Commercial Opaque LLM APIs
Guoheng Sun, Ziyao Wang, Bowei Tian, Meng Liu, Zheyu Shen, Shwai He, Yexiao He, Wanghao Ye, Yiting Wang, Ang Li.
arXiv: 2505.13778

Towards counterfactual fairness through auxiliary variables
Bowei Tian, Ziyao Wang, Shwai He, Wanghao Ye, Guoheng Sun, Yucong Dai, Yongkai Wu, Ang Li.
ICLR 25

SHED: Shapley-Based Automated Dataset Refinement for Instruction Fine-Tuning
Yexiao He, Ziyao Wang, Zheyu Shen, Guoheng Sun, Yucong Dai, Yongkai Wu, Hongyi Wang, Ang Li.
NeurIPS 24

NetGuard: Protecting Commercial Web APIs from Model Inversion Attacks using GAN-generated Fake Samples
Xueluan Gong, Ziyao Wang, Yanjiao Chen, Qian Wang, Cong Wang, Chao Shen.
WWW 23

EdgeLoRA: An Efficient Multi-Tenant LLM Serving System on Edge Devices
Zheyu Shen, Yexiao He, Ziyao Wang, Yuning Zhang, Guoheng Sun, Wanghao Ye, Ang Li.
MobiSys 25

Fair Diagnosis: Leveraging Causal Modeling to Mitigate Medical Bias
Bowei Tian, Ziyao Wang, Meng Liu, Yucong Dai, Shwai He, Guoheng Sun, Zheyu Shen, Wanghao Ye, Yongkai Wu, Ang Li.
arXiv: 2412.04739

Moderator: Moderating Text-to-Image Diffusion Models through Fine-grained Context-based Policies
Peiran Wang, Qiyu Li, Longxuan Yu, Ziyao Wang, Ang Li, Haojian Jin.
CCS 24 (Distinguished Paper Award)

Beowulf: Mitigating Model Extraction Attacks Via Reshaping Decision Regions
Xueluan Gong, Rubin Wei, Ziyao Wang, Yuchen Sun, Jiawen Peng, Yanjiao Chen, Qian Wang.
CCS 24

A GAN-based Defense Framework against Model Inversion Attacks
Xueluan Gong*, Ziyao Wang*, Yanjiao Chen, Qian Wang.
TIFS

Kaleidoscope: Triggering Backdoor with RGB Filters
Xueluan Gong*, Ziyao Wang*, Yuzhe Gu, Yanjiao Chen, Qian Wang.
TDSC

KERBNET: A QoE-aware Kernel-Based Backdoor Attack Framework
Xueluan Gong, Yanjiao Chen, Huayang Huang, Weihan Kong, Ziyao Wang, Qian Wang.
TDSC