近日發(fā)表在《自然醫(yī)學》上的研究結果表明，聯(lián)邦學習能夠建立強大的AI模型并在不同的醫(yī)療機構中進行推廣。這一發(fā)現(xiàn)展示了AI在能源、金融服務、制造業(yè)等領域進一步應用的前景。

由新冠疫情引發(fā)的一項多家醫(yī)院聯(lián)合發(fā)表的倡議表明，任何行業(yè)的機構都可以通過合作開發(fā)預測性AI模型，在準確性和通用性方面樹立新的標桿。

這項合作于今日發(fā)表在領先的同行評議醫(yī)療雜志《自然醫(yī)學》上，展示了保護隱私的聯(lián)邦學習技術如何創(chuàng)建強大的AI模型，這些模型能夠在不同的企業(yè)機構中運行良好，即便是在受到保密數(shù)據(jù)或稀疏數(shù)據(jù)限制的行業(yè)。

該研究的第一作者Ittai Dayan博士表示：“在AI開發(fā)過程中，當在一家醫(yī)院的數(shù)據(jù)上創(chuàng)建一個算法時，這個算法一般在任何其他醫(yī)院都不能很好地工作?！?Ittai Dayan博士在Mass General Brigham領導AI開發(fā)工作并在今年創(chuàng)立了醫(yī)療初創(chuàng)企業(yè)Rhino Health。

“但通過使用聯(lián)邦學習和來自各大洲的客觀多模態(tài)數(shù)據(jù)來開發(fā)模型，我們就能夠建立一個能夠幫助全球一線醫(yī)生的可推廣模型?！?Ittai Dayan表示。

其他大規(guī)模聯(lián)邦學習項目已在醫(yī)療行業(yè)展開，包括一項關于乳房X光檢查評估的五人研究以及制藥巨頭拜耳公司一項用于脾臟的分割解剖的AI模型訓練工作。

除了醫(yī)療行業(yè)之外，聯(lián)邦學習還可以幫助能源公司分析地震和井筒數(shù)據(jù)，幫助金融公司改進欺詐檢測模型，幫助自動駕駛汽車研究人員開發(fā)泛化至不同國家駕駛行為的AI。

聯(lián)邦學習：舉眾人之力開發(fā)AI

開發(fā)AI模型的公司和研究機構通常受到其可用數(shù)據(jù)的限制。這可能意味著較小的企業(yè)機構或小眾研究領域缺乏足夠的數(shù)據(jù)來訓練準確的預測模型。即使是大型數(shù)據(jù)集，也可能因一家企業(yè)機構的病人或客戶人口特征、特定的數(shù)據(jù)記錄方法、甚至所使用的科學設備品牌而出現(xiàn)偏差。

為了收集足夠的訓練數(shù)據(jù)來建立一個強大的、可推廣的模型，大多數(shù)企業(yè)機構需要與同行匯集數(shù)據(jù)。但在許多情況下，數(shù)據(jù)隱私法規(guī)限制在共用的超級計算機或云服務器上直接分享數(shù)據(jù)，例如病人的醫(yī)療記錄或專有數(shù)據(jù)集等。

這就是聯(lián)邦學習的用武之地。

《自然醫(yī)學》雜志發(fā)布的這項新研究被稱為EXAM（EMR CXR AI模型）。該研究由Mass General Brigham和NVIDIA主導，將五大洲的20家醫(yī)院聚集在一起訓練神經(jīng)網(wǎng)絡，該網(wǎng)絡用于預測出現(xiàn)新冠肺炎癥狀的病人在到達急診科等護理點的24和72小時后可能需要的氧氣補充水平。這是迄今為止最大、最多樣化的臨床聯(lián)邦學習研究之一。

多方合作助力AI工作

聯(lián)邦學習使EXAM的合作者能夠創(chuàng)建一個從每家參與醫(yī)院的胸部X光圖像、患者生命體征、人口特征數(shù)據(jù)和實驗室數(shù)值中學習的AI模型，并且無需訪問存儲在各家醫(yī)院私人服務器中的私人數(shù)據(jù)。

每家醫(yī)院都在本地的NVIDIA GPU上訓練同一個神經(jīng)網(wǎng)絡的副本。在訓練過程中，每家醫(yī)院只需定期向中央服務器發(fā)送更新的模型權重，中央服務器中的全局版本神經(jīng)網(wǎng)絡將它們匯總起來，形成一個新的全局模型。

這就像分享考試的答案，但并不透露任何用于得出答案的學習材料。

美國國家衛(wèi)生研究院（NIH）介入腫瘤學中心聯(lián)合作者兼總監(jiān)、臨床中心介入放射科主任Brad Wood博士表示：“EXAM研究的結果表明，能夠在不需要交換私人可識別數(shù)據(jù)的情況下在醫(yī)療領域訓練高性能和可推廣的AI模型，從而保護數(shù)據(jù)隱私?！?/p>

“這些發(fā)現(xiàn)不僅僅可以用于新冠肺炎相關預測的跨醫(yī)院模型 ，并展示了聯(lián)邦學習是該領域的一項有前途的解決方案。這為實現(xiàn)更有效和合規(guī)的大數(shù)據(jù)共享提供了框架，并且為挖掘AI深度學習在醫(yī)學領域的潛力提供支持?！?/p>

與所有參與機構共享的全局EXAM模型使AI模型的平均性能提高了16%。研究人員發(fā)現(xiàn)，與在任何單一站點訓練的模型相比，全局EXAM模型的通用性平均提高了38%。

從上圖中可以看到，對于擁有較小數(shù)據(jù)集的醫(yī)院來說，全局聯(lián)邦學習的性能提升尤其明顯。

泰國朱拉隆功大學和朱拉隆功國王紀念醫(yī)院AI醫(yī)學中心聯(lián)合主任Sira Sriswasdi表示：“聯(lián)邦學習使全球各地的研究人員能夠為一個共同的目標進行合作，開發(fā)出一個能夠從每個人的數(shù)據(jù)中學習和歸納的模型?！霸撫t(yī)院是合作開發(fā)EXAM的20所醫(yī)院之一。“通過NVIDIA GPU和NVIDIA Clara軟件，我們能夠很輕松地參與這項研究，產(chǎn)生有影響的結果?！?/p>

醫(yī)院和初創(chuàng)企業(yè)進一步開展EXAM研究

最初的EXAM研究匯集了北美、南美、歐洲和亞洲的合作者，只用了兩周訓練就實現(xiàn)了對病人氧氣需求的高質量預測 ，這一洞察可以幫助醫(yī)生確定病人需要的護理水平。

此后，該研究的合作者驗證了該AI模型能夠在建立和訓練該模型的地點之外的環(huán)境中進行推廣并表現(xiàn)良好。馬薩諸塞州的另外三家醫(yī)院（Cooley Dickinson醫(yī)院、Martha‘s Vineyard醫(yī)院和Nantucket Cottage醫(yī)院）測試了EXAM，發(fā)現(xiàn)該神經(jīng)網(wǎng)絡在獨立的、從未見過的數(shù)據(jù)上也表現(xiàn)良好。

Cooley Dickinson醫(yī)院發(fā)現(xiàn)，該模型能夠預測病人到達急診室后24小時內的呼吸機需求，其敏感性為95%，特異性超過88%。英國劍橋的Addenbrookes醫(yī)院也發(fā)現(xiàn)了類似的結果。

開發(fā)原始模型的MGH&BWH臨床數(shù)據(jù)科學中心科學主任Quanzheng Li博士表示，Mass General Brigham計劃在不久的將來部署EXAM。該醫(yī)院網(wǎng)絡與Lahey醫(yī)院和醫(yī)療中心以及英國的NIHR劍橋生物醫(yī)學研究中心一起與英偉達初創(chuàng)加速成員Rhino Health合作，使用EXAM開展前瞻性研究。

最初的EXAM模型使用過去新冠肺炎患者的記錄進行回顧性訓練，因此研究人員已經(jīng)掌握了關于患者最終需要的氧氣量的真實數(shù)據(jù)。這項前瞻性研究將AI模型應用于新入院患者的數(shù)據(jù)，朝著在現(xiàn)實環(huán)境中部署AI模型又邁出了一步。

劍橋大學醫(yī)學院放射科主任Fiona Gilbert表示：“聯(lián)邦學習具有革命性的力量，可將AI創(chuàng)新融入到臨床工作流程中。我們與EXAM的長期合作旨在使此類全球合作變得可重復和更有效，滿足臨床醫(yī)生應對復雜健康挑戰(zhàn)和未來流行病的需求。”

EXAM模型通過NVIDIA NGC軟件中心公開，供研究者使用。企業(yè)和研究機構在開始使用聯(lián)邦學習時，可以使用NVIDIA AI Enterprise軟件套件中的AI工具和框架。該套件經(jīng)過優(yōu)化，可在NVIDIA認證系統(tǒng)上運行。

責任編輯：haq