Expectation-Maximization as the Engine of Scalable Medical Intelligence
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , , , , , , , , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2025
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866918237104504832 |
|---|---|
| author | Li, Wenxuan Bassi, Pedro R. A. S. Lin, Tianyu Chou, Yu-Cheng Wasserthal, Jakob Zhou, Xinze Chen, Qi Isensee, Fabian Kirchhoff, Yannick Rokuss, Maximilian Roy, Saikat Ulrich, Constantin Maier-Hein, Klaus Płotka, Szymon Chen, Xiaoxi Wang, Kang Yang, Yang Xu, Daguang Ding, Kai Tang, Yucheng Yuille, Alan L. Zhou, Zongwei |
| author_facet | Li, Wenxuan Bassi, Pedro R. A. S. Lin, Tianyu Chou, Yu-Cheng Wasserthal, Jakob Zhou, Xinze Chen, Qi Isensee, Fabian Kirchhoff, Yannick Rokuss, Maximilian Roy, Saikat Ulrich, Constantin Maier-Hein, Klaus Płotka, Szymon Chen, Xiaoxi Wang, Kang Yang, Yang Xu, Daguang Ding, Kai Tang, Yucheng Yuille, Alan L. Zhou, Zongwei |
| contents | 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. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2501_03410 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Expectation-Maximization as the Engine of Scalable Medical Intelligence Li, Wenxuan Bassi, Pedro R. A. S. Lin, Tianyu Chou, Yu-Cheng Wasserthal, Jakob Zhou, Xinze Chen, Qi Isensee, Fabian Kirchhoff, Yannick Rokuss, Maximilian Roy, Saikat Ulrich, Constantin Maier-Hein, Klaus Płotka, Szymon Chen, Xiaoxi Wang, Kang Yang, Yang Xu, Daguang Ding, Kai Tang, Yucheng Yuille, Alan L. Zhou, Zongwei Computer Vision and Pattern Recognition 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. |
| title | Expectation-Maximization as the Engine of Scalable Medical Intelligence |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2501.03410 |