Polarized images have significant application value in optical research and computer vision. Collaborative training of polarized images can effectively integrate multi-source data and enhance the generalization ability and performance of models. However, due to the fact that these data are usually distributed across multiple devices and have high privacy sensitivity, such as distributed devices or terminals equipped with polarized cameras, direct central training may face the risk of data leakage and insufficient privacy protection. To address this issue, this paper proposes a polarization image enhancement method based on federated learning to effectively safeguard data privacy. To optimize the model aggregation in the federated learning framework, a loss-weighted aggregation strategy is designed, combining multi-source information from each client to construct a global model for distributed low-light image enhancement. Experiments conducted on two real-world datasets demonstrate that the proposed method not only significantly reduces training time but also outperforms traditional centralized methods in terms of privacy protection. To ensure reproducibility, the source code and demonstration videos can be found at https://pan.baidu.com/s/1dzWMZgnGYl92uyabAkfhXQ (Extraction code: zs6h).

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Polarization Image Enhancement Method Based on Federated Learning

  • Meng Wang,
  • Chang-E Ren

摘要

Polarized images have significant application value in optical research and computer vision. Collaborative training of polarized images can effectively integrate multi-source data and enhance the generalization ability and performance of models. However, due to the fact that these data are usually distributed across multiple devices and have high privacy sensitivity, such as distributed devices or terminals equipped with polarized cameras, direct central training may face the risk of data leakage and insufficient privacy protection. To address this issue, this paper proposes a polarization image enhancement method based on federated learning to effectively safeguard data privacy. To optimize the model aggregation in the federated learning framework, a loss-weighted aggregation strategy is designed, combining multi-source information from each client to construct a global model for distributed low-light image enhancement. Experiments conducted on two real-world datasets demonstrate that the proposed method not only significantly reduces training time but also outperforms traditional centralized methods in terms of privacy protection. To ensure reproducibility, the source code and demonstration videos can be found at https://pan.baidu.com/s/1dzWMZgnGYl92uyabAkfhXQ (Extraction code: zs6h).