郁 松,廖文浩
(中南大學(xué)計算機學(xué)院,湖南 長沙 410075)
阿爾茲海默癥AD(Alzheimer’s Disease)是最常見的癡呆癥,占癡呆癥病例的60%~70%,其發(fā)病機制尚不明確,一般都是到了晚期才被發(fā)現(xiàn),而且沒有任何治療可以阻止或逆轉(zhuǎn)其進展,只能通過藥物治療來減緩認(rèn)知衰退的進展[1 - 4]??焖僭鲩L的老齡人口和沉重的照護負(fù)擔(dān),成為當(dāng)下社會不得不面對的緊迫問題。核磁共振成像MRI(Magnetic Resonance Image )是一種被廣泛接受的非侵入性技術(shù),其通常用于臨床AD的輔助診斷[5,6]。然而,傳統(tǒng)的AD診斷不僅耗時耗力,而且無法實現(xiàn)精確辨別,特別是對于早期阿爾茲海默癥患者。此外,現(xiàn)有的方法或工具能夠識別放射科醫(yī)師無法輕易檢測到的AD患者的細(xì)微變化,因此,迫切需要可信而有效的方法或工具來篩查正常人和AD患者。
隨著深度學(xué)習(xí)技術(shù)的快速發(fā)展,特別是卷積神經(jīng)網(wǎng)絡(luò)CNN(Convolutional Neural Network)技術(shù)在圖像處理和自然語言處理等領(lǐng)域取得的顯著成效,越來越多的領(lǐng)域都開始嘗試使用深度學(xué)習(xí)技術(shù)來解決問題。在深度學(xué)習(xí)領(lǐng)域,He等人[7]提出的ResNet模型很好地解決了神經(jīng)網(wǎng)絡(luò)的梯度消失或爆炸等問題,并且加快了網(wǎng)絡(luò)模型的收斂。Selvaraju等人[8]嘗試運用類激活映射CAM(Class Activation Mapping)技術(shù)來反向解釋卷積神經(jīng)網(wǎng)絡(luò),并取得了不錯的成效。近年來,很多學(xué)者采用深度學(xué)習(xí)技術(shù)來進行AD分類。Fulton等人[9]通過50層的殘差網(wǎng)絡(luò)對核磁共振成像數(shù)據(jù)進行分類來預(yù)測阿爾茲海默癥,在OASIS(Open Access Series of Imaging Studies)數(shù)據(jù)集上實現(xiàn)了98.99%的準(zhǔn)確性。Khagi等人[10]構(gòu)造了一個層數(shù)較淺的卷積神經(jīng)網(wǎng)絡(luò)模型對AD進行分類,并在OASIS數(shù)據(jù)集上實現(xiàn)了98.51%的準(zhǔn)確性。Maqsood等人[11]通過預(yù)訓(xùn)練來微調(diào)卷積神經(jīng)網(wǎng)絡(luò)中的AlexNet模型,并使用微調(diào)后的模型對阿爾茲海默癥的核磁共振影像進行分類,該模型采用未分割的圖像進行訓(xùn)練和測試,并在OASIS數(shù)據(jù)集上進行評估,取得了92.85%的準(zhǔn)確性。Luo等人[12]采用卷積神經(jīng)網(wǎng)絡(luò)模型對AD的核磁共振影像進行識別,實驗表明該模型具有較高的AD識別精度,敏感性為100%,特異性為93%。Cheng等人[13]提出構(gòu)建級聯(lián)的3D卷積神經(jīng)網(wǎng)絡(luò)(3D-CNN),以分層學(xué)習(xí)多層圖像特征,并將這些特征集合在一起來對AD進行分類,實驗結(jié)果表明,所提出的方法對AD和正常對照組NC(Normal Control)的分類精度達(dá)到92.2%。Ju等人[14]采用深度學(xué)習(xí)對阿爾茲海默癥進行早期分類,并建立有針對性的自動編碼器網(wǎng)絡(luò),以區(qū)分正常衰老和輕度認(rèn)知障礙(AD的早期階段),所提出的深度學(xué)習(xí)方法可將預(yù)測精度提高約31.21%,并且在最佳情況下可使標(biāo)準(zhǔn)偏差降低51.23%。Payan等人[15]提出了一種深度學(xué)習(xí)方法,包括稀疏自動編碼器和3D卷積神經(jīng)網(wǎng)絡(luò),它可以基于MRI掃描預(yù)測患者的疾病狀態(tài),在預(yù)測AD大腦和健康大腦的數(shù)據(jù)時,達(dá)到了95%的準(zhǔn)確性。Liu等人[16]提出了一種用于多類AD診斷的多模態(tài)神經(jīng)影像特征提取模型,它開發(fā)了一個深度學(xué)習(xí)框架,使用零屏蔽策略來保留成像數(shù)據(jù)中編碼的所有可能信息,實現(xiàn)了87%的高精度。Sarraf 等人[17]設(shè)計了深度學(xué)習(xí)模型用于區(qū)分阿爾茲海默癥的MRI和fMRI與特定年齡組的正常人數(shù)據(jù),其幾乎完美地區(qū)分了阿爾茨海默證患者和正常人的大腦。 Hosseini等人[18]提出用3D卷積神經(jīng)網(wǎng)絡(luò)預(yù)測AD,該網(wǎng)絡(luò)建立在3D卷積自動編碼器上,該自動編碼器經(jīng)過預(yù)先訓(xùn)練,可捕獲結(jié)構(gòu)性腦MRI掃描中的解剖形狀變化,在沒有進行顱骨剝離預(yù)處理的MRI數(shù)據(jù)集上的實驗表明,它的準(zhǔn)確性優(yōu)于幾種傳統(tǒng)的分類器。Vu等人[19]組合稀疏自動編碼器和卷積神經(jīng)網(wǎng)絡(luò)用于多模態(tài)學(xué)習(xí)。
基于卷積神經(jīng)網(wǎng)絡(luò)的阿爾茲海默癥的分類已然成為一大趨勢。然而,現(xiàn)有的基于卷積神經(jīng)網(wǎng)絡(luò)的阿爾茲海默癥分類方法大多只是簡單地將在自然圖像上分類效果較好的卷積神經(jīng)網(wǎng)絡(luò)模型遷移到阿爾茲海默癥的分類,其模型并沒有針對阿爾茲海默癥的核磁共振影像的特點進行改進,且使用的數(shù)據(jù)集規(guī)模較少,因此基于卷積神經(jīng)網(wǎng)絡(luò)的阿爾茲海默癥分類方法還有很大的提升空間。
本文主要研究的是基于3D-ResNet的阿爾茲海默癥分類。通過分析現(xiàn)有阿爾茲海默癥分類方法的不足,本文在2D-ResNet模型的基礎(chǔ)上進行改進,并設(shè)計了一種3D-ResNet模型用于阿爾茲海默癥的分類。首先,對阿爾茲海默癥的核磁共振影像進行預(yù)處理,以生成核磁共振影像的灰質(zhì)GM(Gray Matter)和白質(zhì)WM(White Matter)的分割圖;接著,用預(yù)處理后的核磁共振影像進行模型的端到端的訓(xùn)練和驗證;最后,用訓(xùn)練好的模型進行阿爾茲海默癥的分類。本文主要的工作如下:
(1)設(shè)計了一個3D-ResNet模型用于阿爾茨海默癥分類。
(2)采用類激活映射(CAM)技術(shù)來可視化與AD相關(guān)的腦部區(qū)域。
現(xiàn)有阿爾茲海默癥的分類方法大致可以劃分為人工分類、半自動化分類和自動化分類3種。人工分類阿爾茲海默癥的方式帶有主觀因素。當(dāng)醫(yī)生長時間觀看影像時會感覺到眼睛疲憊,容易導(dǎo)致阿爾茲海默癥患者的漏診或誤診的現(xiàn)象發(fā)生。此外,醫(yī)生通過肉眼分析阿爾茲海默癥的核磁共振影像,對早期阿爾茲海默癥的細(xì)微病理變化不敏感,很難及時檢查出早期的阿爾茲海默癥患者?;跈C器學(xué)習(xí)的阿爾茲海默癥分類的特征提取器是依據(jù)人工經(jīng)驗設(shè)計的,容易導(dǎo)致所提取的特征與輸出任務(wù)弱相關(guān),具有較強的主觀因素,不能充分反映圖像的本質(zhì)特征,進而使最終分類的結(jié)果不夠準(zhǔn)確,產(chǎn)生漏診或誤診的現(xiàn)象。相比基于人工的阿爾茲海默癥分類方法和基于機器學(xué)習(xí)的阿爾茲海默癥分類方法而言,基于卷積神經(jīng)網(wǎng)絡(luò)的阿爾茲海默癥分類方法無需依據(jù)人工經(jīng)驗來提取特征,模型通過訓(xùn)練可以自動從阿爾茲海默癥的核磁共振影像中提取特征,而且所提取的特征與分類任務(wù)高度相關(guān),這也是本文選擇采用卷積神經(jīng)網(wǎng)絡(luò)技術(shù)對阿爾茲海默癥的核磁共振影像進行分類的主要原因之一。
由于卷積神經(jīng)網(wǎng)絡(luò)中的ResNet模型的收斂速度快和收斂效果較理想,因此本文在阿爾茲海默癥的分類問題上引入卷積神經(jīng)網(wǎng)絡(luò)技術(shù)中經(jīng)典的ResNet分類模型。在基于殘差模塊的ResNet模型的基礎(chǔ)上進行修改是為了避免模型訓(xùn)練過程中出現(xiàn)梯度消失問題,如式(1)和式(2)所示:
?Loss/?X1=?FL(XL,WL,bL)/
?XL*…*?F2(X2,W2,b2)/?X1
(1)
?XL/?Xl=(?Xl+?FL(XL,WL,bL))/?Xl=
1+?FL(XL,WL,bL)/?Xl,l∈(1,L)
(2)
其中,?代表求導(dǎo)符號,XL代表第L層的特征圖,WL代表第L層的權(quán)重參數(shù),bL代表第L層的偏置參數(shù),F(xiàn)L(XL,WL,bL)表示XL,WL,bL組成的函數(shù)。
當(dāng)網(wǎng)絡(luò)模型非常深的時候,越往前傳的梯度值就越小,趨近于0,不采用殘差模塊的模型容易出現(xiàn)梯度消失的問題,而采用殘差模塊的模型永遠(yuǎn)不會產(chǎn)生梯度消失的問題,因為回傳的梯度值在1左右。此外,深度神經(jīng)網(wǎng)絡(luò)通常比淺層神經(jīng)網(wǎng)絡(luò)的學(xué)習(xí)能力更強。只是簡單地堆疊卷積層以構(gòu)建深度神經(jīng)網(wǎng)絡(luò)不是一種行之有效的方法,因為隨著神經(jīng)網(wǎng)絡(luò)中的層數(shù)增加,梯度的傳播將受到顯著阻礙。然而,基于殘差模塊的ResNet模型的深度增加不受約束,這樣可以大大提升模型的擬合能力,而且采用殘差模塊的ResNet模型的損失函數(shù)表面非常光滑,如圖1所示。其中,沒有采用殘差模塊的模型損失函數(shù)(如圖1a所示)表面非常崎嶇,這樣很容易使模型收斂于局部最優(yōu)值,從而導(dǎo)致模型的性能較差。采用殘差模塊的模型損失函數(shù)(如圖1b所示)表面非常光滑,這樣既能加快模型的收斂速度,還能使模型收斂得較好。
Figure 1 Loss function surfaces of the models圖1 模型的損失函數(shù)表面
ResNet模型最先應(yīng)用于2D圖像分類,然而,對于像MRI這樣的3D醫(yī)學(xué)圖像數(shù)據(jù),空間信息在阿爾茲海默癥的分類中起著重要作用。3D深度學(xué)習(xí)模型可以從3D MRI圖像中更好地識別解剖位置和病理特征。為了充分利用3D MRI中的上下文信息,本文擴展了原始ResNet支持體積數(shù)據(jù)的能力,使用體積級3D-ResNet代替切片級2D-ResNet。具體來說,本文改進了原始ResNet模型的殘差模塊,將所有2D版本的卷積層、批量標(biāo)準(zhǔn)化層和池化層調(diào)整為3D版本的,如圖2所示,圖2a為原始的2D殘差模塊,圖2b為本文提出的3D殘差模塊。3D殘差模塊的流程如算法1所示,其中,σ表示采用激活函數(shù)進行非線性操作,BN表示歸一化操作。
Figure 2 Residual module圖2 殘差模塊
算法13D殘差模塊
輸入:特征圖x。
輸出:特征圖xr。
步驟13D的1*1*1卷積操作:通道的降維和信息融合。
x11=σ(BN(conv1*1*1(x)))
步驟23D的3*3*3卷積操作:特征提取。
x3=σ(BN(conv3*3*3(x11)))
步驟33D的1*1*1卷積操作:通道的升維和信息融合。
x12=σ(BN(conv1*1*1(x3)))
步驟4殘差特征圖和輸入特征圖的信息融合,特征圖的逐像素相加。
xr=x+x12
本文以3D殘差模塊為基礎(chǔ),構(gòu)建了一種3D-ResNet算法用于AD分類,如圖3所示,其主要由預(yù)處理和3D-ResNet-101 2部分組成,其中101表示模型有101層。預(yù)處理部分負(fù)責(zé)將原始的166*256*256 MRI圖像分割成121*145*124的MRI灰質(zhì)圖像和121*145*121的MRI白質(zhì)圖像;3D-ResNet-101部分主要由一個7*7*7卷積層、一個3*3*3的池化層、一系列對應(yīng)不同特征通道數(shù)的3D殘差模塊和全連接層組成。3D-ResNet-101部分將預(yù)處理后的灰質(zhì)圖像和白質(zhì)圖像疊加在一起進行端到端的訓(xùn)練,其中*3,*4,*6,*3分別代表3D殘差模塊的個數(shù),并以二分類的形式給出模型的預(yù)測結(jié)果,NC和AD分別表示正常人和阿爾茲海默癥患者的分類結(jié)果。3D-ResNet算法如算法2所示。
Figure 3 Classification model structure for Alzheimer’s disease based on 3D-ResNet-101圖3 基于3D-ResNet-101的阿爾茲海默癥分類模型的基本結(jié)構(gòu)
算法23D-ResNet算法
輸入:特征圖x。
輸出:分類結(jié)果NC或AD。
步驟13D的7*7*7的卷積操作:特征提取。
x1=σ(BN(conv7*7*7(x)))
步驟23D的3*3*3的池化操作:減小特征圖分辨率。
x2=pooling3*3*3(x1)
步驟3輸出殘差模塊操作得到的特征圖。
步驟3.1輸出經(jīng)過3個殘差模塊操作的特征圖。
x3=R(x2)*3
步驟3.2輸出經(jīng)過4個殘差模塊操作的特征圖。
x4=R(x3)*4
步驟3.3輸出經(jīng)過23個殘差模塊操作的特征圖。
x5=R(x4)*23
步驟3.4輸出經(jīng)過3個殘差模塊操作的特征圖。
x6=R(x5)*3
步驟4用3D的全連接操作進行分類。
c_vector=(PNC,PAD)=FC(x6)
步驟5取c_vector向量最大概率值所在下標(biāo)對應(yīng)的類別。
class=max(c_vector)
本文使用的數(shù)據(jù)來自阿爾茨海癥神經(jīng)影像學(xué)倡議(ADNI)數(shù)據(jù)集,ADNI的主要目標(biāo)是測試是否可以將連續(xù)核磁共振影像(MRI)、PET、其他生物標(biāo)志物以及臨床和神經(jīng)心理學(xué)評估結(jié)合起來,以測量輕度認(rèn)知障礙和早期AD的進展。ADNI數(shù)據(jù)集主要包含3T類型的MRI數(shù)據(jù)和1.5T類型的MRI數(shù)據(jù)。
本文實驗選用1.5T類型的MRI數(shù)據(jù),其是由639人的掃描數(shù)據(jù)組成,其中一個人有可能有多次掃描數(shù)據(jù),掃描的數(shù)據(jù)類型分為正常人(NC)數(shù)據(jù)、輕度認(rèn)知障礙(MCI)數(shù)據(jù)和阿爾茨海默證患者(AD)3種,如表1所示,ADNI數(shù)據(jù)集中的1.5T類型的MRI數(shù)據(jù)共有3 299個,其中包含1 015個CN數(shù)據(jù),1 709個MCI數(shù)據(jù),575個AD數(shù)據(jù)。
Table 1 1.5T type MRI data in ADNI dataset表1 ADNI數(shù)據(jù)集中1.5T類型的MRI數(shù)據(jù)
本文從ADNI數(shù)據(jù)集中的1.5T類型的AD和NC的MRI數(shù)據(jù)中隨機選取了1 163個數(shù)據(jù),其中包含575個166*256*256的AD數(shù)據(jù)和588個166*256*256的NC數(shù)據(jù),如表2所示。NC和AD數(shù)據(jù)的個數(shù)比趨近于1∶1,這是因為擬合的模型往往會傾向于數(shù)據(jù)量多的那一類數(shù)據(jù),因此將各類數(shù)據(jù)設(shè)置成1∶1可以避免這種數(shù)據(jù)傾斜問題的發(fā)生。
Table 2 Data used in the experiment表2 實驗中使用的數(shù)據(jù)
本文將整個數(shù)據(jù)隨機分成3部分:訓(xùn)練集(64%)、驗證集(16%)和測試集(20%),如表3所示。其中訓(xùn)練集的數(shù)據(jù)量∶驗證的數(shù)據(jù)量∶測試集的數(shù)據(jù)集量約為6∶2∶2,按照這個比例來劃分?jǐn)?shù)據(jù)集是為了讓模型在訓(xùn)練集上得到充分的訓(xùn)練,在驗證集上進行實時的監(jiān)控,在測試集上進行實際的預(yù)測。
Table 3 Partitioning of the experimental data表3 實驗數(shù)據(jù)的劃分
本文采用的MRI數(shù)據(jù)均使用基于Matlab的SPM12軟件包進行預(yù)處理。本文引入了基于SPM12的標(biāo)準(zhǔn)化CAT12(用于SPM的計算解剖工具箱)包用于數(shù)據(jù)預(yù)處理,并選取CAT12包中的121*145*121的templates_1.50 mm圖像作為數(shù)據(jù)預(yù)處理的模板,選取CAT12包中的121*145*121的neuromorphometrics圖像作為數(shù)據(jù)預(yù)處理中區(qū)域分割的標(biāo)簽?zāi)0?。其中?biāo)簽?zāi)0鍖⒛X部一共劃分為142個腦區(qū),選取DPABI 2.2包中的ch2圖像作為原圖模板,并將其尺寸從181*217*181重置為同templates_1.50 mm的尺寸121*145*121相同大小,進而用于顯示與CAM技術(shù)生成的mask圖像的疊加。預(yù)處理的步驟包括空間配準(zhǔn)、組織分類(灰質(zhì)、白質(zhì)和腦脊液)和強度非均勻性的偏差校正。本文將預(yù)處理后AD病患和正常人生成的灰質(zhì)(GM)和白質(zhì)(WM)數(shù)據(jù)相疊加用于3D-ResNet-101模型的訓(xùn)練、驗證和測試。如圖4所示,輸入的原始MRI圖像的尺寸為166*256*256,輸出的預(yù)處理后的MRI圖像尺寸分別為121*145*121的灰質(zhì)圖像和121*145*121的白質(zhì)圖像。
Figure 4 Preprocessing of magnetic resonance imaging圖4 核磁共振影像的預(yù)處理
本文所設(shè)計的3D-ResNet算法采用的初始化方法為kaiming initialization[20],且算法中的參數(shù)都是從頭到尾訓(xùn)練,所有的實驗都是在一塊Tesla K40 GPU 上完成的,GPU的顯存約為11 GB。模型訓(xùn)練的超參數(shù)如表4所示,其中模型總的迭代次數(shù)設(shè)置為50,批量訓(xùn)練的大小為6,學(xué)習(xí)率從0.001開始,每迭代20次后,學(xué)習(xí)率除以10。
Table 4 Hyper parameters for model training表4 模型訓(xùn)練的超參數(shù)
為了評估所設(shè)計的3D-ResNet-101模型對AD分類的性能,本文將其與其他模型進行了準(zhǔn)確性(accuracy)的對比,如表5所示。accuracy的定義如式(3)所示,其反映了模型準(zhǔn)確分類AD病例和正常人的數(shù)量,accuracy越大,模型正確分類AD病例和正常人的數(shù)量就越多。
(3)
其中,TP、TN、FP和FN分別代表真陽性、真陰性、假陽性和假陰性。
Table 5 Comparison of accuracy among different models表5 不同模型的準(zhǔn)確性對比
為了進一步評估所設(shè)計的3D-ResNet-101模型對AD分類的性能,本文評估了基于3D-ResNet-101模型的敏感性(sensitivity)和特異性(specificity)指標(biāo),如表6所示。sensitivity的定義如式(4)所示,其反映了模型準(zhǔn)確分類的AD病例數(shù)量,sensitivity越大,漏診的AD病例就越少。specificity的定義如式(5)所示,其反映了模型準(zhǔn)確分類的正常人的數(shù)量,specificity越大,正常人被誤診為AD的病例就越少。
(4) Table 6 Sensitivity and specificity results of 3D-ResNet-101 model on verification set and testing set表6 3D-ResNet-101模型在驗證集和測試集上的敏感性和特異性
(5)
本文還探討了模型訓(xùn)練參數(shù)不同對分類結(jié)果的影響。當(dāng)初始學(xué)習(xí)率參數(shù)設(shè)置為較大的數(shù)0.1時,模型訓(xùn)練過程中會出現(xiàn)收斂抖動的現(xiàn)象,從而導(dǎo)致模型收斂時間延長或無法收斂。由于實驗條件的限制,批量處理的大小最大可設(shè)置為6。當(dāng)批量處理大小小于6時,模型最終同樣會收斂,但所需要的收斂時間會延長。
類激活映射(CAM)技術(shù)是將最后一個卷積層的輸出特征圖和該特征圖對應(yīng)的輸入圖像的仿射矩陣相結(jié)合,進而生成與輸入圖像尺寸相同的掩碼圖,如圖5所示,其中響應(yīng)越高的圖像區(qū)域,越能代表3D-ResNet算法的分類依據(jù)。
Figure 5 Structure of class activation mapping圖5 類激活映射技術(shù)的基本結(jié)構(gòu)
此外,本文將原始的MRI圖像進行預(yù)處理,并生成相應(yīng)的灰質(zhì)圖像和白質(zhì)圖像,然后采用CAM技術(shù)生成掩碼圖像,并將掩碼圖像中的高響應(yīng)區(qū)域可視化到標(biāo)準(zhǔn)的腦模板上,如圖6所示,其通過Matlab 2016b的DPABI Viewer來展示疊加效果,并從axial、coronal和sagittal 3個不同的方向進行具體顯示。
為了探索3D-ResNet算法的深度對AD分類的影響,本文對比了3D-ResNet-50模型和3D-ResNet-101模型在訓(xùn)練過程中accuracy和loss的變化情況,結(jié)果如圖7所示。其中3D-ResNet-101模型比3D-ResNet-50模型擬合目標(biāo)數(shù)據(jù)的效果更好,由此可以推出適當(dāng)增加3D-ResNet算法的層數(shù)能提升模型的擬合能力。
Figure 6 Visualization of Alzheimer’s disease classification based on 3D-ResNet圖6 基于3D-ResNet的阿爾茲海默癥分類的可視化
Figure 7 Model depth impact on Alzheimer’s disease classification圖7 模型的深度對阿爾茲海默癥分類的影響
Figure 8 Acquisition of 2D MRI slices圖8 2D MRI切片的獲取
此外,本文通過取MRI圖像的中間切片將3D圖像轉(zhuǎn)成2D圖像的形式,以便于采用2D神經(jīng)網(wǎng)絡(luò)模型做AD的分類。以3D單通道灰質(zhì)圖像轉(zhuǎn)2D單通道灰質(zhì)圖像為例,圖8顯示了本文將3D MRI圖像轉(zhuǎn)為2D圖像的原理,其中(37,44,37)表示分別從sagittal、coronal和axial 3個方向取121*145*121的第37個2D圖像和第44個2D圖像、第37個2D圖像,同理,類似的還有(61,73,61)和(97,116,97)。由于(61,73,61)位置的切片能將腦部輪廓全部展現(xiàn)出來,因此本文選取的2D圖像就是從該位置獲得的,并將獲取的2D圖像重置為28*28的單通道圖像。本文將axial方向的GM和WM對應(yīng)的2D圖像疊加在一起,然后輸入到2D神經(jīng)網(wǎng)絡(luò)模型進行訓(xùn)練進和預(yù)測。本文探討了使用體積級3D-ResNet代替切片級2D-ResNet的區(qū)別,如表7所示,在模型參數(shù)和訓(xùn)練時間上,體積級3D-ResNet的模型都比切片級2D-ResNet的模型增加了很多,但在準(zhǔn)確性上,體積級3D-ResNet的模型卻比切片級2D-ResNet的模型提升將近一倍。
Table 7 Difference between volume-level 3D-ResNet and slice-level 2D-ResNet表7 使用體積級3D-ResNet代替切片級2D-ResNet的區(qū)別
本文提出了一種3D-ResNet模型用于準(zhǔn)確而快速地對AD進行分類,并用accuracy、sensitivity和specificity性能指標(biāo)對3D-ResNet模型進行了評價。針對AD的發(fā)病機制尚不明確的問題,本文通過類激活映射(CAM)技術(shù)來可視化與AD相關(guān)的腦部區(qū)域。此外,本文的模型還可用于診斷其他類似的神經(jīng)疾病,如帕金森和小兒麻痹癥等。未來我們希望將本文所提出的基于3D-ResNet的AD分類算法應(yīng)用于實際生活中,并從精度和速度上進一步優(yōu)化。