Improving Learning of New Diseases through Knowledge-Enhanced Initialization for Federated Adapter Tuning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Peng, Danni, Wang, Yuan, Cai, Kangning, Ning, Peiyan, Xu, Jiming, Liu, Yong, Goh, Rick Siow Mong, Wei, Qingsong, Fu, Huazhu
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909736687894528
author Peng, Danni
Wang, Yuan
Cai, Kangning
Ning, Peiyan
Xu, Jiming
Liu, Yong
Goh, Rick Siow Mong
Wei, Qingsong
Fu, Huazhu
author_facet Peng, Danni
Wang, Yuan
Cai, Kangning
Ning, Peiyan
Xu, Jiming
Liu, Yong
Goh, Rick Siow Mong
Wei, Qingsong
Fu, Huazhu
contents In healthcare, federated learning (FL) is a widely adopted framework that enables privacy-preserving collaboration among medical institutions. With large foundation models (FMs) demonstrating impressive capabilities, using FMs in FL through cost-efficient adapter tuning has become a popular approach. Given the rapidly evolving healthcare environment, it is crucial for individual clients to quickly adapt to new tasks or diseases by tuning adapters while drawing upon past experiences. In this work, we introduce Federated Knowledge-Enhanced Initialization (FedKEI), a novel framework that leverages cross-client and cross-task transfer from past knowledge to generate informed initializations for learning new tasks with adapters. FedKEI begins with a global clustering process at the server to generalize knowledge across tasks, followed by the optimization of aggregation weights across clusters (inter-cluster weights) and within each cluster (intra-cluster weights) to personalize knowledge transfer for each new task. To facilitate more effective learning of the inter- and intra-cluster weights, we adopt a bi-level optimization scheme that collaboratively learns the global intra-cluster weights across clients and optimizes the local inter-cluster weights toward each client's task objective. Extensive experiments on three benchmark datasets of different modalities, including dermatology, chest X-rays, and retinal OCT, demonstrate FedKEI's advantage in adapting to new diseases compared to state-of-the-art methods.
format Preprint
id arxiv_https___arxiv_org_abs_2508_10299
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Improving Learning of New Diseases through Knowledge-Enhanced Initialization for Federated Adapter Tuning
Peng, Danni
Wang, Yuan
Cai, Kangning
Ning, Peiyan
Xu, Jiming
Liu, Yong
Goh, Rick Siow Mong
Wei, Qingsong
Fu, Huazhu
Machine Learning
Computer Vision and Pattern Recognition
In healthcare, federated learning (FL) is a widely adopted framework that enables privacy-preserving collaboration among medical institutions. With large foundation models (FMs) demonstrating impressive capabilities, using FMs in FL through cost-efficient adapter tuning has become a popular approach. Given the rapidly evolving healthcare environment, it is crucial for individual clients to quickly adapt to new tasks or diseases by tuning adapters while drawing upon past experiences. In this work, we introduce Federated Knowledge-Enhanced Initialization (FedKEI), a novel framework that leverages cross-client and cross-task transfer from past knowledge to generate informed initializations for learning new tasks with adapters. FedKEI begins with a global clustering process at the server to generalize knowledge across tasks, followed by the optimization of aggregation weights across clusters (inter-cluster weights) and within each cluster (intra-cluster weights) to personalize knowledge transfer for each new task. To facilitate more effective learning of the inter- and intra-cluster weights, we adopt a bi-level optimization scheme that collaboratively learns the global intra-cluster weights across clients and optimizes the local inter-cluster weights toward each client's task objective. Extensive experiments on three benchmark datasets of different modalities, including dermatology, chest X-rays, and retinal OCT, demonstrate FedKEI's advantage in adapting to new diseases compared to state-of-the-art methods.
title Improving Learning of New Diseases through Knowledge-Enhanced Initialization for Federated Adapter Tuning
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2508.10299