3. 鲁棒性威胁建模

在医学影像模型鲁棒性测试工程中,威胁建模是连接“模型可能失效”与“测试用例如何设计”的关键桥梁。它的核心任务不是简单罗列异常现象,而是把真实临床链路中可能出现的扰动,按来源、作用位置、统计特性和对模型决策的影响方式,整理成可分类、可参数化、可复现的威胁集合。只有先建立清晰的威胁分类,后续的测试数据构造、指标选择和通过判据才有依据。

一、威胁建模的基本视角

医学影像从采集到模型输出,通常经历“成像设备—重建/后处理—存储传输—预处理—模型推理”多个环节。威胁可以发生在任一环节,并沿信号链向下游传播。因此,威胁建模应同时回答三个问题:扰动从哪里来、以什么形式叠加到图像或数据上、它主要破坏模型的哪类能力。按来源划分,常见威胁可分为自然噪声、扫描仪差异、对比度变化、运动伪影、金属伪影和对抗攻击六大类;按作用方式,又可分为加性扰动、乘性扰动、几何形变、局部强干扰和语义定向扰动。

需要特别说明的是,医学影像中的“威胁”并不等同于“图像质量差”。有些扰动人眼几乎不可见,却可能显著改变模型输出,例如对抗攻击;有些扰动视觉上非常明显,但模型仍可能保持正确判断,例如局部金属伪影未覆盖病灶区域。因此,威胁建模必须同时记录扰动的物理来源和模型响应,而不能只凭视觉严重程度判断风险。

医学影像鲁棒性威胁分类与作用链 采集端 扫描仪 / 参数 / 患者 重建 / 后处理 滤波 / 归一化 / 重采样 模型推理 分割 / 检测 / 分类 输出与判读 病灶 / 分级 / 置信度 威胁注入点 六类威胁及其作用特征 自然噪声 来源:量子噪声、电子噪声、热噪声 形式:加性高斯/泊松,像素级随机 影响:低剂量CT、低场MRI细节退化 参数:σ、SNR、噪声功率谱 典型测试:σ=0.01~0.10 归一化灰度 扫描仪差异 来源:厂商、型号、协议、校准 形式:灰度分布偏移、分辨率差异 影响:跨中心泛化性能下降 参数:窗宽窗位、层厚、重建核 典型测试:A设备训练→B设备推理 对比度变化 来源:窗宽窗位、曝光、造影剂 形式:线性/非线性灰度映射 影响:低对比病灶漏检、边界模糊 参数:γ、窗宽、窗位、对比度增益 典型测试:γ=0.5~2.0 幂律变换 运动伪影 来源:呼吸、心跳、患者移动 形式:模糊、重影、条带、几何错位 影响:结构边界失真、配准失败 参数:幅度、频率、方向、持续时间 典型测试:平移/旋转/弹性形变 金属伪影 来源:植入物、牙科材料、穿刺针 形式:放射状条带、暗带、杯状伪影 影响:局部CT值失真、结构遮挡 参数:金属尺寸、密度、位置、kVp 典型测试:金属植入区域叠加 对抗攻击 来源:人为构造、梯度优化 形式:微小扰动、语义定向、补丁 影响:高置信度误判、目标消失 参数:ε、迭代步数、攻击目标 典型测试:FGSM/PGD ε=0.01~0.05 威胁建模输出:来源分类 + 扰动参数 + 注入位置 + 预期模型响应 + 测试优先级

二、六类威胁的专业特征与测试参数

自然噪声主要来自成像物理过程。CT中的量子噪声近似泊松分布,低剂量条件下尤为明显;MRI中的热噪声通常建模为复高斯噪声。测试时常用归一化灰度标准差σ控制强度,例如在[0,1]灰度范围内取σ=0.01、0.03、0.05、0.10四档,观察模型Dice或敏感度随噪声水平的变化曲线。若模型在σ=0.05时Dice下降超过5个百分点,说明其对噪声较敏感,需要加入噪声增强训练或测试时去噪。

扫描仪差异是跨中心部署中最常见的威胁。同一解剖部位,不同厂商设备的重建核、层厚、窗宽窗位设置会导致灰度分布和纹理特征明显偏移。建模时应记录设备型号、管电压、管电流、重建算法、层厚、像素间距等元数据,并构造“源域训练—目标域测试”的跨域评估。例如,用A医院数据训练,在B医院数据上直接推理,若敏感度从0.92降至0.78,则需引入域适应或标准化预处理。

对比度变化可由窗宽窗位调节、曝光差异或造影剂浓度变化引起。它通常表现为全局或局部的灰度映射改变,可用幂律变换模拟:I' = I^γ,γ<1提升暗部对比度,γ>1压缩暗部。测试时取γ=0.5、0.8、1.2、2.0,重点观察低对比病灶的分割完整性和检测召回率。若模型对γ变化敏感,说明其特征提取过度依赖绝对灰度值,而非结构关系。

运动伪影包括呼吸运动、心脏搏动和患者自主移动。在图像域,可用平移、旋转、弹性形变模拟;在k空间,则表现为相位编码方向的条带。测试参数包括最大位移(如2mm、5mm、10mm)、旋转角度(如2°、5°)和形变场平滑度。评估时应关注病灶边界的Hausdorff距离和体积误差,而不仅是Dice。运动伪影常导致边界模糊,使分割模型倾向于欠分割或过分割。

金属伪影常见于CT,表现为放射状条带和暗带,导致金属周围CT值严重失真。建模时可在图像中嵌入不同尺寸和密度的金属掩膜,或使用真实金属植入病例。关键参数包括金属材料(钛、不锈钢)、尺寸、位置和扫描kVp。评估重点不是全图指标,而是金属附近感兴趣区的CT值误差和病灶可识别性。若模型在金属附近出现大面积误分割,需引入金属伪影校正或伪影感知训练。

对抗攻击是人为构造的、以误导模型为目标的扰动。与自然噪声不同,对抗扰动通常幅度很小(如ε=0.01~0.05),但方向经过梯度优化,能造成高置信度误判。医学影像中常见的攻击包括FGSM、PGD和通用扰动。测试时应固定攻击目标(如使病灶消失、使良性判为恶性),记录攻击成功率、扰动范数和模型置信度变化。对抗鲁棒性测试不仅关注准确率下降,更要关注误判的临床后果。

三、威胁建模的交付物与优先级

完成分类后,应形成一份可执行的威胁建模表,至少包含:威胁类别、来源环节、扰动形式、关键参数、注入位置、评估指标、通过判据和优先级。优先级可依据“发生概率×临床影响”排序。例如,运动伪影和扫描仪差异发生概率高,应优先覆盖;对抗攻击发生概率低但后果严重,也应纳入安全关键测试。下面给出一个简化的威胁建模示例表,参数为教学假设值,实际项目需根据设备、病种和临床要求调整。

威胁类别 注入位置 示例参数 主要评估指标 示例通过判据 优先级
自然噪声 图像域 σ=0.01~0.10 Dice、敏感度 σ=0.05时Dice下降≤3% 高
扫描仪差异 采集/重建 设备A→设备B 跨域敏感度、AUC 跨域敏感度下降≤5% 高
对比度变化 后处理 γ=0.5~2.0 召回率、边界F1 γ=0.8~1.2时召回率下降≤3% 中
运动伪影 采集/图像域 位移≤5mm,旋转≤3° HD95、体积误差 HD95增加≤2mm 高
金属伪影 图像域 钛棒直径5mm,kVp=120 局部CT值误差、病灶可识别性 金属旁ROI误分割面积≤10% 中
对抗攻击 模型输入 PGD ε=0.03,10步 攻击成功率、置信度偏移 目标病灶攻击成功率≤20% 高
威胁建模不是一次性工作。当模型迭代、设备更新或临床流程变化时,应重新评估威胁清单和参数范围。建议将威胁建模表与测试用例库关联,使每个威胁类别都能追溯到具体的测试样本、扰动参数和通过判据,形成可审计的鲁棒性证据链。

最后需要强调,六类威胁并非彼此独立。真实场景中,低剂量扫描可能同时带来自然噪声和对比度变化,金属植入患者可能同时存在金属伪影和运动伪影。因此,在完成单威胁测试后,还应设计组合威胁场景,观察模型在多重扰动下的性能退化是否呈非线性叠加。组合测试的优先级可依据临床实际共现概率确定,例如“低剂量+运动”在急诊CT中较为常见,应优先纳入。