让视觉语言模型在联邦学习中自适应测试分布,兼顾个性化与协作。
Latte: Collaborative Test-Time Adaptation of Vision-Language Models in Federated Learning

- 客户端本地存历史数据嵌入,跨客户端共享类别原型以增强记忆。
- 在多个数据分布下显著提升性能,通信与计算开销极低。
- 适合资源受限的分布式场景,尤其适用于测试时分布漂移问题。
使用预训练视觉语言模型进行测试时自适应,已成为应对测试阶段分布偏移的重要方法。其中,基于记忆的方法因其无需训练且可利用历史测试数据而备受关注。然而,现有方法多针对单一领域且依赖充足数据,在联邦学习等去中心化场景中,各客户端独立应用会因测试数据有限导致效果不佳;若直接由服务器共享全局记忆,则难以满足客户端各自的分布特性。为此,我们提出 Latte 框架:每个客户端维护本地记忆(存储自身历史测试数据嵌入)和外部记忆(存储其他相关客户端的类别原型)。通信时,在服务器协调下,客户端从相似客户端检索原型以扩展自身记忆。本地适应阶段,结合嵌入相似性与不确定性来提升模型表现。理论分析表明,Latte 能有效利用同分布客户端,同时对异分布客户端保持鲁棒性。在域迁移与噪声污染基准上的大量实验验证了其在去中心化设置下的优越性能,且通信与计算开销几乎可忽略。代码已公开于 https://github.com/baowenxuan/Latte。
原文摘要 · Abstract (English)
Test-time adaptation with pre-trained vision-language models has gained increasing attention for addressing distribution shifts during testing. Among these approaches, memory-based algorithms stand out due to their training-free nature and ability to leverage historical test data. However, existing test-time adaptation methods are typically designed for a single domain with abundant data. In decentralized settings such as federated learning, applying these methods individually to each client suffers from limited test data, while directly sharing a single global memory via the server prevents proper personalization to each client's unique distribution. To address this, we propose Latte, a novel framework where each client maintains a local memory to store embeddings from its own historical test data and an external memory to store class prototypes from other relevant clients. During communication, each client retrieves prototypes from similar clients under the server's coordination to expand its memory. For local adaptation, Latte utilizes both embedding similarity and uncertainty to enhance model performance. Our theoretical analysis shows that Latte effectively leverages in-distribution clients while remaining robust to out-of-distribution clients. Extensive experiments on domain adaptation and corruption benchmarks validate that Latte achieves superior performance in decentralized settings, while introducing only negligible communication and computation costs. Our code is available at https://github.com/baowenxuan/Latte .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。