為促進(jìn)醫(yī)學(xué)研究、保護(hù)數(shù)據(jù)隱私以及改善患者腦腫瘤識(shí)別結(jié)果,NVIDIA 攜手倫敦國(guó)王學(xué)院(King’s College London)于近日推出了首個(gè)用于醫(yī)學(xué)影像分析且具有隱私保護(hù)能力的聯(lián)邦學(xué)習(xí)系統(tǒng)( federated learning system),標(biāo)志著在醫(yī)療健康 AI 領(lǐng)域?qū)崿F(xiàn)了突破性進(jìn)展。
該技術(shù)論文在 MICCAI 大會(huì)召開(kāi)期間發(fā)布——該會(huì)議于 10 月 14 日在中國(guó)深圳拉開(kāi)帷幕,是全球最高端的醫(yī)學(xué)影像會(huì)議之一。
NVIDIA 與倫敦國(guó)王學(xué)院研究人員介紹了該技術(shù)的實(shí)施細(xì)節(jié)。
聯(lián)邦學(xué)習(xí)(federated learning)是一種能夠讓開(kāi)發(fā)者與各企業(yè)機(jī)構(gòu)利用分散在多個(gè)位置的訓(xùn)練數(shù)據(jù)對(duì)中心深度神經(jīng)網(wǎng)絡(luò)(DNN)進(jìn)行訓(xùn)練的學(xué)習(xí)范式,該方法可以支持各企業(yè)機(jī)構(gòu)針對(duì)共享模型開(kāi)展協(xié)作,而無(wú)需共享任何臨床數(shù)據(jù)。
研究人員在論文中闡述道:“聯(lián)邦學(xué)習(xí)在無(wú)需共享患者數(shù)據(jù)的情況下,即可實(shí)現(xiàn)協(xié)作與分散化的神經(jīng)網(wǎng)絡(luò)訓(xùn)練。各節(jié)點(diǎn)負(fù)責(zé)訓(xùn)練其自身的本地模型,并定期提交給參數(shù)服務(wù)器。該服務(wù)器不斷累積并聚合各自的貢獻(xiàn),進(jìn)而創(chuàng)建一個(gè)全局模型,分享給所有節(jié)點(diǎn)?!?/p>
研究人員解釋道,雖然聯(lián)邦學(xué)習(xí)可以保證極高的隱私安全性,但通過(guò)模型反演,仍可以設(shè)法使數(shù)據(jù)重現(xiàn)。為了幫助提高聯(lián)盟學(xué)習(xí)的安全性,研究人員研究試驗(yàn)了使用ε- 差分隱私框架的可行性。該框架是一種正式定義隱私損失的方法,該方法可以借助其強(qiáng)大的隱私保障性來(lái)保護(hù)患者與機(jī)構(gòu)數(shù)據(jù)。
上述突破性試驗(yàn)是基于取自 BraTS 2018 數(shù)據(jù)集的腦腫瘤分割數(shù)據(jù)實(shí)施的。BraTS 2018 數(shù)據(jù)集包含有 285 位腦腫瘤患者的 MRI 掃描結(jié)果。
該數(shù)據(jù)集旨在評(píng)估面向多模態(tài)與多級(jí)分割任務(wù)的聯(lián)邦學(xué)習(xí)算法。在客戶端一側(cè),研究團(tuán)隊(duì)改寫(xiě)了一個(gè)原本用于數(shù)據(jù)集中式訓(xùn)練的一流訓(xùn)練管道,并將其用作 NVIDIA Clara Train SDK 的一部分。
此外,研究團(tuán)隊(duì)還將 NVIDIA V100 Tensor Core GPU 用于訓(xùn)練與推理。
相比于數(shù)據(jù)集中式系統(tǒng),聯(lián)邦學(xué)習(xí)所提供的方法可以在不共享機(jī)構(gòu)數(shù)據(jù)的情況下實(shí)現(xiàn)相當(dāng)大的分割性能。此外,試驗(yàn)結(jié)果顯示,隱私保護(hù)與受訓(xùn)模型質(zhì)量之間產(chǎn)生了自然折中。而且,通過(guò)使用稀疏向量技術(shù),聯(lián)盟學(xué)習(xí)系統(tǒng)可以實(shí)現(xiàn)嚴(yán)格隱私保護(hù),且對(duì)模型性能僅產(chǎn)生合理的輕微影響。
深度學(xué)習(xí)是一種從醫(yī)學(xué)數(shù)據(jù)中自動(dòng)提取知識(shí)的強(qiáng)大技術(shù)。聯(lián)邦學(xué)習(xí)有望有效聚合各機(jī)構(gòu)從私有數(shù)據(jù)中本地習(xí)得的知識(shí),從而進(jìn)一步提高深度模型的準(zhǔn)確性、穩(wěn)健性與通用化能力。
此項(xiàng)研究為部署安全聯(lián)邦學(xué)習(xí)方面做出了巨大的推動(dòng),并將廣泛推動(dòng)數(shù)據(jù)驅(qū)動(dòng)型精準(zhǔn)醫(yī)學(xué)的進(jìn)步。
點(diǎn)擊教程鏈接:https://www.nvidia.com/en-us/events/miccai/,閱讀論文詳情。