用自监督预训练的视觉模型提升联邦域泛化性能。
Boosting Federated Domain Generalization: Understanding the Role of Advanced Pre-Trained Architectures
- 采用ViT、ConvNeXt等先进架构,结合大尺度自监督预训练。
- 在Office-Home和PACS上分别达到84.46%和92.55%平均准确率。
- 小参数模型胜过大型ResNet,适合资源受限场景。
本研究探讨了Vision Transformers (ViT)、ConvNeXt、Swin Transformer等先进预训练架构在提升联邦域泛化(Federated Domain Generalization, FDG)中的有效性。这些架构能捕捉全局上下文特征并建模长程依赖,具备提升跨域泛化潜力。研究系统评估了多种架构变体,使用ImageNet-1K、ImageNet-21K、JFT-300M、ImageNet-22K等大规模预训练数据集,并对比自监督与监督预训练策略。结果表明,以掩码图像块重建为核心的自监督方法能更好捕捉图像内在结构,表现优于监督方法。在Office-Home和PACS数据集上的全面评估显示,采用大规模预训练的先进架构建立新基准,平均准确率分别达84.46%和92.55%。值得注意的是,某些参数更少的先进模型仍超越更大规模的ResNet,凸显了复杂架构与多样化预训练策略对提升FDG的关键作用,尤其在计算资源受限时更具优势。研究为未来联邦学习中的域泛化提供了重要启示。
原文摘要 · Abstract (English)
In this study, we explore the efficacy of advanced pre-trained architectures, such as Vision Transformers (ViT), ConvNeXt, and Swin Transformers in enhancing Federated Domain Generalization. These architectures capture global contextual features and model long-range dependencies, making them promising candidates for improving cross-domain generalization. We conduct a broad study with in-depth analysis and systematically evaluate different variants of these architectures, using extensive pre-training datasets such as ImageNet-1K, ImageNet-21K, JFT-300M, and ImageNet-22K. Additionally, we compare self-supervised and supervised pre-training strategies to assess their impact on FDG performance. Our findings suggest that self-supervised techniques, which focus on reconstructing masked image patches, can better capture the intrinsic structure of images, thereby outperforming their supervised counterparts. Comprehensive evaluations on the Office-Home and PACS datasets demonstrate that adopting advanced architectures pre-trained on larger datasets establishes new benchmarks, achieving average accuracies of 84.46\% and 92.55\%, respectively. Additionally, we observe that certain variants of these advanced models, despite having fewer parameters, outperform larger ResNet models. This highlights the critical role of utilizing sophisticated architectures and diverse pre-training strategies to enhance FDG performance, especially in scenarios with limited computational resources where model efficiency is crucial. Our results indicate that federated learning systems can become more adaptable and efficient by leveraging these advanced methods, offering valuable insights for future research in FDG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。