← Search

Kang Wei

6 accepted papers

2026

Rethinking Convergence in MoE Training: The Role of Routing Sparsity

ICML 2026poster

In Mixture-of-Experts (MoE) training, sparse routing, i.e., activating only the top-$K$ experts per token, is essential for balancing convergence speed and computational cost. However, existing works typically choose $K$ empirically, without theoretical guidance. To address this gap, we characterize…

Cited by 0SourceScholar
2026

The Hidden Risk: Membership Inference Attacks on Multimodal Federated Learning via Modality Imbalance

ICML 2026poster

Federated learning (FL) faces significant challenges from modality heterogeneity, which motivates multimodal federated learning (MFL) to leverage complementary modalities across decentralized clients for improved performance. However, modality imbalance introduces a new attack surface, making MFL mo…

Cited by 0SourceScholar
2024

Dual Expert Distillation Network for Generalized Zero-Shot Learning

IJCAI 2024poster

Zero-shot learning has consistently yielded remarkable progress via modeling nuanced one-to-one visual-attribute correlation. Existing studies resort to refining a uniform mapping function to align and correlate the sample regions and subattributes, ignoring two crucial issues: 1) the inherent asymm…

2023

Improving the Model Consistency of Decentralized Federated Learning

ICML 2023poster

To mitigate the privacy leakages and communication burdens of Federated Learning (FL), decentralized FL (DFL) discards the central server and each client only communicates with its neighbors in a decentralized communication network. However, existing DFL suffers from high inconsistency among local c…

Cited by 68SourcePDFScholar
2023

Make Landscape Flatter in Differentially Private Federated Learning

CVPR 2023poster

To defend the inference attacks and mitigate the sensitive information leakages in Federated Learning (FL), client-level Differentially Private FL (DPFL) is the de-facto standard for privacy protection by clipping local updates and adding random noise. However, existing DPFL methods tend to make a s…