arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.09757cs.CVcs.AI

UFPS:异构数据分布下部分标注联邦分割的统一框架

UFPS: A unified framework for partially-annotated federated segmentation in heterogeneous data distribution

Shanghai University(上海大学);Chinese University of Hong Kong(香港中文大学);Elsevier B.V.(爱思唯尔出版集团);STM Document Engineering Pvt Ltd.(STM文档工程私人有限公司)

Le Jiang, Li Yan Ma, Tie Yong Zeng, Shi Hui Ying

AI总结:

针对部分标注联邦分割中的类别异构与客户端漂移问题,提出UFPS框架,通过统一标签学习和稀疏统一锐度感知最小化,实现无类别冲突的全类别分割,并在真实医学数据集上验证了其优越性。

摘要:

部分监督分割是一种基于数据集中部分类别被标注且类别间存在交集的标签节省方法。然而,由于隐私问题和数据异构性,该方法仍远未应用于真实世界的医学场景。作为无隐私泄露的解决方案,本文提出了联邦部分监督分割(FPSS)。FPSS的主要挑战是类别异构性和客户端漂移。我们提出了一种统一联邦部分标注分割(UFPS)框架,通过训练一个全能全局模型,在不发生类别冲突的情况下,对部分标注数据集中的所有类别进行像素分割。该框架包括统一标签学习和稀疏统一锐度感知最小化,分别用于统一类别空间和特征空间。通过实证研究,我们发现传统部分监督分割与联邦学习方法简单组合的主要障碍是类别冲突。我们在真实医学数据集上的综合实验表明,与改进方法相比,UFPS具有更好的去冲突能力和泛化能力。

英文摘要:

Partially supervised segmentation is a label-saving method based on datasets with fractional classes labeled and intersectant. However, it is still far from landing on real-world medical applications due to privacy concerns and data heterogeneity. As a remedy without privacy leakage, federated partially supervised segmentation (FPSS) is formulated in this work. The main challenges for FPSS are class heterogeneity and client drift. We propose a Unified Federated Partially-labeled Segmentation (UFPS) framework to segment pixels within all classes for partially-annotated datasets by training a totipotential global model without class collision. Our framework includes Unified Label Learning and sparsed Unified Sharpness Aware Minimization for unification of class and feature space, respectively. We find that vanilla combinations for traditional methods in partially supervised segmentation and federated learning are mainly hampered by class collision through empirical study. Our comprehensive experiments on real medical datasets demonstrate better deconflicting and generalization ability of UFPS compared with modified methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑