解决联邦学习中因分辨率差异导致的性能下降问题
Mitigating Resolution-Drift in Federated Learning: Case of Keypoint Detection
- 通过热图知识蒸馏实现多分辨率自适应联邦学习
- 在关键点检测任务中显著提升低分辨率客户端表现
- 适用于需要保留空间细节的视觉任务,如姿态估计
联邦学习(FL)可在保护用户数据隐私的同时实现分布式协同学习。现有研究主要关注统计异构性和通信效率,在分类任务中已取得成功,但在非分类任务(如人体姿态估计)中的应用仍不充分。本文首次识别并分析了“分辨率漂移”问题:由于客户端间分辨率差异,模型性能显著下降。与类别级异构性不同,分辨率漂移凸显了分辨率作为非独立同分布(non-IID)数据的另一维度的重要性。为此,本文提出分辨率自适应联邦学习(RAF),采用基于热图的知识蒸馏机制,在高分辨率输出(教师)与低分辨率输出(学生)之间进行多分辨率知识迁移,增强对分辨率变化的鲁棒性,同时避免过拟合。大量实验与理论分析表明,RAF能有效缓解分辨率漂移,带来显著性能提升,并可无缝集成至现有联邦学习框架。尽管本文聚焦人体姿态估计,但t-SNE分析揭示分类任务与高分辨率表征任务存在本质差异,支持RAF在其他依赖空间细节的任务中的通用性。
原文摘要 · Abstract (English)
The Federated Learning (FL) approach enables effective learning across distributed systems, while preserving user data privacy. To date, research has primarily focused on addressing statistical heterogeneity and communication efficiency, through which FL has achieved success in classification tasks. However, its application to non-classification tasks, such as human pose estimation, remains underexplored. This paper identifies and investigates a critical issue termed ``resolution-drift,'' where performance degrades significantly due to resolution variability across clients. Unlike class-level heterogeneity, resolution drift highlights the importance of resolution as another axis of not independent or identically distributed (non-IID) data. To address this issue, we present resolution-adaptive federated learning (RAF), a method that leverages heatmap-based knowledge distillation. Through multi-resolution knowledge distillation between higher-resolution outputs (teachers) and lower-resolution outputs (students), our approach enhances resolution robustness without overfitting. Extensive experiments and theoretical analysis demonstrate that RAF not only effectively mitigates resolution drift and achieves significant performance improvements, but also can be integrated seamlessly into existing FL frameworks. Furthermore, although this paper focuses on human pose estimation, our t-SNE analysis reveals distinct characteristics between classification and high-resolution representation tasks, supporting the generalizability of RAF to other tasks that rely on preserving spatial detail.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。