用迭代优化方法自动纠错标注,让医学影像数据集规模翻4倍
Expectation-Maximization as the Engine of Scalable Medical Intelligence
- 通过期望-最大化框架让模型与标注数据互相迭代改进
- 构建4.8万例CT数据集,肿瘤标注量达416万像素,超现有最大数据集
- 模型诊断准确率超人类专家7%,在两大权威评测中表现显著提升
高质量标注的医学数据集是医疗AI研究的基础,但构建小型中等质量数据集需跨学科团队多年努力。尽管主动学习可优先标注内容,扩展仍需大量人工修正噪声标注。本文将此问题建模为缺失数据问题,提出ScaleMAI框架,通过期望-最大化(EM)过程统一数据标注与模型开发的共同演化。在迭代过程中,模型自动识别并纠正标注错误(期望步),而优化后的标注数据重新训练模型以提升精度(最大化步)。除经典EM算法外,还引入人工专家对无法由两步解决的标注(<5%)进行复核。最终,ScaleMAI逐步构建出包含47,315例CT扫描的数据集(比最大公开数据集PanTS大4.8倍),涵盖4,163,720个体素级良性/恶性肿瘤标注及88个解剖结构。该框架迭代训练的模型在肿瘤诊断上超越人类专家7%,在两个权威基准上肿瘤检测和分割性能分别提升10%和14%,显著优于基于小规模中等质量数据集训练的模型。
原文摘要 · Abstract (English)
Large, high-quality, annotated datasets are the foundation of medical AI research, but constructing even a small, moderate-quality, annotated dataset can take years of effort from multidisciplinary teams. Although active learning can prioritize what to annotate, scaling up still requires extensive manual efforts to revise the noisy annotations. We formulate this as a missing-data problem and develop ScaleMAI, a framework that unifies data annotation and model development co-evolution through an Expectation-Maximization (EM) process. In this iterative process, the AI model automatically identifies and corrects the mistakes in annotations (Expectation), while the refined annotated data retrain the model to improve accuracy (Maximization). In addition to the classical EM algorithm, ScaleMAI brings human experts into the loop to review annotations that cannot be adequately addressed by either Expectation or Maximization step (<5%). As a result, ScaleMAI progressively creates an annotated dataset of 47,315 CT scans (4.8x larger than the largest public dataset, PanTS) including 4,163,720 per-voxel annotations for benign/malignant tumors and 88 anatomical structures. ScaleMAI iteratively trains a model that exceeds human expert performance in tumor diagnosis (+7%), and outperforms models developed from smaller, moderate-quality datasets, with statistically significant gains in tumor detection (+10%) and segmentation (+14%) on two prestigious benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。