核心概念
本文提出了一種基於熵引導的弱監督細胞核偵測方法,利用點標註生成細胞核像素分佈的熵估計,並通過分水嶺算法生成實例分割結果,最終使用Mask-RCNN模型進行細胞核偵測,並取得了與全監督方法相當的效能。
論文概述
本論文提出了一種基於熵引導的弱監督細胞核偵測方法,旨在解決醫學影像分析中細胞核標註成本高昂的問題。
研究背景
細胞核偵測在病理學診斷中扮演著至關重要的角色,深度學習技術的發展為自動化細胞核偵測提供了新的途徑。然而,深度學習模型的訓練需要大量的標註數據,而細胞核的精確標註非常耗時耗力。為了解決這一問題,弱監督學習方法應運而生,其目標是利用少量標註信息(如點標註)訓練出高精度的細胞核偵測模型。
方法介紹
本論文提出的方法主要分為三個步驟:
熵引導: 利用貝葉斯分割網絡對輸入圖像進行分析,並根據點標註信息計算每個像素屬於細胞核的概率,進而得到細胞核像素分佈的熵估計。
實例轉換: 基於熵估計結果,利用自適應閾值分割、Voronoi圖分割和分水嶺算法生成細胞核的實例分割結果。
Mask-RCNN模型訓練與評估: 使用生成的實例分割結果訓練Mask-RCNN模型,並利用標準的目標檢測指標(如mAP)評估模型的細胞核偵測效能。
實驗結果
實驗結果表明,本論文提出的方法在僅使用5%的像素標註情況下,取得了與全監督方法相當的細胞核偵測效能。
總結
本論文提出了一種基於熵引導的弱監督細胞核偵測方法,有效降低了細胞核標註成本,並取得了良好的細胞核偵測效能,為醫學影像分析領域提供了一種新的思路。
統計資料
使用的數據集:PanNuke 數據集,包含 2,656 張圖像,其中 80% 用於訓練,20% 用於測試。
點標註方法:在每個細胞核內隨機選擇一個像素,並在其周圍添加一個半徑為 3 個像素的鄰域。
熵估計網絡:基於函數變分推斷的貝葉斯分割網絡。
實例分割方法:自適應閾值分割、Voronoi 圖分割和分水嶺算法。
細胞核偵測模型:Mask-RCNN,使用 ResNet50、Swin Transformer (小型) 和 Swin Transformer (完整) 作為骨幹網絡。
評估指標:Dice 相似係數、AUROC、mAP50、mAP75。