聯邦學習(FL)透過聚合客戶端模型更新實現去中心化訓練,無需共享原始資料,從而保護資料隱私。分割聯邦學習(SplitFed)進一步將模型分割至客戶端和伺服器,客戶端僅需處理部分模型,大幅降低計算和通訊開銷。然而,在醫療場景中,不同機構通常執行不同的分割任務,例如某些機構專注於肝臟分割,而其他機構可能關注腦腫瘤分割。這種任務異構性導致標準的FL和SplitFed難以直接應用,因為全域性模型無法同時適應所有任務。多工聯邦學習(MTFL)允許客戶端各自處理不同的任務,但往往引入訓練不穩定性和額外的隱私風險,例如伺服器可能從共享的表示中重建客戶的私有資料。
為了解決這些問題,研究者提出了MuCALD-SplitFed,一個新穎的多工分割聯邦學習框架。該框架將因果表示學習與潛在擴散模型有機結合。因果表示學習旨在提取任務無關且因果穩健的特徵,而潛在擴散模型則透過生成式建模增強表示的多樣性和隱私性。具體來說,在分割點(即客戶端與伺服器交換中間表示的位置),MuCALD-SplitFed引入因果潛在擴散過程,使得不同任務的表示在潛在空間中趨於對齊,從而促進多工協同訓練。同時,擴散過程中的噪聲注入機制有效防止了攻擊者從中間表示中恢復原始影像或推斷成員資訊。
實驗在多個醫學影像分割資料集上進行,包括視網膜血管分割、心臟分割和肺部分割等任務。結果表明,基線SplitFed由於任務衝突而無法收斂,而MuCALD-SplitFed不僅穩定收斂,而且分割效能持續提升,達到與集中式訓練相當的水平。在隱私評估中,基於重構和成員推斷的攻擊成功率顯著下降,證明了該方法在隱私保護方面的優越性。此外,與當前最先進的個性化FL(如pFL)和多工FL(如FedAvg變體)相比,MuCALD-SplitFed在Dice係數、Hausdorff距離等指標上均取得最佳結果。該工作已被IEEE國際影像處理大會(ICIP 2026)接收為口頭報告,程式碼已開源在GitHub上。