Federated learning (FL) is a distributed learning paradigm that collaboratively learns a global model from multiple decentralized clients without sharing raw data, providing an effective solution for privacy preservation. However, current FL works primarily focus on high-level visual tasks (e.g., classification), the exploration for image super-resolution (SR), a fundamental low-level visual task, is rarely addressed. In this study, we propose Federated Image Super-Resolution (FedSR), a novel FL approach for image SR task, aiming to generate high-quality images while preserving data privacy. Specifically, we introduce a detail-assisted contrastive loss at the decentralized client, aligning the shallow representations by rectifying the relationship of low-level features between local clients and global server. In addition, we develop a hierarchical aggregation policy at the central server to better integrate dispersed client models into the global model, with the aggregation weight determined based on the layer-wise similarity between the updated local models and the historical global model. Extensive experiments conducted on general SR benchmarks and facial image datasets demonstrate the superiority of our FedSR in terms of image quality and model performance. Notably, FedSR can be seamlessly integrated with various prevalent image SR methods, including CNN-based and Transformer-based architectures.
Read full abstract