當前位置: 華文世界 > 科技

多模態AI是醫學的未來,谷歌三個新模型,Med-Gemini迎來大升級

2024-05-15科技

編輯 | 白菜葉

許多臨床任務需要了解專業數據,例如醫學影像、基因組學,這類專業知識資訊在通用多模態大模型的訓練中通常不存在。

在上一篇論文的描述中,Med-Gemini 在各種醫學成像任務上超越 GPT-4 系列模型實作了 SOTA!

在這裏,Google DeepMind 撰寫了第二篇關於 Med-Gemini 的論文。

在 Gemini 的多模態模型的基礎上,該團隊為 Med-Gemini 系列開發了多個模型。這些模型繼承了 Gemini 的核心功能,並透過 2D 和 3D 放射學、組織病理學、眼科、皮膚病學和基因組數據的微調,針對醫療用途進行了最佳化:

1、Med-Gemini-2D:能夠處理放射學、病理學、皮膚科、眼科影像;

2、Med-Gemini-3D:能夠處理 CT 影像;

3、Med-Gemini-Polygenic:能夠處理基因組「影像」。

該研究以「 Advancing Multimodal Medical Capabilities of Gemini 」為題,於 2024 年 5 月 6 日釋出在 arXiv 預印平台。

來自生物樣本庫、電子健康記錄、醫學成像、可穿戴器材、生物傳感器和基因組測序等不同來源的醫療數據正在推動多模態人工智能解決方案的開發,從而更好地捕捉人類健康和疾病的復雜性。

雖然醫學領域的人工智能主要專註於具有單一輸入和輸出類別的狹窄任務,但生成人工智能的最新進展顯示出在解決醫療環境中的多模態、多工挑戰方面的前景。

以 Gemini 等強大模型為代表的多模態生成人工智能擁有徹底改變醫療保健的巨大潛力。雖然醫學是這些新模型快速叠代的數據來源,但由於其高度專業化的數據,通用模型在醫學領域套用時通常表現不佳。

基於 Gemini 的核心功能,DeepMind 推出了 Med-Gemini 系列的三個新模型,Med-Gemini-2D、Med-Gemini-3D、Med-Gemini-Polygenic。

圖示:Med-Gemini 概述。(來源:論文)

來自 370 萬張醫學影像和病例的超過 700 萬個數據樣本用於訓練模型。使用各種視覺問答和影像字幕數據集,包括一些來自醫院的私人數據集。

為了處理 3D 數據 (CT),使用了 Gemini 影片編碼器,其中時間維度被視為深度維度。為了處理基因組數據,各種性狀的風險評分被編碼為影像中的 RGB 像素。

圖示:使用個人的 PRS 影像和人口統計資訊預測冠狀動脈疾病的範例。(來源:論文)

Med-Gemini-2D

Med-Gemini-2D 根據專家評估,為基於人工智能的胸部 X 射線(CXR)報告生成制定了新標準,超過了之前兩個獨立數據集的最佳結果,絕對優勢為 1% 和 12%,其中 AI 的正常病例報告為 57% 和 96%,異常病例報告為 43% 和 65%,與原始放射科醫生的報告相比質素「相當」甚至「更好」。

圖示:Med-Gemini-2D 在胸部 X 射線分類任務上的效能。(來源:論文)

Med-Gemini-2D 在分布胸部 X 射線分類任務上優於通用的較大 Gemini 1.0 Ultra 模型(在訓練期間看到了來自同一數據集的範例)。對於分布外的任務,效能各不相同。

圖示:Med-Gemini-2D 組織病理學影像分類效能。(來源:論文)

在組織病理學分類任務上,Med-Gemini 的表現大多優於 Gemini Ultra,但未能超越病理學特定的基礎模型。

圖示:PAD-UFES-20 分類任務的效能。(來源:論文)

在皮膚病變分類上,觀察到類似的趨勢(特定領域模型 > Med-Gemini > Gemini Ultra),盡管 Med-Gemini 與特定領域模型非常接近。

圖示:Med-Gemini-2D、Gemini Ultra 和使用眼底影像分類附加數據進行訓練的監督模型的效能比較。(來源:論文)

對於眼科分類,再次看到類似的情況。請註意,特定領域模型是在約 200 倍的數據上進行訓練的,因此相比之下,Med-Gemini 的表現相當不錯。

圖示:VOA 任務的評估詳細資訊。(來源:論文)

該團隊還評估了醫學視覺問答(VQA)方面的 Med-Gemini-2D 模型。在這裏,他們的模型在許多 VQA 任務上都非常強大,經常擊敗 SOTA 模型。Med-Gemini-2D 在 CXR 分類和放射學 VQA 方面表現良好,在 20 項任務中的 17 項上超過了 SOTA 或基線。

圖示:評估詳細說明了胸部 X 光檢查報告的生成。(來源:論文)

除了對醫學影像的簡單狹義解釋之外,作者還評估了 Med-Gemini-2D 在胸部 X 射線放射學報告生成方面的表現,並觀察到它根據放射學專家的評估實作了 SOTA!

Med-Gemini-3D

圖示:頭部 CT 體積報告生成的人工評估結果。(來源:論文)

Med-Gemini-3D 不僅僅適用於 2D 影像,還套用於自動化端到端 CT 報告生成。根據專家評估,其中 53% 的 AI 報告被認為是臨床可接受的,盡管需要進行額外的研究來滿足專家放射科醫生的報告質素,但這是第一個能夠完成此任務的生成模型。

Med-Gemini-Polygenic

最後,根據各種性狀的多基因風險評分,對 Med-Gemini-Polygenic 的健康結果預測進行了評估。該模型通常優於現有基線。

圖示:使用 Med-Gemini-Polygenic 與分布不均和分布外結果的兩個基線進行比較的健康結果預測。(來源:論文)

以下是 Med-Gemini 支持的多模態對話的一些範例!

圖示:透過開放式問答的 2D 醫學影像對話範例。(來源:論文)

在組織病理學、眼科和皮膚科影像分類中,Med-Gemini-2D 在 20 項任務中的 18 項中超越了基線,並接近特定任務的模型效能。

結語

總體而言,這項工作在通用多模態醫療人工智能模型方面取得了有益的進展,但顯然還有很大的改進空間。許多特定領域模型的效能優於 Med-Gemini,但 Med-Gemini 能夠在數據較少和更通用的方法下表現良好。有趣的是,Med-Gemini 在依賴更多語言理解的任務(例如 VQA 或放射學報告生成)上似乎表現得更好。

研究人員設想未來所有這些單項功能都整合到綜合系統中,從而執行一系列復雜的多學科臨床任務。AI 與人類一起工作,從而最大限度地提高臨床療效並改善患者的治療結果。

論文連結: https://arxiv.org/abs/2405.03162

相關內容: https://twitter.com/iScienceLuvr/status/1789216212704018469