Zexi Li 0001

dblp:151/9187-1 · DBLP profile ↗
← Back
5ranked-venue papers in the field
2as first author
5since 2021 · last 2026
0000-0003-0831-3549ORCID · conflict

Domains — venue-derived; a paper can count in several

Data Mining & Knowledge Discovery · 4 (1 first)Information Retrieval & Web Search · 1 (1 first)
YearPublicationVenuePosition
2026 Each Rank Could be an Expert: Single-Ranked Mixture of Experts LoRA for Multi-task Learning
abstract
Low-Rank Adaptation (LoRA) is widely used for adapting large language models (LLMs) to specific domains due to its efficiency and modularity. However, vanilla LoRA struggles with task conflicts in multi-task scenarios. Recent works adopt Mixture of Experts (MoE) by treating each LoRA module as an expert, thereby mitigating task interference through multiple specialized LoRA modules. While effective, these methods often isolate knowledge within individual tasks, failing to fully exploit the shared knowledge across related tasks. In this paper, we establish a connection between single LoRA and multi-LoRA MoE, integrating them into a unified framework. We demonstrate that the dynamic routing of multiple LoRAs is functionally equivalent to rank partitioning and block-level activation within a single LoRA. To systematically study the role of expert granularity in multi-task learning, we conduct an in-depth investigation within our unified framework. Our empirical results show that a finer-grained expert partitioning not only yields significant performance gains but also captures more diverse parameter patterns. These empirical findings are supported by our theoretical analysis, which proves that finer granularity expands parameter space diversity and tightens the model's error bound. Building on these findings, we propose Single-ranked Mixture of Experts LoRA (SMoRA ), which embeds MoE into LoRA by treating each rank as an independent expert. With a dynamic rank-wise activation mechanism, SMoRA facilitates a flexible composition of knowledge, enabling the model to learn deeper and more diverse features while mitigating task conflicts. Experiments demonstrate that SMoRA activates fewer parameters yet achieves better performance in multi-task scenarios.
Ziyu Zhao 0001, Yixiao Zhou 0001, Zhi Zhang 0005, Didi Zhu, Tao Shen 0002, Zexi Li 0001, Jinluan Yang, Xuwu Wang, Jing Su 0005, Kun Kuang 0001, Zhongyu Wei, Fei Wu 0001, Yu Cheng 0001
KDD (1)7
2025 FedGuCci: Making Local Models More Connected in Landscape for Federated Learning
abstract
Federated learning (FL) involves multiple heterogeneous clients collaboratively training a global model via iterative local updates and model fusion.The generalization of FL's global model has a large gap compared with centralized training, which is its bottleneck for broader applications.In this paper, we study and improve FL's generalization through a fundamental "connectivity" perspective, which means how the local models are connected in the parameter region and fused into a generalized global model.The term "connectivity" is derived from linear mode connectivity (LMC), studying the interpolated loss landscape of two different solutions (e.g., modes) of neural networks.Bridging the gap between LMC and FL, in this paper, we leverage fixed anchor models to empirically and theoretically study the transitivity property of connectivity from two models (LMC) to a group of models (model fusion in FL).Based on the findings, we propose FedGuCci(+), improving group connectivity for better generalization.It is shown that our methods can boost the generalization of FL under client heterogeneity across various tasks (4 CV datasets and 6 NLP datasets) and model architectures (e.g., ViTs and PLMs).The code is available here: FedGuCci Codebase.
Zexi Li 0001, Zhiqi Li 0004, Didi Zhu, Tao Shen 0002, Tao Lin 0004, Chao Wu 0001, Nicholas D. Lane
KDD (2)1
2024 OpenFedLLM: Training Large Language Models on Decentralized Private Data via Federated Learning
abstract
Trained on massive publicly available data, large language models (LLMs) have demonstrated tremendous success across various fields.While more data contributes to better performance, a disconcerting reality is that high-quality public data will be exhausted in a few * Siheng Chen is the corresponding author.
Rui Ye 0001, Wenhao Wang 0002, Jingyi Chai, Dihan Li, Zexi Li 0001, Yinda Xu, Yaxin Du, Yanfeng Wang 0001, Siheng Chen
KDD5
2024 Neural Collapse Anchored Prompt Tuning for Generalizable Vision-Language Models
abstract
Large-scale vision-language (V-L) models have demonstrated remarkable generalization capabilities for downstream tasks through prompt tuning. However, the mechanisms behind the learned text representations are unknown, limiting further generalization gains, and the limitations are more severe when faced with the prevalent class imbalances seen in web-sourced datasets. Recent advances in the neural collapse (NC) phenomenon of vision-only models suggest that the optimal representation structure is the simplex ETF, which paves the way to study representations in V-L models. In this paper, we make the first attempt to use NC for examining the representations in V-L models via prompt tuning. It is found that NC optimality of text-to-image representations shows a positive correlation with downstream generalizability, which is more severe under class imbalance settings. To improve the representations, we propose Neural-collapse-anchored Prompt Tuning (NPT), a novel method that learns prompts with text and image representations that satisfy the same simplex Equiangular Tight Frame (ETF). NPT incorporates two regularization terms: language-modality collapse and multi-modality isomorphism; and it is compatible with other prompt tuning methods. Extensive experiments show that NPT can consistently help to improve existing prompt tuning techniques across 11 datasets for both balanced and imbalanced settings.
Didi Zhu, Zexi Li 0001, Min Zhang 0068, Junkun Yuan, Kun Kuang 0001, Chao Wu 0001
KDD2
2023 Edge-cloud Collaborative Learning with Federated and Centralized Features
abstract
Federated learning (FL) is a popular way of edge computing that does not compromise user's privacy. Current FL paradigms assume data only resides on the edge, while cloud servers only perform model averaging. However, in real-life situations such as recommender systems, the cloud server usually has abundant features and computation resources. Specifically, the cloud stores historical and interactive features, and the edge stores privacy-sensitive and real-time features. In this paper, our proposed Edge-Cloud Collaborative Knowledge Transfer Framework (ECCT) jointly utilizes the edge-side features and the cloud-side features, enabling bi-directional knowledge transfer between the two by sharing feature embeddings and prediction logits. ECCT consolidates various benefits, including enhancing personalization, enabling model heterogeneity, tolerating training asynchronization, and relieving communication burdens. Extensive experiments on public and industrial datasets demonstrate the effectiveness of ECCT.
Zexi Li 0001, Qunwei Li, Yi Zhou 0017, Leon Wenliang Zhong, Chao Wu 0001
SIGIR1