跳到主要內容
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融合因果表示學習和潛在擴散,實現多工分割聯邦學習。
  • 與基線分割聯邦學習相比,該方法能夠收斂且分割效能持續提升。
  • 顯著降低分割點的資訊洩露,抵禦重構和成員推斷攻擊。
  • 在多項指標上超越當前最優的個性化聯邦學習和多工聯邦學習方法。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。