核心概念
本文提出了一種名為 Sli2Vol+ 的新型自我監督遮罩傳播框架,用於僅使用每個訓練和測試體積中的一個標註切片來分割 3D 醫學影像中的任何解剖結構,透過物件估計引導的對應流網路來學習可靠的對應關係,有效解決誤差累積和不連續性問題。
文獻資訊
An, D., Gu, P., Sonka, M., Wang, C., & Chen, D. Z. (2024). Sli2Vol+: Segmenting 3D Medical Images Based on an Object Estimation Guided Correspondence Flow Network. arXiv preprint arXiv:2411.13873.
研究目標
本研究旨在開發一種新的自我監督遮罩傳播框架,解決現有方法在 3D 醫學影像分割中面臨的標註負擔問題。具體而言,目標是僅使用每個訓練和測試體積中的一個標註切片,即可準確分割 3D 醫學影像中的解剖結構。
方法
本研究提出了一種名為 Sli2Vol+ 的新型自我監督遮罩傳播框架,其核心是一種稱為物件估計引導的對應流網路 (OEG-CFN) 的新型網路。
訓練階段
偽標籤生成: 使用 Sli2Vol 方法將訓練體積中已標註切片的標籤傳播到其他切片,生成偽標籤。
偽標籤優化: 使用 3D 模型 (UNETR++) 對生成的偽標籤進行優化,以提升偽標籤的品質。
OEG-CFN 訓練: 使用 OEG-CFN 學習相鄰切片和對應偽標籤之間的像素級對應關係。OEG-CFN 包含兩個路徑,分別從相鄰切片和偽標籤中學習特徵,並將其串聯起來計算最終的對應關係。
測試階段
對應關係計算: 使用訓練好的 OEG-CFN 計算測試體積中相鄰切片之間的對應關係。
遮罩傳播: 利用計算出的對應關係,將已標註切片的遮罩傳播到其他切片,生成整個體積的分割結果。
主要發現
Sli2Vol+ 在九個公開數據集(包括 CT 和 MRI)上進行了評估,涵蓋十種不同的感興趣結構 (SOI)。
與需要完整標註數據訓練的模型相比,Sli2Vol+ 在跨域評估中表現出更小的效能下降,顯示其解決領域轉移問題的潛力。
與其他僅需要每個測試體積中一個標註切片的半監督方法相比,Sli2Vol+ 在所有數據集上都取得了顯著的效能提升。
Sli2Vol+ 在跨模態評估中也表現出優於其他方法的效能,顯示其在跨模態任務中的泛化能力更強。
主要結論
Sli2Vol+ 是一種有效的自我監督遮罩傳播框架,能夠在僅使用每個訓練和測試體積中的一個標註切片的情況下,實現準確的 3D 醫學影像分割。透過物件估計引導的對應流網路,Sli2Vol+ 有效解決了現有方法面臨的誤差累積和不連續性問題,並在跨域和跨模態評估中表現出更強的泛化能力。
意義
本研究提出的 Sli2Vol+ 方法為解決 3D 醫學影像分割中的標註負擔問題提供了一種新的思路,並在多個公開數據集上取得了令人鼓舞的結果。這項工作有望促進醫學影像分析領域的發展,並為臨床診斷和治療提供更有效的工具。
限制與未來研究方向
Sli2Vol+ 方法需要額外標註每個訓練體積中的一個切片,雖然標註成本相對較低,但未來可以探索利用分割基礎模型或互動式醫學影像分割工具來自動生成訓練體積的標註,以進一步降低標註成本。
未來可以進一步研究如何提升 OEG-CFN 的效能,例如探索更有效的網路架構或訓練策略。
可以將 Sli2Vol+ 方法應用於更多類型的醫學影像數據和臨床應用場景,以驗證其泛化能力和臨床價值。
統計資料
與完全監督方法相比,Sli2Vol+ 在跨域評估中顯示出不到 7 個 Dice 分數的下降。
與僅使用每個測試體積中一個標註切片的全監督方法 (FS-SS) 相比,Sli2Vol+ 的平均 Dice 分數提高了 20 多分。
與其他基於傳播的方法(VM、Sli2Vol 和 Vol2Flow)相比,Sli2Vol+ 在所有數據集上的表現均優於它們。
與 Sli2Vol 相比,Sli2Vol+ 的平均 Dice 分數提高了 5.6 分。
在使用 ScribblePrompt 生成的分割結果時,Sli2Vol+ 取得了與使用真實標籤相似的效能。
使用 OEG-CFN 學習相鄰切片和偽標籤之間的對應關係後,CT 和 MRI 數據集上的 Dice 分數分別平均提高了 3.4 和 1.9。
使用梯度增強影像生成器後,CT 和 MRI 數據集上的效能分別平均提高了 1.6 和 1.2。