使用機(jī)器學(xué)習(xí)創(chuàng)建計(jì)算機(jī)生成的X射線影像以增強(qiáng)AI的訓(xùn)練集

AI 特別是深度學(xué)習(xí)在提升醫(yī)療診斷的速度和準(zhǔn)確率上擁有相當(dāng)?shù)臐摿?，但在臨床醫(yī)師可以駕馭 AI 的能力識(shí)別醫(yī)學(xué)圖像（例如 X 射線影像）中反映的疾病之前，他們需要先對(duì)模型進(jìn)行大量訓(xùn)練，而罕見疾病標(biāo)記圖像的匱乏一直是深度學(xué)習(xí)醫(yī)療的一大障礙?，F(xiàn)在，多倫多大學(xué)教授 Shahrokh Valaee 和他的團(tuán)隊(duì)設(shè)計(jì)了一種 AI 醫(yī)療領(lǐng)域中的新方法：使用生成對(duì)抗網(wǎng)絡(luò)創(chuàng)建的 X 射線影像來增強(qiáng) AI 訓(xùn)練集。實(shí)驗(yàn)表明，通過數(shù)據(jù)增強(qiáng)，無論對(duì)常見疾病還是罕見疾病，系統(tǒng)的識(shí)別準(zhǔn)確率都有大幅提高。

Shahrokh Valaee 教授和博士生 Hojjat Salehinejad 正在使用機(jī)器學(xué)習(xí)創(chuàng)建虛擬 X 射線胸透影像來訓(xùn)練 AI 系統(tǒng)識(shí)別罕見疾病。攝影：Jess MacInnis

由于以監(jiān)督學(xué)習(xí)方式訓(xùn)練 AI 系統(tǒng)的醫(yī)學(xué)圖像非常稀少，識(shí)別醫(yī)學(xué)圖像中的罕見疾病對(duì)于研究者而言一直是很有挑戰(zhàn)的問題。

Shahrokh Valaee 教授和他的團(tuán)隊(duì)設(shè)計(jì)了一種新方法：使用機(jī)器學(xué)習(xí)創(chuàng)建計(jì)算機(jī)生成的 X 射線影像以增強(qiáng) AI 的訓(xùn)練集。

在每個(gè)象限的左側(cè)是一張真實(shí)的病人 X 射線胸透影像，其旁邊是由 DCGAN 生成的合成 X 射線影像。在 X 射線影像的下方是對(duì)應(yīng)的熱圖，是機(jī)器學(xué)習(xí)系統(tǒng)觀察影像后得到的。圖源：Hojjat Salehinejad/MIMLab

「在某種意義上，我們?cè)谑褂脵C(jī)器學(xué)習(xí)來做機(jī)器學(xué)習(xí)，」多倫多大學(xué) Edward S. Rogers 高級(jí)電氣與計(jì)算機(jī)工程系（ECE）教授 Valaee 說。「我們正在創(chuàng)建模擬的但反映了特定罕見疾病的 X 射線影像，從而我們可以將它們和真實(shí) X 射線影像結(jié)合起來，以得到足夠規(guī)模的數(shù)據(jù)庫(kù)，來訓(xùn)練神經(jīng)網(wǎng)絡(luò)在其它 X 射線影像上識(shí)別這些疾病?！?/p>

Valaee 是醫(yī)學(xué)實(shí)驗(yàn)室機(jī)器智能（MIMLab）的成員，MIMLab 是一個(gè)由醫(yī)師、科學(xué)家和工程研究員組成的團(tuán)隊(duì)，致力于將他們的專業(yè)知識(shí)結(jié)合到圖像處理、人工智能和醫(yī)學(xué)中以解決醫(yī)學(xué)挑戰(zhàn)?！窤I 在醫(yī)療領(lǐng)域中有無數(shù)種方式能提供幫助，」Valaee 說，「為此我們需要大量數(shù)據(jù)。但對(duì)于某些罕見疾病，并沒有讓這些系統(tǒng)正常工作而必要的數(shù)千張標(biāo)記圖像?！?/p>

為了創(chuàng)建這些人工 X 射線影像，該團(tuán)隊(duì)使用了稱為深度卷積生成對(duì)抗網(wǎng)絡(luò)（DCGAN）的 AI 技術(shù)，來生成和不斷提升模擬圖像。GAN 是一類由兩個(gè)網(wǎng)絡(luò)構(gòu)成的算法：一個(gè)網(wǎng)絡(luò)生成圖像，另一個(gè)嘗試將合成圖像和真實(shí)圖像區(qū)分開來（判別器）。這兩個(gè)網(wǎng)絡(luò)被持續(xù)訓(xùn)練直到判別器無法將合成圖像和真實(shí)圖像區(qū)分。一旦創(chuàng)建了足夠多的人工 X 射線影像，它們就和真實(shí) X 射線影像結(jié)合起來以訓(xùn)練一個(gè)深度卷積神經(jīng)網(wǎng)絡(luò)（即分類器），然后該分類器用于對(duì)其它圖像進(jìn)行診斷（正?；蛴屑膊。?。

「我們的實(shí)驗(yàn)表明，由 DCGAN 生成的人工數(shù)據(jù)可以用于增強(qiáng)真實(shí)數(shù)據(jù)集，」Valaee 說。「這為我們提供了更多的訓(xùn)練數(shù)據(jù)，并提升了這些系統(tǒng)在識(shí)別罕見疾病上的性能。」

MIMLab 將用增強(qiáng)數(shù)據(jù)集獲得的識(shí)別準(zhǔn)確率和原始數(shù)據(jù)集獲得的準(zhǔn)確率進(jìn)行對(duì)比，發(fā)現(xiàn)對(duì)于常見疾病，其識(shí)別準(zhǔn)確率提高了 20%。對(duì)于某些罕見疾病，準(zhǔn)確率提高了約 40%，并且由于合成的 X 射線影像不是源于真實(shí)的個(gè)體，該數(shù)據(jù)集可以輕易地提供給醫(yī)院外的研究者，而不會(huì)侵犯?jìng)€(gè)人隱私。

「這實(shí)在令人激動(dòng)，通過證實(shí)這些增強(qiáng)數(shù)據(jù)集幫助提高了分類準(zhǔn)確率，我們已經(jīng)可以克服將人工智能應(yīng)用到醫(yī)療中的一大障礙，」Valaee 說。「深度學(xué)習(xí)僅當(dāng)訓(xùn)練數(shù)據(jù)足夠多的時(shí)候才有效，而數(shù)據(jù)增強(qiáng)是可以確保神經(jīng)網(wǎng)絡(luò)能以高精度分類圖像的一種方式。