AI News HubLIVE
站内改写1 分鐘閱讀

AMN:一種用於細胞核分割的自適應多尺度融合網絡,具有邊界和不確定性建模

提出AMN(自適應多尺度細胞核網絡),一種雙編碼器分割框架,結合Swin Transformer和ResNet-50特徵金字塔,通過學習的每通道門控機制動態加權各編碼器貢獻。在CoNIC基準上平均Dice 0.82,F1 0.68,優於八種基線模型,並在MoNuSeg上展示強泛化能力。

來源arXiv Computer Vision作者: Spoorthi M, Suja Palaniswamy

在病理圖像分析中,準確分割和分類細胞核亞型對於腫瘤分級、免疫浸潤量化及預後預測等下游任務至關重要。然而,現有方法通常單獨使用卷積神經網絡(CNN)或Transformer作為編碼器,難以同時捕捉精細的局部紋理和長程空間上下文。針對這一挑戰,研究者提出了AMN(自適應多尺度細胞核網絡),一種創新的雙編碼器分割框架。

AMN的核心創新在於聯合利用Swin Transformer和ResNet-50特徵金字塔作為雙編碼器,並通過一個學習的每通道門控機制,在每個尺度上動態權衡每個編碼器的貢獻。這種設計使得網絡能夠自適應地融合來自兩種架構的特徵,從而兼顧局部細節和全局結構。此外,AMN採用多目標損失函數進行訓練,包括類別加權焦點損失、強調陽性像素的邊界感知損失,以及一種新穎的不確定性調製分類項,後者能夠抑制過度自信的錯誤預測,提升模型的魯棒性。

在CoNIC基準數據集上,AMN對七種細胞核類別進行了評估,取得了平均Dice係數0.82和平均F1分數0.68的成績,其中在診斷上具有挑戰性的淋巴細胞類別上F1分數達到0.67。與其他八種基線模型(包括純CNN、純Transformer及近期混合架構,如U-Net、ResU-Net、DeepLabV3+、SegNet、ViT-Small、HmsU-Net、ConvFormer-UNet和BEFUnet)相比,AMN在所有指標上均表現更優。進一步的跨數據集評估顯示,AMN在MoNuSeg數據集上無需重新訓練即可實現強大的泛化能力,驗證了其學習表示在領域遷移中的魯棒性。

這項研究表明,通過精心設計的雙編碼器融合與多任務學習策略,可以顯著提升病理圖像中細胞核分割的準確性和可靠性,為臨牀診斷和醫學研究提供有力工具。