Improving Learning of New Diseases through Knowledge-Enhanced Initialization for Federated Adapter Tuning
2508.10299v1
cs.LG, cs.CV
2025-08-15
Авторы:
Danni Peng, Yuan Wang, Kangning Cai, Peiyan Ning, Jiming Xu, Yong Liu, Rick Siow Mong Goh, Qingsong Wei, Huazhu Fu
Резюме на русском
## Контекст
Современные медицинские исследования сталкиваются с вопросами быстрого адаптирования к новым заболеваниям и высокой защиты конфиденциальности данных. Федеративное обучение (FL) широко применяется в здравоохранении, позволяя нескольким медицинским учреждениям обмениваться знаниями без передачи чувствительных данных. Большие фундаментальные модели (FMs) продемонстрировали вакумный потенциал в области медицинского анализа данных. Однако применение этих моделей через адаптерный тюнинг часто оказывается дорогостоящим и неэффективным. Более того, новые заболевания требуют не только персонализированного подхода, но и эффективного использования предыдущих знаний для ускорения обучения. Это влечет за собой необходимость разработки новых методов, которые бы ускоряли адаптацию к новым задачам, учитывая имеющиеся знания.
## Метод
Мы предлагаем Federated Knowledge-Enhanced Initialization (FedKEI) — новую архитектуру, основанную на понятии знаний, которое усилено с помощью кросс-клиентского и кросс-задачного переноса. FedKEI сначала выполняет глобальный кластеризационный процесс на сервере для обобщения знаний по всем задачам. Затем он оптимизирует веса аггрегации между кластерами (интер-кластерные веса) и внутри каждого кластера (интерьер-кластерные веса), чтобы улучшить личный трансфер для каждого клиента. Для этого используется билевел-оптимизационный схема, в которой глобальные веса интерьера учитываются вместе со специфичными локальными весами клиентов. Эта структура позволяет FedKEI эффективно адаптироваться к новым заболеваниям, используя предыдущие знания для оптимизации начальных параметров адаптера.
## Результаты
Мы проверили FedKEI на трех бенчмарк-сетях, относящихся к разным модальностям: дерматологии, травм брюшины и ретинальных OCT-сканах. Наши эксперименты показали, что наше решение превосходит состояние технологий по эффективности и скорости обучения. В частности, FedKEI показал существенное улучшение в скорости построения первых моделей для новых заболеваний, а также в их точности и потреблении ресурсов. Эти результаты демонстрируют высокую эффективность FedKEI в ситуациях, требующих быстрого адаптирования к новым задачам в сфере здравоохранения.
## Значимость
FedKEI может применяться в различных медицинских областях, таких как дерматология, радиология и окулология. Его преимущества заключаются в ускорении обучения моделей для новых заболеваний, повышении точности и экономии ресурсов. Этот подход имеет потенциал для ускорения развития медицинских технологий, обеспечивая быстрое реагирование на новые заболевания
Abstract
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.
Ссылки и действия
Дополнительные ресурсы: