基于深度神經網絡的癌細胞識別系統研究

楊曉玲　王振奇　李嘉

摘 要：癌癥是發病率和死亡率極高的疾病，癌細胞正確識別與癌癥等級正確判斷具有極其重要的意義。深度神經網絡（DNN）可用神經網絡模擬大腦識別過程，底層提取初級特征，高層對底層特征進行組合與抽象。以乳腺癌細胞圖像為例，采用BreaKHis官網數據集，在Linux操作系統安裝Pycharm開發軟件，以Tensorflow為框架，搭載Python2.7編譯環境，增加現有神經網絡的卷積層數和全連接層數，提出一種優化的深度神經網絡癌細胞識別方法。實驗結果表明，該方法能更加準確地識別癌細胞圖像特征，有效降低現有神經網絡分類錯誤，對癌細胞平均識別率達89.58%，對惡性癌細胞識別率最高可達96.75%。

關鍵詞：癌細胞識別;數據集;神經網絡;訓練速度

DOI：10. 11907/rjdk. 191574

中圖分類號：TP303 ? 文獻標識碼：A??????????????? 文章編號：1672-7800（2020）003-0065-04

Research Cancer Cell Recognition System Based on Deep Neural Network

YANG Xiao-ling，WANG Zhen-qi，LI Jia

（School of Electronic Information Engineering，Zhuhai College of Jilin University，Zhuhai 519041，China）

Abstract： Cancer has become a major disease with high morbidity and mortality in China. Correct identification of cancer cells and correct judgement of cancer grade are of great significance to the development of Chinese medicine. This system employs official websites BreaKHis Data and takes breast cancer cell image as an example. Pycharm development software was installed on Linux operating system in Python2.7 compiler environment within the framework of Tensorflow to speed up network training and deepen convolutional layers and fully layers of existing neural networks. A cancer cell recognition method based on optimized deep neural network is proposed.? The experimental results show that， this method can recognize the image features of cancer cells more accurately， effectiving reduce the existing neural networks classification errors， the average recognition rate of cancer cells was 89.58%， and the highest recognition rate of malignant cancer cells was 96.75%.

Key Words： cancer cell recognition; data set; neural network; training speed

0 引言

深度學習算法是近幾年興起的特征學習及分類算法，具有強大的特征學習能力，可同時實現特征學習與分類。深度學習在癌細胞識別中的應用尚不普遍，目前最具代表性的研究是Cruz-Roa等[1]基于深度學習網絡的基底細胞癌（Basal-Cell Carcinoma Cancer）自動檢測系統。深度學習模型不僅能大幅提高圖像識別精度，而且避免消耗大量時間進行人工特征提取，使運行效率大大提升。普通神經網絡和深度神經網絡區別如圖1所示[2]。深度學習通過多個處理層組成復雜的計算模型，自動獲取數據的表示與多個抽象級別，利用深度神經網絡具有的網絡深度和大樣本量訓練集，將其應用于癌細胞識別，可以更好地表達癌細胞圖像特征并區分細胞，提高癌細胞識別率。本文以乳腺癌細胞識別為例，通過深度學習進行癌細胞識別應用，對癌癥的臨床判斷具有極其重要的意義。

1 深度神經網絡模型架構

基于經典的LeNet-5[3]構建深度神經網絡，主要包括兩個卷積層、一個池化層和兩個全鏈接層。卷積神經網絡為癌細胞識別系統的主體部分，通過網絡中的多個隱含層實現對癌細胞數據更深層次的特征提取。對現有的卷積神經網絡結構進行優化，通過增加并聯卷積層擴寬網絡寬度，生成一個訓練樣本學習系統。通過增加訓練樣本數量使網絡學到更多并且更加準確。深度神經網絡模型如圖2所示。

2 癌細胞識別模型

基于深度神經網絡的癌細胞識別系統流程分為癌細胞圖像預處理、癌細胞特征提取和分類3個部分，實現方案如圖3所示。

癌細胞圖像預處理主要對癌細胞圖像的形態學特征進行融合，將融合后的一維向量轉化為二維圖像。特征提取由深度卷積神經網絡自動完成，最后使用一種常用的分類器Softmax[4]進行分類并得到識別結果。癌細胞識別過程如圖4所示。通過加深卷積層數和全連接層數，提高訓練速度與識別率，圖4右邊為每層網絡的數據結構。

2.1 癌細胞圖像預處理

首先通過使用高階中值濾波器算法檢測癌細胞圖像噪聲，去除癌細胞圖像漂移噪聲，并利用小波變換算法，選擇合適的小波基函數和閾值去除信號干擾并重構時域信號，得到去噪后的癌細胞圖像;然后利用融合技術將癌細胞圖像的形態學特征進行融合，將特征的時間值歸一化到[0，1]之間，實現數據的歸一化處理;最后將一維特征融合向量轉化為二維的二值圖像，作為卷積神經網絡的輸入，利用跳白格和游程編碼對二值圖像進行降維。

2.2 癌細胞圖像特征提取

癌細胞圖像特征提取步驟如下：①對特定類型癌細胞圖像通過特定選擇程序進行預提取，選擇出更具代表性的癌細胞圖像作為訓練樣本集;②構建深度卷積神經網絡模型，如圖2所示。設置模型中的特征圖片數量及各層參數（卷積核、步長等）;③確定訓練集、驗證集及樣本集中的癌細胞數量;④利用深度卷積神經網絡對輸入自動提取高層特征;⑤利用改進的深度神經網絡方法（如圖4所示）加快訓練的收斂速度、提高分類準確率。

2.3 Softmax分類

使用Softmax分類器進行特征識別，將目標變量分為多類算法。分類器利用Logistic 模型[3]對多分類問題進行推廣。假設有[N]幅輸入圖像[xi，yiNi=1]，每幅圖像標記[yi∈{1，2，？，k}，k2]共[k] 類，本文設[k=2]。對于給定的測試圖像[xi]，用假設函數估計出其屬于每個類別[j]的概率值[p（yi=j|xi）]，則假設函數[hθ（xi）]為：

式（1）中，[1j=1keθTjxi]代表對概率分布進行歸一化，全部概率之和為1。[θ]表示Softmax分類器的參數。

Softmax分類器損失函數為：

其中[1（yi=j）]為指示性函數，其取值規則為：1{值為真的表達式}=1，1{值為假的表達式}=0。最后通過隨機梯度下降法得到最小化誤差loss函數。

3 方案實現與實驗結果分析

3.1 數據集

本文采用BreaKHis官網公開的數據集[5]，該數據集包含82位患者的7 909幅已標注乳腺癌病理圖像，其中良性腫瘤圖像2 480幅，惡性腫瘤圖像5 429幅。每幅癌細胞圖像均采用4種不同的放大倍數（40X、100X、200X、400X），固定大小為700×460像素，模式為RGB三通道圖像（24位顏色，每個通道8位）。不同放大倍數的良、惡性腫瘤圖像分布情況如表1所示。

不同放大倍數的良性和惡性癌細胞圖例如圖5所示，肉眼可以直觀看到：隨著放大倍數的增大，良性癌細胞和惡性癌細胞的細胞核特征區別很大，惡性癌細胞明顯呈現顏色深等特點。

3.2 系統設計與實現

首先讀取圖像數據和標簽數據，對封裝的數據加載并進行亂序處理，如果不進行亂序處理會有某種特征數據連續出現從而影響訓練效果。把相關數據抽象為數據類進行處理，包括圖像信息、標簽信息、圖像總數據、數據批次;然后對所得圖像進行訓練或預測，將圖像數據傳入卷積層，經過處理得到壓縮的圖像數據。把數據傳入全連接層，經過數據分析輸出特征，再通過Softmax函數把特征提取出來，對結果進行誤差計算以及優化，最后打印出訓練和預測結果。系統實現過程如圖6所示。

3.3 實驗結果分析

由于每個病變文件夾下都包含40X、100X、200X、400X 等幾種分辨率圖片，進行數據提取后得到訓練數據和驗證數據兩種結構的數據集，每部分包含良性和惡性兩種。兩種癌細胞的召回率、識別精確率和平均識別率如表2所示，可以看到系統對惡性癌細胞的整體識別效果更好。

隨著訓練次數增加，通過模型預測的準確率平均值是89.58%，最高可達96.75%。訓練結束后的準確率和誤差如圖7所示，可以看出，loss隨訓練次數的增大衰減很快，到訓練100次時基本衰減為0.012。第100次訓練時，訓練集的loss為0.001 1，準確率為100%;驗證集的loss為0.510 3，準確率為87.44%。

4 結語

本文以乳腺癌為例，研究了利用深度神經網絡方法實現癌細胞圖像識別系統，采用加深現有神經網絡模型的卷積層和全連層方法，使其具有更深、更復雜的結構，加快網絡訓練速度。系統對乳腺癌癌細胞圖像識別分類效果良好，識別準確率最高可達96.75%，改善了現有神經網絡分類識別率不高的問題。后續將繼續學習深度神經網絡模型，研究影響模型性能的因素，如染色對于圖片的影響、預處理方法、分類器改進等，滿足更高臨床要求。

參考文獻：

[1]Y L， Y B， G H. Deep learning [J]. Nature， 2015， 521（7）：353-436.

[2]M Y，WANG Y. Research on image classification model based on deep convolution neural network [J]. Springer，2019（1）：1186-1191.

[3]AHIALE AKOGO，DARLINGTON，PALMER XAVIER LEWIS. End- to-end learning via a convolutional neural network for cancer cell line classification[J].? ResearchGate，2018（3）：889-902.