arXiv:2502.21012cs.DCcs.CV2025-02被引 3

提出动态记忆联邦学习方法,解决工业医疗图像异常检测的隐私与数据偏移问题。

FedDyMem: Efficient Federated Learning with Dynamic Memory and Memory-Reduce for Unsupervised Image Anomaly Detection

  • 用客户端动态记忆库替代模型参数共享,实现隐私保护下的知识传递。
  • 通过加权平均压缩记忆库,通信量减少62%,降低数据重建风险。
  • 适用于多客户端工业/医疗异常检测,尤其适合正类样本单一且分布不均场景。

无监督图像异常检测(UAD)在工业和医疗应用中至关重要,但日益严峻的数据隐私问题带来挑战。由于单类分类任务固有的类别多样性不足,以及客户端间和内部产品差异导致的分布偏移,传统联邦学习难以兼顾隐私保护与性能。为此,本文提出一种高效的联邦学习方法FedDyMem,通过客户端动态记忆库实现知识共享,而非传输模型参数。在本地客户端,利用记忆生成器与度量损失提升正常样本特征分布的一致性,并动态更新记忆库。为降低通信开销与隐私泄露风险,提出基于加权平均的记忆压缩方法,显著减小记忆库规模。服务器端通过k-means聚合构建全局记忆并分发至各客户端。在六个工业与医疗数据集上的实验表明,这些数据集源自十一个公开数据源,涵盖六种产品或健康筛查类型,验证了该方法的有效性。

原文摘要 · Abstract (English)

Unsupervised image anomaly detection (UAD) has become a critical process in industrial and medical applications, but it faces growing challenges due to increasing concerns over data privacy. The limited class diversity inherent to one-class classification tasks, combined with distribution biases caused by variations in products across and within clients, poses significant challenges for preserving data privacy with federated UAD. Thus, this article proposes an efficient federated learning method with dynamic memory and memory-reduce for unsupervised image anomaly detection, called FedDyMem. Considering all client data belongs to a single class (i.e., normal sample) in UAD and the distribution of intra-class features demonstrates significant skewness, FedDyMem facilitates knowledge sharing between the client and server through the client's dynamic memory bank instead of model parameters. In the local clients, a memory generator and a metric loss are employed to improve the consistency of the feature distribution for normal samples, leveraging the local model to update the memory bank dynamically. For efficient communication and data privacy, a memory-reduce method based on weighted averages is proposed to significantly decrease the scale of memory banks. This reduced representation inherently, thereby mitigating the risk of data reconstruction. On the server, global memory is constructed and distributed to individual clients through k-means aggregation. Experiments conducted on six industrial and medical datasets, comprising a mixture of six products or health screening types derived from eleven public datasets, demonstrate the effectiveness of FedDyMem.

联邦学习异常检测隐私保护动态记忆

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