跳到主要內容
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系統奠定基礎。

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