Concepts de base
針對超聲圖像中病灶輪廓模糊的問題,本文提出了一種基於輪廓概率模型的醫學超聲圖像分割網絡 CP-UNet,通過多組通道移位下採樣模塊和輪廓概率建模模塊,增強網絡對輪廓的表示能力,從而提高分割精度。
Résumé
CP-UNet:基於輪廓的醫學超聲圖像分割概率模型研究
論文概述
本論文提出了一種名為 CP-UNet 的新型醫學超聲圖像分割網絡,旨在解決超聲圖像中病灶輪廓模糊的難題。該網絡採用基於輪廓的概率模型,通過多組通道移位下採樣模塊和輪廓概率建模模塊,增強網絡對輪廓的表示能力,從而提高分割精度。
研究背景
超聲成像技術因其低成本、易操作和無創性等優點,被廣泛應用於各種疾病的診斷。然而,超聲圖像固有的局限性,如病灶區域分佈不均勻、斑點噪聲和成像偽影等,增加了分割任務的難度。特別是輪廓模糊問題,包括病灶與正常組織之間邊界線不明確以及病灶形狀和紋理的不規則變化,嚴重影響了分割的準確性。
研究方法
CP-UNet 主要由三個核心模塊組成:
- 多組通道移位下採樣模塊(MgCSD): 該模塊取代了傳統的卷積下採樣,通過多組通道的空間移位加權,建立全局-局部特徵鏈接,在保留細節特徵的同時,提供更全面的特徵信息。
- 輪廓概率建模模塊(CPM): 該模塊利用混合高斯分佈擬合輪廓特徵,突破了單一損失約束的限制,建立了通用的輪廓特徵,以指導分割網絡中的輪廓表示。
- 基於門控機制的特徵過濾模塊(GF): 該模塊融合了上採樣結果、下採樣結果和相同維度的 CPM 採樣結果,並通過門控機制過濾輸入特徵,保留重要特徵,抑制無關或噪聲特徵,有效地將輪廓特徵融合到解碼階段。
實驗結果
論文在公開的乳腺超聲數據集 BUSI、甲狀腺超聲數據集 DDTI 以及一個私有的甲狀腺超聲數據集 TUI 上進行了實驗驗證。結果表明,CP-UNet 在乳腺結節和甲狀腺結節的分割精度方面均優於現有的醫學圖像分割方法,特別是在輪廓描繪方面表現出色。
研究結論
CP-UNet 通過引入基於輪廓的概率模型,有效地解決了超聲圖像中病灶輪廓模糊的問題,提高了分割精度。該方法為醫學超聲圖像分割領域提供了一種新的思路,具有重要的臨床應用價值。
Stats
在 BUSI 和 DDTI 數據集上,CP-UNet 的分割精度均優於基於卷積的基線模型,提升幅度約為 10%。
與在 ImageNet-21K 數據集上進行預訓練的 TransUNet 相比,CP-UNet 在分割指標相近的情況下,能夠更好地捕捉輪廓的鋸齒狀變化。
在數據量充足的情況下,基於注意力機制的方法比基於卷積的基線模型表現更穩定,效果更好。
CP-UNet 在較小的數據集 BUSI 上比在較大的數據集 TUI 上的改進更顯著,表明該網絡在數據量較小的情況下也能提取更有用的信息以增強分割效果。
Citations
"When the local edge line of the contour is blurred, it needs to be complemented by the global edge line characteristics, such as pixels around the edge line, contour shape, and range characteristics."
"We use the Gaussian Mixture Model to simulate the distributional features of the synthetic contours to break the limitation of a single loss constraint and establish generalized features to guide the contour representation in the segmentation network."