atlas

Research map / Federated learning and privacy / Personalized and heterogeneous FL

Dual-Personalizing Adapter for Federated Foundation Models

NeurIPS 2024 · Federated learning and privacy › Personalized and heterogeneous FL · approach: Sparse and collaboration-graph personalization · 13 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes FedDPA, a federated dual-adapter method (global LoRA shared via FedAvg plus a client-local LoRA) with instance-wise similarity-based weighting, to achieve personalization and test-time personalization for federated LLMs.

Problem

Federated foundation models are personalized to client tasks, but existing methods ignore test-time distribution shifts (unseen tasks/distributions at inference), and conventional test-time FL methods are ill-suited to LLMs that require training all parameters.

Approach

Each client keeps a frozen LLM with a globally aggregated LoRA adapter (for generic, test-time knowledge) and a local LoRA adapter (for personalization); at inference, the two are mixed per instance using cosine similarity between last-token LLM embeddings of the test input and sampled local training instances.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add instance-wise dynamic weighting (ablation)FedDPA-F · Fed Dataset 1 and Fed Dataset 2 personalizationROGUE-1: 79.06 -> 79.48; 73.17 -> 73.48Helped
add instance-wise dynamic weighting (ablation)FedDPA-T · Fed Dataset 1 and Fed Dataset 2 personalizationROGUE-1: 79.57 -> 80.22; 73.75 -> 75.33Helped
add instance-wise dynamic weighting (ablation)FedDPA-F · Fed Dataset 1 and Fed Dataset 2 test-time personalizationROGUE-1: 76.97 -> 77.12; 71.70 -> 71.64Mixed
add instance-wise dynamic weighting (ablation)FedDPA-T · Fed Dataset 1 and Fed Dataset 2 test-time personalizationROGUE-1: 60.06 -> 75.47; 63.57 -> 71.27Helped
use updating weight alpha 0.3 rather than 0.5 or 0.7 (ablation)FedDPA-T · Fed Dataset 1 test-time personalizationROGUE-1: 75.47; 75.01; 75.85Helped
vary updating weight alpha (ablation)FedDPA-T · Fed Dataset 1 and Fed Dataset 2 personalizationROGUE-1: 79.69; 80.22; 79.88; 75.33; 74.10; 74.04Mixed
use cosine similarity instead of L2 norm or Pearson correlation (ablation)FedDPA-F · Fed Dataset 1 personalizationROGUE-1: 77.69; 79.09; 79.48Helped
use cosine similarity instead of L2 norm or Pearson correlation (ablation)FedDPA-F · Fed Dataset 1 test-time personalizationROGUE-1: 77.10; 76.78; 77.12Mixed
use cosine similarity instead of L2 norm or Pearson correlation (ablation)FedDPA-T · Fed Dataset 1 personalizationROGUE-1: 77.58; 79.73; 80.22Helped
use cosine similarity instead of L2 norm or Pearson correlation (ablation)FedDPA-T · Fed Dataset 1 test-time personalizationROGUE-1: 77.30; 75.27; 75.47Mixed
use the final-token embedding instead of averaging all token embeddings (ablation)FedDPA-F · Fed Dataset 1 personalization and test-time personalizationROGUE-1: 79.30 -> 79.48; 76.77 -> 77.12Helped
use the final-token embedding instead of averaging all token embeddings (ablation)FedDPA-T · Fed Dataset 1 personalization and test-time personalizationROGUE-1: 79.65 -> 80.22; 73.36 -> 75.47Helped

Surprising findings

Limitations

Experimental setup

Datasets: Flan (GLUE QQP, SNLI, WebNLG, fix_punct, CoLA, WiC, definite pronoun resolution, TREC; paws_wiki, hellaswag, qnli, word_

Benchmarks: Federated Dataset 1 Federated Dataset 2

Metrics: ROUGE-1 accuracy

Baselines: Centralized Local-finetuned FedIT FedLoRA FedProx

Builds on: LoRA FedAvg FedProx FedIT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy