arXiv:2511.08130cs.CVcs.DC2025-11

用联邦学习+SAM2实现污水厂泡沫自动分割,保护数据隐私且效果好

Foam Segmentation in Wastewater Treatment Plants: A Federated Learning Approach with Segment Anything Model 2

  • 通过联邦学习在多厂间协作训练,不共享原始数据
  • 结合SAM2预训练模型,少量本地数据也能快速收敛
  • 适合关注工业数据隐私与自动化监测的水务从业者

污水厂泡沫问题严重影响处理效率并增加成本。传统机器学习需大量标注数据,但不同厂区因隐私顾虑难以共享数据,导致系统开发缓慢。本文提出一种融合联邦学习(FL)与图像分割前沿模型SAM2的新框架。该框架通过分布式客户端(边缘节点)在Flower框架下微调SAM2,中央雾服务器仅聚合模型权重而不接触私有数据,保障隐私。实验使用西班牙格拉纳达真实污水厂图像、合成泡沫数据集及公开数据集进行训练与验证,显著提升在小样本下的分割性能与收敛速度。研究证明,将大模型与联邦学习结合,可有效应对分布式、敏感的工业数据挑战,为污水厂泡沫实时监控提供可扩展、隐私友好的解决方案。

原文摘要 · Abstract (English)

Foam formation in Wastewater Treatment Plants (WTPs) is a major challenge that can reduce treatment efficiency and increase costs. The ability to automatically examine changes in real-time with respect to the percentage of foam can be of great benefit to the plant. However, large amounts of labeled data are required to train standard Machine Learning (ML) models. The development of these systems is slow due to the scarcity and heterogeneity of labeled data. Additionally, the development is often hindered by the fact that different WTPs do not share their data due to privacy concerns. This paper proposes a new framework to address these challenges by combining Federated Learning (FL) with the state-of-the-art base model for image segmentation, Segment Anything Model 2 (SAM2). The FL paradigm enables collaborative model training across multiple WTPs without centralizing sensitive operational data, thereby ensuring privacy. The framework accelerates training convergence and improves segmentation performance even with limited local datasets by leveraging SAM2's strong pre-trained weights for initialization. The methodology involves fine-tuning SAM2 on distributed clients (edge nodes) using the Flower framework, where a central Fog server orchestrates the process by aggregating model weights without accessing private data. The model was trained and validated using various data collections, including real-world images captured at a WTPs in Granada, Spain, a synthetically generated foam dataset, and images from publicly available datasets to improve generalization. This research offers a practical, scalable, and privacy-aware solution for automatic foam tracking in WTPs. The findings highlight the significant potential of integrating large-scale foundational models into FL systems to solve real-world industrial challenges characterized by distributed and sensitive data.

联邦学习图像分割水务智能隐私保护

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。