跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

FedCC:一種用於胎兒超聲圖像中胼胝體穩健定位的低資源聯邦基礎模型適配方法

文章摘要

FedCC提出了一種基於聯邦學習的框架,結合凍結的DINOv2骨幹網絡、輕量級YOLO檢測頭和低秩適配(LoRA)模塊,實現胎兒超聲圖像中胼胝體的精準定位。在包含10,970幀多中心數據集的評估中,該方法在FedAvg策略下達到平均mAP@50為0.857、F1分數為0.803,同時將可訓練參數從24.4M降至2.9M,通信成本減少約8.5倍。

來源arXiv Machine Learning作者: Alessandro Di Matteo, Sara Moccia, Giuseppe Rizzo, Gianpaolo Grisolia, Ricciarda Raffaelli, Lorenzo Vasciaveo, Francesco D'Antonio, Maria Chiara Fiorentino
FedCC:一種用於胎兒超聲圖像中胼胝體穩健定位的低資源聯邦基礎模型適配方法
報告錯誤

更正渠道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

胼胝體(CC)是大腦中重要的白質結構,在胎兒超聲圖像中準確識別其對早期發現神經發育異常至關重要。然而,由於超聲成像固有的低對比度、散斑噪聲以及胼胝體解剖結構的顯著變異,這項任務極具挑戰性。來自多所研究機構的研究人員在arXiv上發佈了一項名為FedCC的新方法,旨在通過聯邦學習框架解決這一問題,該框架特別設計用於真實的多中心和資源受限臨牀環境,無需共享患者數據。

FedCC的核心創新在於整合了三種關鍵技術:凍結的DINOv2視覺Transformer骨幹網絡、輕量級YOLO檢測頭以及低秩適配(LoRA)模塊。DINOv2作為基礎模型提供強大的特徵提取能力,但其參數在訓練過程中保持凍結,以降低計算負擔;YOLO檢測頭負責目標定位;而LoRA模塊僅允許少量可訓練參數進行更新和客户端間交換。這種參數高效適配策略大幅減少了計算和通信開銷,使框架適合低資源環境。研究團隊在包含3個臨牀中心、58名孕婦的10,970幀超聲圖像數據集上評估了FedCC,這些圖像由不同成像設備採集,覆蓋了真實世界的異質性。

在聯邦平均(FedAvg)策略下,DINOv2與LoRA的組合取得了平均mAP@50為0.857、F1分數為0.803的成績,超越了全微調和編碼器凍結基線。值得注意的是,該方法的可訓練參數僅2.9M,相比全微調的24.4M減少了約88%,通信成本相應降低約8.5倍。這些結果標誌着在開發可擴展、保護隱私且可臨牀部署的胎兒神經超聲AI系統方面邁出了有希望的一步。FedCC不僅證明了基礎模型在醫學圖像分析中的潛力,還展示了聯邦學習在克服數據隱私和資源限制方面的價值。未來工作可能包括在更大規模數據集上的驗證以及與其他胎兒結構的聯合檢測。

展開要點與分析

文章情報

投資人進階

要點

  • FedCC整合DINOv2、YOLO和LoRA,實現高效的聯邦學習。
  • 在10,970幀多中心數據集上達到mAP@50 0.857,參數減少至2.9M。
  • 通信成本降低8.5倍,適用於低資源環境。
  • 為可擴展、保護隱私的胎兒神經超聲AI系統奠定基礎。

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