跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

MuCALD-SplitFed:用于隐私保护多任务分割联邦学习的因果潜在扩散模型

文章摘要

本文提出MuCALD-SplitFed,一种结合因果表示学习和潜在扩散的多任务分割联邦学习框架,旨在解决医疗图像分割中跨机构任务不一致和隐私泄露问题。实验表明,该方法在分割性能上持续提升,同时有效减少信息泄露,优于现有先进方法。

来源arXiv Computer Vision作者: Chamani Shiranthika, Hadi Hadizadeh, Parvaneh Saeedi
MuCALD-SplitFed:用于隐私保护多任务分割联邦学习的因果潜在扩散模型
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

联邦学习(FL)通过聚合客户端模型更新实现去中心化训练,无需共享原始数据,从而保护数据隐私。分割联邦学习(SplitFed)进一步将模型分割至客户端和服务器,客户端仅需处理部分模型,大幅降低计算和通信开销。然而,在医疗场景中,不同机构通常执行不同的分割任务,例如某些机构专注于肝脏分割,而其他机构可能关注脑肿瘤分割。这种任务异构性导致标准的FL和SplitFed难以直接应用,因为全局模型无法同时适应所有任务。多任务联邦学习(MTFL)允许客户端各自处理不同的任务,但往往引入训练不稳定性和额外的隐私风险,例如服务器可能从共享的表示中重建客户的私有数据。

为了解决这些问题,研究者提出了MuCALD-SplitFed,一个新颖的多任务分割联邦学习框架。该框架将因果表示学习与潜在扩散模型有机结合。因果表示学习旨在提取任务无关且因果稳健的特征,而潜在扩散模型则通过生成式建模增强表示的多样性和隐私性。具体来说,在分割点(即客户端与服务器交换中间表示的位置),MuCALD-SplitFed引入因果潜在扩散过程,使得不同任务的表示在潜在空间中趋于对齐,从而促进多任务协同训练。同时,扩散过程中的噪声注入机制有效防止了攻击者从中间表示中恢复原始图像或推断成员信息。

实验在多个医学图像分割数据集上进行,包括视网膜血管分割、心脏分割和肺部分割等任务。结果表明,基线SplitFed由于任务冲突而无法收敛,而MuCALD-SplitFed不仅稳定收敛,而且分割性能持续提升,达到与集中式训练相当的水平。在隐私评估中,基于重构和成员推断的攻击成功率显著下降,证明了该方法在隐私保护方面的优越性。此外,与当前最先进的个性化FL(如pFL)和多任务FL(如FedAvg变体)相比,MuCALD-SplitFed在Dice系数、Hausdorff距离等指标上均取得最佳结果。该工作已被IEEE国际图像处理大会(ICIP 2026)接收为口头报告,代码已开源在GitHub上。

展开要点与分析

文章情报

工程师进阶

要点

  • MuCALD-SplitFed融合因果表示学习和潜在扩散,实现多任务分割联邦学习。
  • 与基线分割联邦学习相比,该方法能够收敛且分割性能持续提升。
  • 显著降低分割点的信息泄露,抵御重构和成员推断攻击。
  • 在多项指标上超越当前最优的个性化联邦学习和多任务联邦学习方法。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。