← 返回 2026-07-15

面向自主且可审计的医学影像模型开发:多智能体框架 AMID Towards Autonomous and Auditable Medical Imaging Model Development

Shengyuan Liu, Jia-Xuan Jiang, Boyun Zheng, Cheng Wang, Zipei Wang, Wentao Pan, Hongtao Wu, Houwen Peng, Yu Gu, Lichao Sun, Yixuan Yuan 📅 2026-07-12 👍 18 2026-07-19 18:30
LLM Agent 医学影像 可审计AI 多智能体系统 自主机器学习工程

首个通用医学影像自主多智能体建模框架,以数据条件规划与验证引导搜索,19/20任务超越基线。

前置知识

自主机器学习工程(Autonomous MLE)

自主机器学习工程(Autonomous Machine Learning Engineering)是指用大语言模型(LLM)智能体自动完成模型开发全流程的范式。智能体将规划、代码执行、调试和经验反馈耦合在一起,把模型构建从一次性的代码生成转变为迭代式、反馈驱动的实验循环:每次实验产生代码、验证分数、日志和提交产物,这些证据反过来指导下一次细化。代表性系统包括 AIDE、ML-Master、R&D-Agent,它们用树搜索、图结构或自组织团队等方式组织探索。

本文正是把这种自主 MLE 范式迁移到医学影像领域,理解智能体如何迭代实验、积累证据,才能理解 AMID 的两阶段优化和验证层为何必要。

医学影像建模任务与评估协议

医学影像建模任务涵盖分割(如 Dice)、检测(如 AP、IoU)、分类(如 Accuracy)、图像质量评估与增强(如 Score、LNCC)等,每种任务对应特定的预测产物格式(分割掩膜、检测框、提交文件等)。评估协议要求严格的患者级(patient-level)或站点级(site-level)数据划分、锁定的折(locked-fold)划分、正确的指标方向与计算方式,以及与评测器匹配的提交格式。任何指标方向错误、数据泄漏、折划分错误或格式不匹配都会使分数失效。

AMID 的核心动机正是医学影像任务对评估协议和产物格式的严格要求,理解这一点才能理解为何需要独立的验证/审查层来防止不可信证据驱动搜索。

医学影像基础模型与工具箱

医学影像领域有一批可复用的基础模型和工具箱:nnU-Net 提供自适应的分割流水线(含数据转换、折管理、预处理、训练与推理);MONAI 提供跨任务的可复用组件(数据加载、空间/强度变换、损失函数、网络结构和预训练 bundle);MedSAM/MedSAM2 提供可提示的分割能力;TotalSegmentator、VISTA3D 提供解剖级工具;病理领域有 UNI、Phikon-v2、CellViT、PathoSAM 等编码器与提议器。它们是构建方法的实现锚点而非强制基线。

AMID 的资源寻址机制正是围绕这些基础模型和工具箱展开,理解这些资源的用途才能理解数据条件方法规划如何把资源与任务匹配成可执行的方法车道。

数据条件方法规划(DCMP)

数据条件方法规划(Data-Conditioned Method Planning)是 AMID 的核心组件之一。它先对任务的公开数据视图做证据支撑的画像(模态、维度、体素间距、标注密度、患者分组、指标定义、输出格式等),再把这些画像当作搜索谓词,结合本地资源库和基础模型注册表,把粗糙的任务级搜索空间细化为若干可执行、可并行的方法车道(method lanes)。每条车道携带建模假设、所需资源、验证义务、训练预算与可能失败模式。

这是本文区别于通用 MLE 智能体的关键创新,理解 DCMP 才能理解为何 AMID 能在异质的医学影像任务上避免在医学上不合理或不可运行的流水线上浪费算力。

验证引导的两阶段优化

验证引导的两阶段优化(Verification-Guided Two-Stage Optimization)把搜索分为两阶段。Stage 1 是行为门控探索:把 worker 智能体分配到不同方法族,每个尝试绑定提交代码、验证分数、产物与车道归属;一个独立于 worker 的审查者(reviewer)校验冻结提交是否真正在活动协议下执行、证据是否可采信(验证集、指标方向、完整产物、无重复、可追溯),只有通过审查的尝试才成为可比证据,达到覆盖后按最优有效证据晋升最强候选。Stage 2 是选择性利用:worker 从晋升锚点出发做模型/推理/后处理/集成改进、修复和终稿准备,审查门控始终生效。

这是 AMID 区别于纯分数驱动优化的关键,理解该机制才能理解为何高分数也可能来自错误划分或格式错误,以及为何验证层会反过来改变优化本身。

研究动机

通用的自主 MLE 智能体(如 AIDE、ML-Master、R&D-Agent)直接套用到医学影像时会暴露两类问题。其一,它们构造的搜索空间对异质的医学影像任务过于粗糙:一个任务可能被识别为分割或检测,但合适的方法族实际上取决于模态、解剖、采集协议、标注语义、临床目标和提交约束。若不把这些医学影像属性显式转化为可执行、可并行的方法车道,智能体可能把算力花在技术上能跑但医学上不合适、与数据不匹配、或无法产出有效最终产物的流水线上。其二,医学影像的实验反馈昂贵、延迟且可能不可信:与轻量基准不同,医学影像实验需要高 GPU 预算、长训练、内存感知推理和严格的患者级/站点级划分。在此条件下智能体容易偏向低成本捷径实验、训练不完整、验证协议薄弱或产物无效,一个验证分数因此可能在小组样本、患者级依赖、扫描仪/站点效应、泄漏、指标失配、折错误或无效预测产物下不稳定甚至误导。作者在 ReX-MLE 基准上观察到 AIDE 在 PUMA-T1 组织分割上直接 FAIL、ML-Master 在 USenhance 上 FAIL、多个基线在 SEG.A 上得分仅 0.00–0.02。

本文的目标是本文的目标是设计首个面向通用医学影像的自主多智能体 MLE 框架 AMID,把医学影像模型开发从劳动密集的手工工程转变为一个智能体工作流,端到端地产出高性能、已验证且可审计的模型产物包:可执行训练与推理代码、模型权重/检查点、预测文件、验证分数、所需时的最终提交产物,以及证明结果在正确的数据、指标、划分和提交契约下产出的审计轨迹。具体地,AMID 要在给定一个数据集和任务定义(目标输出、评估指标、提交协议)的最小输入下,自动完成从原始文件到最终可审计候选模型的全部开发,并让中间选择对人类审查者可见。

与已有工作不同的是,本文的独特切入角度是把自主医学影像建模重新表述为一种「数据条件 + 验证控制」的搜索问题,而不是自由形式的想法生成或一次性提示转代码。与通用 MLE 智能体相比,AMID 不只搜索通用流水线,而是先用 DCMP 把任务与数据画像、可执行资源清单共同转化为多条方法车道,使搜索空间在昂贵实验之前就可被检查;它不只靠分数驱动优化,而是把验证当作控制层——审查者独立校验每次尝试的验证协议、指标计算和预测/提交产物,不可信证据只能用于诊断或修复,不能驱动晋升或最终选择。这种「显式构造候选方法空间 + 严格验证选择证据」的双重设计,是 AMID 区别于既有医学 autoresearch 系统(如只做小规模约束任务、只靠临床医师引导)的本质差异。

核心方法

AMID 的整体思路是:把医学影像模型开发当作一条「建立任务契约 → 画像数据 → 规划方法车道 → 两阶段探索-利用优化 → 产出审计包」的多智能体流水线。系统首先从任务文件物化一个工作区,记录建模目标、评估指标、预测模式和所需输出契约,然后对公开数据视图做画像,写出模态、维度、监督类型、样本组织、几何、标签结构以及泄漏/不平衡/稀疏掩膜/无效输出等风险的证据支撑摘要。接着 DCMP 利用该画像构建方法车道组合,把实现支撑的路线分组为方法族用于并行探索。验证引导的两阶段优化把组合变成可执行过程:Stage 1 是行为门控探索(约 1 折验证,覆盖多车道),Stage 2 是选择性利用(多折交叉验证、产物有效性校验)。两阶段中审查者都校验验证协议、指标计算和预测/提交产物。整个执行由自组织智能体循环承载:每个 worker 在隔离的 git worktree 中运行,但共享同一份任务契约、方法车道、尝试记录、笔记、技能、资源和审计产物;一个中心化的生命周期管理器负责 worker 的创建、上下文注入、打断、重指派、退役与终稿化。系统支持 Codex、Claude Code、OpenCode、Cursor Agent、Kiro 等后端以及混合后端运行。

核心创新点是「验证即控制层」与「数据条件的方法空间显式构造」二者的结合。与已有 MLE 智能体的本质区别在于:第一,AMID 不让 worker 自行认证自己的证据——每个提交先被冻结为待定尝试(含提交代码、worker 身份、车道元数据、共享状态检查点、产物),再由独立审查者在该 session 之外评估,校验是否真正在活动协议下执行、证据是否可采信;只有通过审查的尝试才成为真实、可归属、可比的证据。第二,AMID 把「想法生成」刻意做成一个领域条件搜索系统:DCMP 展开多条医学上合理的路线,对仅论文级或不兼容的资源降权或排除,在昂贵实验之前让候选空间可被检查。这两点使得 AMID 不是更强的提示工程,而是围绕现代编码智能体运行时构建的高层基底(任务契约、共享文件系统记忆、方法车道记录、尝试账本、心跳干预、可复用技能、生命周期调度、最终产物门控)。

方法步骤详情

AMID 的完整流程可拆为以下步骤。步骤 1:任务与产物契约建立。系统物化工作区,记录建模目标、评估指标、预测模式和所需输出契约。步骤 2:任务与数据画像。检查任务描述、README、样例提交、指标定义、可见训练数据、标签与元数据,按四层记录证据——清单层(文件类型、目录布局、训练/测试组织、样本计数、配对关系、缺失/重复、可见分层)、图像表示层(采样原始文件估计形状、通道数、dtype、强度范围、间距、朝向、仿射一致性、各向异性、体素尺寸分布)、监督层(类别平衡/前景比/连通分量/框数与目标尺寸范围/对齐与对比偏移/模态缺失与配准关系)、评估层(主指标、方向、折或分组约束、患者级泄漏风险、提交列、引用的预测文件、格式校验)。步骤 3:资源寻址式方法搜索。DCMP 区分两类资源:预置本地资源库(nnU-Net、MONAI 等)和策划的基础模型与外部资源注册表(MedSAM/MedSAM2、TotalSegmentator、VISTA3D、病理/胸片/CT/MRI/视网膜领域模型),逐个检查是否匹配画像数据、可用产物能否被验证或获取,产出混合方法图,最终车道组合含保守 nnU-Net/MONAI 基线、基础模型适配、任务专用架构、预处理/后处理密集型、集成、以及只校验数据加载/指标计算/提交机制的标定车道。步骤 4:Stage 1 行为门控探索。worker 分散到方法族,提交绑定到特定车道的尝试(提交代码、预测/提交产物、验证分数、元数据、车道归属),审查者校验验证集、指标方向与计算、完整产物、无重复、必要日志/检查点、代码到产物的可追溯性;达到所需车道覆盖并出现平台行为或文档豁免后,管理器按最优有效证据排名并晋升最强候选(可选保留紧邻次优车道作为挑战者)。步骤 5:Stage 2 选择性利用。优化器 worker 改进领头配方,挑战者 worker 保留有限容量给次优车道,修复 worker 处理有原始分但协议/产物失败的候选,终稿 worker 准备最终预测/提交/模型包;审查门控要求正确指标方向、合法数据访问、锁定折使用、完整折输出、正确均分计算、折来源可追溯、折不变流水线、有效预测模式、完整所需文件和从最终包回溯到已接受尝试的可追溯性。步骤 6:自组织智能体循环与生命周期管理。生命周期管理器维护每个 worker 的显式状态,监控进程健康、最近尝试、空闲时间、上下文压力和进展停滞迹象,必要时用定向提示打断并恢复(而非替换为新会话);心跳干预分反射式心跳(把结果转成实验笔记)、整合心跳(综合共享笔记、找矛盾、总结团队状态)、平台心跳(强制在停滞时做转向决策)。

技术新颖性

技术新颖性体现在四个层面。其一,据作者所知 AMID 是首个面向通用医学影像模型开发的自主多智能体 MLE 框架,能在异质任务和模态间产出可审计模型包(含可执行代码、验证证据、可提交预测产物)。其二,DCMP 首次把粗糙任务级搜索空间在「真实数据条件」和「可运行资源清单」双重约束下细化为可执行、可并行的方法车道,并对仅论文级、不兼容或超预算的想法降权或排除而非静默删除,使搜索空间在昂贵实验前就可被检查。其三,验证引导的两阶段优化首次把「审查者作为控制层」贯穿探索与利用:独立审查者校验验证协议、指标计算和预测/提交产物,不可信证据只能用于诊断/修复,这反过来改变了优化本身(无法满足协议的候选不能驱动晋升或最终选择)。其四,系统在执行基底层面做了专门化:后端无关(Codex、Claude Code、OpenCode、Cursor Agent、Kiro 及混合后端可参与同一运行),每个 worker 在隔离 git worktree 运行但共享同一份任务契约与方法车道,中心化生命周期管理器 + 共享文件系统记忆 + 技能(如组织文件、读论文、适配基础模型的技能)+ 心跳干预,构成一种「管理器中介的自组织」而非固定身份的智能体社会。

Overview of AMID's verification-guided optimization and benchmark performance.
Figure 1: Overview of AMID's verification-guided optimization and benchmark performance.
Overview of AMID.
Figure 2: Overview of AMID.

实验结果

核心发现来自 ReX-MLE 基准(20 个跨模态、跨预测类型的医学影像挑战,含全景 X 光、CT、MRI、CTA、MRA、组织病理、超声、显微镜;主实验用 Codex+GPT-5.5 后端,每任务 24 小时墙钟预算 + 1 张 48GB RTX A6000)。第一,AMID 在全部 20 个挑战上产出有效的已接受结果,并在 19 个任务上改进最强基线,仅 TopCoW-CTA-Cls 持平(0.33 Accuracy)。第二,分割与检测任务增益最大——这是通用智能体的主要瓶颈。分割方面,SEG.A Dice 提升 $\Delta=+0.89$(0.91 vs 基线最优 0.02)、ISLES'22 Dice 提升 $+0.67$(0.71 vs 0.04),并把此前 FAIL 或接近 0 的 PUMA 组织分割转化为 0.5 以上的有效 Dice;检测方面,DENTEX AP 提升 $+0.40$(0.49 vs 0.09)、PUMA-T1 细胞核检测 F1 提升 $+0.46$(0.54 vs 0.08)。第三,相对人类参考(图 3 归一化到 1.0),最强 AMID 结果在 DENTEX、NeurIPS-CellSeg、PUMA-T2-Det、LDCT-IQA 上达到或超过参考,在 SEG.A 上接近。第四,统一后端对照(Table 2)把 AIDE/ML-Master/R&D-Agent 都固定到 GPT-5.5,AMID 仍在三个代表性任务上胜出:DENTEX 0.49 vs 0.08 AP、PUMA-T1-Seg 0.56 vs 0.52 Dice、TopCoW-MRA-Cls 0.46 vs 0.39 Accuracy,排除了「基线弱」的解释。第五,运行时后端对照显示 Codex+GPT-5.5 在 DENTEX 最强(0.49 AP),Claude Code 在 PUMA-T1-Seg(0.64 Dice)和 TopCoW-MRA-Cls(0.50 Accuracy)最强,说明 AMID 的增益来自搜索组织与验证而非单一运行时。第六,案例分析显示强项源于任务匹配的方法搜索:DENTEX 用 YOLOv8 检测 + 解剖校准层(FDI 象限映射、保守诊断回退)达到 0.48825 mAP;PUMA 病理用锁定五折 UNI 冻结特征集成(Track1)和 Phikon-v2 微调(Track2)把此前 0 分任务变成 0.56395/0.56214 Dice,PUMA-T2 细胞核检测的 0.28422 macro-F1 在官方赛道排名第一。

Focused comparison on three representative ReX-MLE challenges.
Table 2: Focused comparison on three representative ReX-MLE challenges.
Challenge-wise comparison with human reference performance on ReX-MLE.
Figure 3: Challenge-wise comparison with human reference performance on ReX-MLE.
查看结构化数据
任务指标本文基线提升
SEG.A 分割 Dice(↑) 0.91 0.02(最强基线) +0.89 绝对提升
ISLES'22 中风病灶分割 Dice(↑) 0.71 0.04 +0.67 绝对提升
DENTEX 牙齿检测 AP(↑) 0.49 0.09 +0.40 绝对提升
PUMA-T1 细胞核检测 F1(↑) 0.54 0.08 +0.46 绝对提升
PUMA-Track1/2 组织分割 Dice(↑) 0.56 / 0.56 FAIL / 0.00 由失败转为有效 0.56
NeurIPS-CellSeg 细胞分割 F1(↑) 0.90 0.36 +0.54 绝对提升

局限与改进

作者承认的主要局限是成本:长时程医学 MLE 同时消耗大量 GPU 时间和编码智能体 token,尤其在探索多条方法车道、运行阶段级审查校验、跨 worker 保留证据时。因此当前实验并未在所有最新模型-运行时组合上穷举每个挑战,全套结果用主后端,新/备选后端只在聚焦的代表性任务上考察;跨模型后端、算力预算、方法车道消融、验证消融和审查式有效性分析仍待补齐,以分离搜索设计的贡献与底层运行时的强度。AMID 还依赖资源注册表质量、数据画像准确性和后端智能体在预算下实现强方法的能力。在我自己的观察中还有两点:第一,挑战性能是系统级证据而非临床验证——它不建立临床安全性、前瞻泛化、部署偏移下的校准或患者适用性,这一点作者自己也明确;第二,若干弱项任务暴露了系统设计的天花板,USenhance 的 LNCC 仅 0.19(人类参考 0.91)说明在需要长训练的对抗式图像翻译任务上,短期证据会错误地淘汰本可能更强的 CycleGAN 候选;两个 TopCoW 分类任务(CTA 收敛到固定拓扑先验、MRA 仅 0.46/0.17 准确率)暴露了「分割-拓扑-分类」耦合瓶颈,说明当下游目标依赖小分支接触和图标签时,体素级掩膜并不足够。

独立分析的弱点

第一个弱点是成本-性能权衡尚不明确。论文只给出主后端的全套结果和少数聚焦任务的对照,缺少对算力预算、方法车道数量、审查频次的系统消融,难以判断 DCMP 与验证层各自的贡献占比,也难判断在更小预算下增益是否仍成立——改进方向是成本感知的长时程自治:更便宜的早期过滤器、更强的停止规则、可复用技能蒸馏、缓存资源审计、代理验证,以及对值得深验的候选才升级到昂贵模型。第二个弱点是短时程证据偏差。USenhance 案例显示,固定预算下短期证据会偏好稳定但弱的 U-Net 配方而淘汰需要更长训练的 CycleGAN——改进方向是引入损失曲线外推、早停置信区间或基于代理训练的候选预筛,让短期不稳定的强候选不被过早拒绝。第三个弱点是耦合任务的瓶颈。TopCoW 分类暴露出当目标依赖中间分割和拓扑提取时,AMID 的车道设计可能让各环节各自优化却无法联合提升——改进方向是把「分割→拓扑→分类」建模为一条强耦合车道或引入端到端可微的拓扑后处理。第四个弱点是资源注册表与后端的依赖。系统强依赖注册表的覆盖与画像准确性,且 Claude Code 与 Codex 在不同任务上各擅胜场——改进方向是自动化的注册表维护与质量审计、以及按任务自适应选择后端的调度策略。

未来方向

作者明确提出的未来方向包括:扩展受控后端对照;为数据画像、方法车道规划、两阶段优化和验证门控分别加消融实验;强化审查式审计分析;评估成本感知的长时程运行策略(便宜早期过滤器、更强停止规则、技能蒸馏、缓存审计、代理验证、选择性升级);在更多任务、模型-运行时组合、资源设置和面向部署的产物校验上拓宽验证。基于本文成果可延伸的方向包括:把验证引导搜索推广到非挑战的临床建模请求(如预测服务、Docker 化推理、供人类审查的证据包);研究如何把审查者的有效性规则蒸馏为可学习规则或神经校验器,降低人工设计成本;探索把 AMID 的共享记忆与技能机制用于跨任务的技能迁移(一次运行中学到的技能复用到下一次);研究在资源受限或数据敏感(如不能下载基础模型)场景下的轻量化适配;以及对系统产出的「审计轨迹」做临床下游验证,连接系统级证据与临床安全性的鸿沟。

复现评估

论文的可复现性总体较好。作者提供了项目主页(https://github.com/CUHK-AIM-Group/AMID)并声明「为所有挑战获得的解决方案已在项目页公开」。实验设置交代清晰:基准为 ReX-MLE(Kenia et al., 2025,20 个 MICCAI 风格医学影像挑战,覆盖分割/检测/分类/图像质量/增强,跨全景 X 光、CT、MRI、CTA、MRA、组织病理、超声、显微镜);主实验用 Codex+GPT-5.5 后端,每任务 24 小时墙钟预算和 1 张 48GB NVIDIA RTX A6000 GPU;基线分数用 ReX-MLE 报告的已接受基准分,AMID 用同样的已接受产物标准,人类列为原始竞赛参考分。Table 1 与 Table 2 给出了逐任务的指标值(Dice、AP、IoU、Accuracy、Score、LNCC 等)和绝对提升 $\Delta$,便于核对。但论文也注明是「正在进行的前期技术报告」,关键挑战在于完整复现需要相当资源:大量 GPU 时间与编码智能体 token,且论文未给出 DCMP 方法图的完整节点/边定义、审查者的具体校验规则伪代码、技能模板的内容和心跳触发的精确阈值,这些是实现细节层级的复现障碍。此外 ReX-MLE 的「人类参考」是基于重划分公开训练数据构造的近似参考而非严格匹配的榜单对照,复现时需注意该口径差异。