← 返回 2026-04-08

MedGemma 1.5 技术报告:在统一架构中扩展 3D 影像、病理 WSI、时序分析与文档理解 MedGemma 1.5 Technical Report

Andrew Sellergren, Chufan Gao, Fereshteh Mahvar, Timo Kohlberger, Fayaz Jamil, Madeleine Traverse, Alberto Tono, Bashir Sadjad, Lin Yang, Charles Lau, Liron Yatziv, Tiffany Chen, Bram Sterling, Kenneth Philbrick, Richa Tiwari, Yun Liu, Madhuram Jajoo, Chandrashekar Sankarapu, Swapnil Vispute, Harshad Purandare, Abhishek Bijay Mishra, Sam Schmidgall, Tao Tu, Anil Palepu, Chunjong Park, Tim Strother, Rahul Thapa, Yong Cheng, Preeti Singh, Kat Black, Yossi Matias, Katherine Chou, Avinatan Hassidim, Kavi Goel, Joelle Barral, Tris Warkentin, Shravya Shetty, Dale Webster, Sunny Virmani, David F. Steiner, Can Kirmizibayrak, Daniel Golden 📅 2026-04-06 👍 15 2026-07-13 08:36
3D 影像 医疗 AI 多模态大模型 文档理解 病理 WSI 知识蒸馏

Google 4B 医疗多模态模型,新增 3D 影像与时序分析能力

前置知识

视觉-语言模型(VLM)

把视觉编码器(如 SigLIP)与语言解码器(如 Gemma)联合训练的多模态架构,通过对比学习与指令微调让模型同时理解图像与文本。

MedGemma 1.5 由 Gemma 3 语言骨干加 400M MedSigLIP 视觉编码器组成,看懂 VLM 是理解本报告架构与训练流程的前提。

知识蒸馏(Distillation)

用大教师模型的输出分布作为软标签训练小模型。本报告每 token 采样 256 个教师 logits,按教师概率加权后用交叉熵损失让学生逼近教师分布。

蒸馏是 MedGemma 1.5 后训练三大支柱之一(继续预训练/蒸馏/RL),对医学领域任务的能力提升至关重要。

全切片病理图像(WSI)

数字病理切片单图可达数十亿像素,通常先做组织分割,再切成固定大小 patch(如 $896 \times 896$)后送入模型。

WSI 是 MedGemma 1.5 新增的核心模态,本报告提出的 patch 采样与随机缩放策略是其病理能力跃升的关键。

Hounsfield Unit(HU)窗宽窗位

CT 体素值的标准化单位。由于视觉编码器只接受 256 级强度,需把 HU 值经多通道窗宽映射到 RGB 三通道才能保留组织对比。

MedGemma 1.5 用 R$(-1024, 1024)$、G$(-135, 215)$、B$(0, 80)$ 三通道分别兼顾肺、纵隔、脑实质,是 3D CT 性能的预处理基石。

强化学习后训练(RL)

用奖励信号微调语言模型使其输出更符合期望。本报告在 CXR、CT、MRI、WSI 等多种医学任务上同时施加 RL,对齐临床视觉任务。

RL 与蒸馏共同贡献了 MedGemma 1.5 在高维影像任务上的提升,是 WSI ROUGE-L 从 2.2% 跃升至 49.4% 的关键手段。

研究动机

现有开源医疗多模态模型几乎都只能处理 2D 影像与文本,难以应对真实临床的高维数据:CT/MRI 体数据包含数十至上百层 2D 切片,单独切片处理会丢失 3D 空间上下文;全切片病理图像(WSI)单张可达数十亿像素,patch 采样策略不成熟;多时点胸片分析需要时间序列推理能力;PDF 化验单的结构化抽取对文档理解提出新要求。MedGemma 1 发布时仅支持 2D 影像,在 3D MRI 分类上仅 51.3%、WSI 报告生成 ROUGE-L 仅 2.2%、解剖定位 IoU 仅 3.1%,远达不到临床可用水平;而专攻单一模态的模型又缺乏统一的通用接口,开发者集成成本极高。

本文的目标是本文发布 MedGemma 1.5 4B,目标是在单一 4B 参数架构中同时原生支持四大新能力:3D 放射学解读(CT/MRI 体数据)、病理 WSI 报告生成、胸部 X 光精细解剖定位(bounding box)、多时点胸片纵向分析;同时显著提升医学文档(化验单 PDF/EHR)结构化抽取与文本临床推理能力,使模型成为医疗多模态开发者社区的通用可微调基座。报告还设置了量化目标:在 MRI 分类上相对前代提升 10 个百分点以上、WSI 报告生成 ROUGE-L 提升 40 个百分点以上、解剖定位 IoU 提升 30 个百分点以上,且不显著回退现有基准。

与已有工作不同的是,现有工作要么专攻单一高维模态(如 PathChat、BioViL-T),要么是通用多模态模型但不支持 3D/WSI/时序。MedGemma 1.5 的独特切入点是:在保持 4B 轻量级可部署规模的前提下,通过长上下文 3D 切片采样(最多 85 张)、WSI 随机缩放 patch 采样(最多 126 张)与多通道 CT 窗宽映射,把多种高维模态塞进同一组视觉 token 预算($\le 32K$ token),并用三阶段后训练(继续预训练 + 蒸馏 + RL)一次性对齐所有模态,是首个在单一架构中同时达成这些基线能力的开源模型。

核心方法

MedGemma 1.5 4B 沿用 MedGemma 1 的架构:400M MedSigLIP 视觉编码器 + Gemma 3 语言骨干,视觉侧冻结,语言侧继续训练。整体思路是"冻结视觉、扩数据、改预处理、堆后训练"。新增训练数据涵盖 605,732 张印度医院胸片、282,963 例 CT、167,674 例 MRI、335,825 张 WSI、25,560 + 87,879 张皮肤图像与 4 个 EHR/化验单数据集。3D 体数据切成最多 85 张 $896 \times 896$ 轴位切片(21,760 视觉 token)按 z 轴等距采样;WSI 在 5x/10x/20x 三档放大率(概率 0.34/0.33/0.33)随机选一档再切最多 126 个 $896 \times 896$ patch(32,256 token)。后训练采用 Gemma 3 同款配方(PT → Distill → RL),蒸馏时每 token 采样 256 个教师 logits 加权求 KL 散度,并新增领域专用教师(CT/MRI/WSI 各一名)增强高维模态。

核心创新点是把"高维医学影像如何编码进固定大小上下文"这一工程难题做了系统化处理:一是用三通道 HU 窗宽映射($R:\,[-1024,1024]$、$G:\,[-135,215]$、$B:\,[0,80]$)让 2D 视觉编码器同时感知肺、纵隔、脑三种解剖区域,把 3D 切片序列塞进语言模型上下文窗口;二是对 WSI 设计"低分辨率组织分割 → 高分辨率随机放大率 → 等距网格切 patch → 固定 126 个 patch"流水线,并显式保留空间顺序;三是引入领域专用教师蒸馏(在 CT Dataset 1、MRI Dataset 1、WSI 上各训一名专门教师),让 4B 学生能逼近远超自身容量的专科模型。

方法步骤详情

训练流程分四步:(1)继续预训练,冻结 MedSigLIP,把上述新增图像-文本对用监督微调方式继续训练 LLM,目标函数为标准交叉熵 $\mathcal{L}_{PT}=-\sum_i \log p_\theta(y_i|x_i)$;(2)多教师蒸馏,对通用大教师与领域专用教师各采样 256 个 logits/token,按教师概率 $p_T$ 加权得到软标签,最小化 $\mathcal{L}_{distill}=-\sum_i p_T(y_i)\log p_\theta(y_i)$;(3)强化学习,在 CXR、CT、MRI、WSI 等多任务奖励信号上做 RLHF;(4)评估时采用 temperature $=0.0$,所有数据严格在训练集外。预处理细节上,3D 体数据先按 $512\times 512$、轴位、等层厚、$\ge 5$ 张筛选,再限制总切片数 $\le 85$,超出则 z 轴等距采样;CT 用上述三通道窗宽映射,MRI 用 min-max 归一化后三通道复制同一灰度。

技术新颖性

与同期工作相比,新颖性体现在三方面:第一,把"切片序列 + 文本报告"当成单一长上下文输入而非独立样本,使模型在 3D MRI 分类上获得 11% 绝对提升($51.3\%\to 64.7\%$);第二,WSI 随机放大率采样策略在保留空间顺序的前提下控制视觉 token 数,使 WSI 报告生成 ROUGE-L 取得 47% 巨幅提升($2.2\%\to 49.4\%$);第三,把解剖定位(bounding box)作为原生输出能力,通过 IoU 目标微调使 Chest ImaGenome IoU 从 3.1% 飙升至 38.0%,首次让一个 4B 模型同时具备空间定位与时序对比推理能力。

Overview of MedGemma 1.5 3D radiology capability
Figure 3: Overview of MedGemma 1.5 3D radiology capability

实验结果

核心发现分四组:(1)3D 影像:MRI Dataset 1 分类准确率 $51.3\%\to 64.7\%$(+11%),CT Dataset 1 $58.2\%\to 61.1\%$(+3%);(2)WSI:内部 WSI Histopath 报告生成 ROUGE-L 从 $2.2\%$ 跃升至 $49.4\%$(+47%),逼近 PolyPath SOTA 的 $49.8\%$;(3)解剖定位:Chest ImaGenome 平均 IoU 从 $3.1\%$ 飙升至 $38.0\%$(+35%),与 Gemini 3 Pro 的 $39.1\%$ 持平;(4)文本与时序:MS-CXR-T 多时点 macro accuracy $61.1\%\to 65.7\%$(+4%),MedQA $64.4\%\to 69.1\%$(+5%),EHRQA $67.6\%\to 89.6\%$(+22%),4 个化验单抽取平均 macro F1 提升至 78%,MedXpertQA 多模态推理从 $18.8\%$ 升至 $26.4\%$,超过同尺寸 Qwen3 VL 4B 的 $21.9\%$。

Additional training data for MedGemma 4B 1.5 relative to MedGemma 1
Table 1: Additional training data for MedGemma 4B 1.5 relative to MedGemma 1
Additional evaluation datasets for MedGemma 1.5
Table 2: Additional evaluation datasets for MedGemma 1.5
MedGemma 1.5 performance for original MedGemma 1 evaluation tasks
Table 3: MedGemma 1.5 performance for original MedGemma 1 evaluation tasks
New Evaluation Task Results
Table 4: New Evaluation Task Results
Performance across medical text Q&A and diverse medical imaging tasks
Figure 2: Performance across medical text Q&A and diverse medical imaging tasks
查看结构化数据
任务指标本文基线提升
3D MRI 条件分类 (MRI Dataset 1) Accuracy 64.7% MedGemma 1 4B: 51.3% +11% (绝对), 已逼近 Gemini 3 Flash 60.3%
3D CT 条件分类 (CT Dataset 1) Accuracy 61.1% MedGemma 1 4B: 58.2% +3% (绝对), 与 Gemini 3 Pro 61.0% 持平
WSI 病理报告生成 ROUGE-L 49.4% MedGemma 1 4B: 2.2% +47% (绝对), 逼近 PolyPath SOTA 49.8%
胸部 X 光解剖定位 Mean IoU 38.0% MedGemma 1 4B: 3.1% +35% (绝对), 略胜 Gemini 3 Flash (38.5%)
多时点胸片分类 (MS-CXR-T) Macro Accuracy 65.7% MedGemma 1 4B: 61.1% +4% (绝对), 优于 Gemini 3 Pro (62.9%)
医学文本 QA (MedQA) Accuracy 69.1% MedGemma 1 4B: 64.4% +5% (绝对), 但低于 Qwen3 VL 4B 76.8%
EHR 问答 (EHRQA) Accuracy 89.6% MedGemma 1 4B: 67.6% +22% (绝对), 与 Gemini 3 Pro (95.2%) 仅差 5.6%
化验单信息抽取 (EHR Dataset 4) Macro F1 64 MedGemma 1 4B: 25 +39 (绝对), 提升 2.5 倍

局限与改进

作者承认的局限:(1)模型成为"医疗通才"后在某些传统基准上轻微回退——SLAKE tokenized F1 从 72.3% 降至 59.8%(-12.5%),VQA-RAD 从 49.9% 降至 48.1%(-1.8%),PubMedQA 从 73.4% 降至 67.6%,作者认为这些基准依赖 token 重叠、答案未标准化,质量有限;(2)3D 切片数被硬限 85 张、WSI 限 126 个 patch,超长病例(如全腹部 CT)必须下采样可能丢失细节;(3)部分新基准(Qwen3 VL 4B)因内部推理框架不兼容未能完整对比。我的额外观察:CT 内部数据集仅 +3% 提升幅度远小于 MRI 的 +11%,说明头部多通道窗宽映射对肺/脑区域更友好;EHR Dataset 4 虽从 25 飙到 64,但仍只有 Gemini 3 Pro (81) 的 79%,说明小模型在稀有化验单模板上仍有较大差距。

独立分析的弱点

独立分析三个关键弱点:(1)prompt 敏感性高,作者明确承认"prompt 改动显著影响 benchmark 性能,进一步优化仍有空间",意味着当前数字可能并非模型真实能力上界,实际部署需要 prompt 工程;(2)3D 切片被强制 $\le 85$ 张且需等距采样,对全腹部 CT 或高分辨率 MRI 容易遗漏微小病灶,且不同层厚的扫描会改变采样密度,可能引入偏差——改进方向是引入自适应切片采样或滑动窗口;(3)WSI 仅保留单放大率档位且仅 126 patch,对肿瘤异质性高的病例可能漏掉关键区域,改进方向是结合多倍率金字塔输入或基于注意力的 patch 选择;(4)边界框定位 IoU 38% 虽比前代好但仍偏低,输出 JSON 格式也限制了下游使用灵活性,改进方向是引入更结构化的位置编码。

未来方向

作者明确方向:扩展内部评估框架以兼容 Qwen 等异构架构做更公平对比;针对 SLAKE/VQA-RAD 等回退基准做针对性微调以恢复窄任务性能。基于成果可延伸的方向:(1)把 3D 切片数动态化或引入压缩器,让模型处理更长病例;(2)把 WSI 多放大率金字塔原生接入,让模型在不同尺度上联合推理;(3)把 bounding box 能力与 EHRQA 文档理解结合,支持"指向 + 解释"联合输出;(4)在 RL 阶段引入临床医生反馈的多轮偏好数据,对齐真实临床推理;(5)探索多模态 RAG 方案,让模型调用外部医学知识库弥补训练数据盲区。

复现评估

复现评估:模型权重与教程已开源(https://goo.gle/medgemma),但训练数据绝大部分是内部数据集(CT Dataset 1、MRI Dataset 1、Internal WSI 335,825 例、Dermatology 4/5、CXR-IND1 605,732 例、EHR Dataset 2-5 等),外部研究者无法复现完整训练流程,只能基于开源权重做微调与评估;评估代码基于内部框架,附录 A/B 提供了 prompt 模板但缺乏统一推理脚本。算力层面未公开,但 4B 模型 + 400M 视觉编码器、$\le 32K$ 上下文,预估单卡 H100 80GB 可微调。整体复现难度:权重获取易(Apache 类许可),训练流程复现难(数据不可得),评估可参照公开评测集(CT-RATE、MS-CXR-T、Chest ImaGenome、Mendeley)做近似复现。