activity
20202026
most citedFedML: A Research Library and Benchmark for Federated Machine Learning

354 citations · 634 across the 70 of their papers we have counts for

collaborators
Showing cs.LGShow all

50 papers · 1 filter

cs.LG2026

LLM Evolution as an Industry-Scale Ecosystem: A Lifecycle Perspective on Continual Learning

Hao Jiang, Enneng Yang, Guojie Zhu +7

Continual learning capability is critical for Industrial LLMs, as deployed models must be continuously updated to meet evolving requirements and environments, rather than repeatedl…

cs.LG2025

Network Sparsity Unlocks the Scaling Potential of Deep Reinforcement Learning

Guozheng Ma, Lu Li, Zilin Wang +3

Effectively scaling up deep reinforcement learning models has proven notoriously difficult due to network pathologies during training, motivating various targeted interventions suc…

cs.LG2025

Safety Reasoning with Guidelines

Haoyu Wang, Zeyu Qin, Li Shen +3

Training safe LLMs remains a critical challenge. The most widely used method, Refusal Training (RT), struggles to generalize against various Out-of-Distribution (OOD) jailbreaking…

cs.LG2025

TeZO: Empowering the Low-Rankness on the Temporal Dimension in the Zeroth-Order Optimization for Fine-tuning LLMs

Yan Sun, Tiansheng Huang, Liang Ding +2

Zeroth-order optimization (ZO) has demonstrated remarkable promise in efficient fine-tuning tasks for Large Language Models (LLMs). In particular, recent advances incorporate the l…

cs.LG2024

Communication Learning in Multi-Agent Systems from Graph Modeling Perspective

Shengchao Hu, Li Shen, Ya Zhang +1

In numerous artificial intelligence applications, the collaborative efforts of multiple intelligent agents are imperative for the successful attainment of target objectives. To enh…

cs.LG2024

A-FedPD: Aligning Dual-Drift is All Federated Primal-Dual Learning Needs

Yan Sun, Li Shen, Dacheng Tao

As a popular paradigm for juggling data privacy and collaborative training, federated learning (FL) is flourishing to distributively process the large scale of heterogeneous datase…

cs.LG2024

Convergent Differential Privacy Analysis for General Federated Learning

Yan Sun, Qixin Zhang, Li Shen +1

The powerful cooperation of federated learning (FL) and differential privacy~(DP) provides a promising paradigm for the large-scale private clients. However, existing analyses in F…

cs.LG2024★ 4 cited

Learning Multi-Agent Communication from Graph Modeling Perspective

Shengchao Hu, Li Shen, Ya Zhang +1

In numerous artificial intelligence applications, the collaborative efforts of multiple intelligent agents are imperative for the successful attainment of target objectives. To enh…

cs.LG2024

Communication-Efficient Distributed Learning with Local Immediate Error Compensation

Yifei Cheng, Li Shen, Linli Xu +6

Gradient compression with error compensation has attracted significant attention with the target of reducing the heavy communication overhead in distributed learning. However, exis…

cs.LG2024★ 1 cited

Representation Surgery for Multi-Task Model Merging

Enneng Yang, Li Shen, Zhenyi Wang +4

Multi-task learning (MTL) compresses the information from multiple tasks into a unified backbone to improve computational efficiency and generalization. Recent work directly merges…

cs.LG2024

Solving Continual Offline Reinforcement Learning with Decision Transformer

Kaixin Huang, Li Shen, Chen Zhao +2

Continuous offline reinforcement learning (CORL) combines continuous and offline reinforcement learning, enabling agents to learn multiple tasks from static datasets without forget…

cs.LG2023★ 2 cited

Efficient Federated Prompt Tuning for Black-box Large Pre-trained Models

Zihao Lin, Yan Sun, Yifan Shi +4

With the blowout development of pre-trained models (PTMs), the efficient tuning of these models for diverse downstream applications has emerged as a pivotal research concern. Altho…

cs.LG2023★ 1 cited

Revisiting Plasticity in Visual Reinforcement Learning: Data, Modules and Training Stages

Guozheng Ma, Lu Li, Sen Zhang +6

Plasticity, the ability of a neural network to evolve with new data, is crucial for high-performance and sample-efficient visual reinforcement learning (VRL). Although methods like…

cs.LG2023

Towards Understanding Generalization and Stability Gaps between Centralized and Decentralized Federated Learning

Yan Sun, Li Shen, Dacheng Tao

As two mainstream frameworks in federated learning (FL), both centralized and decentralized approaches have shown great application value in practical scenarios. However, existing…

cs.LG2023★ 5 cited

AdaMerging: Adaptive Model Merging for Multi-Task Learning

Enneng Yang, Zhenyi Wang, Li Shen +4

Multi-task learning (MTL) aims to empower a model to tackle multiple tasks simultaneously. A recent development known as task arithmetic has revealed that several models, each fine…

cs.LG2023★ 1 cited

FedLALR: Client-Specific Adaptive Learning Rates Achieve Linear Speedup for Non-IID Data

Hao Sun, Li Shen, Shixiang Chen +4

Federated learning is an emerging distributed machine learning method, enables a large number of clients to train a model without exchanging their local data. The time cost of comm…

cs.LG2023★ 1 cited

DFedADMM: Dual Constraints Controlled Model Inconsistency for Decentralized Federated Learning

Qinglun Li, Li Shen, Guanghao Li +2

To address the communication burden issues associated with federated learning (FL), decentralized federated learning (DFL) discards the central server and establishes a decentraliz…

cs.LG2023★ 2 cited

Towards Understanding the Generalizability of Delayed Stochastic Gradient Descent

Xiaoge Deng, Li Shen, Shengwei Li +3

Stochastic gradient descent (SGD) performed in an asynchronous manner plays a crucial role in training large-scale machine learning models. However, the generalization performance…

cs.LG2023

Efficient Federated Learning via Local Adaptive Amended Optimizer with Linear Speedup

Yan Sun, Li Shen, Hao Sun +2

Adaptive optimization has achieved notable success for distributed learning while extending adaptive optimizer to federated Learning (FL) suffers from severe inefficiency, includin…

cs.LG2023★ 3 cited

Understanding How Consistency Works in Federated Learning via Stage-wise Relaxed Initialization

Yan Sun, Li Shen, Dacheng Tao

Federated learning (FL) is a distributed paradigm that coordinates massive local clients to collaboratively train a global model via stage-wise local training processes on the hete…

cs.LG2023★ 1 cited

Instructed Diffuser with Temporal Condition Guidance for Offline Reinforcement Learning

Jifeng Hu, Yanchao Sun, Sili Huang +6

Recent works have shown the potential of diffusion models in computer vision and natural language processing. Apart from the classical supervised learning fields, diffusion models…

cs.LG2023★ 3 cited

Towards More Suitable Personalization in Federated Learning via Decentralized Partial Model Training

Yifan Shi, Yingqi Liu, Yan Sun +4

Personalized federated learning (PFL) aims to produce the greatest personalized model for each client to face an insurmountable problem--data heterogeneity in real FL systems. Howe…

cs.LG2023

Prompt-Tuning Decision Transformer with Preference Ranking

Shengchao Hu, Li Shen, Ya Zhang +1

Prompt-tuning has emerged as a promising method for adapting pre-trained models to downstream tasks or aligning with human preferences. Prompt learning is widely used in NLP but ha…

cs.LG2023★ 1 cited

Towards the Flatter Landscape and Better Generalization in Federated Learning under Client-level Differential Privacy

Yifan Shi, Kang Wei, Li Shen +4

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 f…