A Data-scalable Transformer for Medical Image Segmentation: Architecture, Model Efficiency, and Benchmark
Yunhe Gao Mu Zhou Di Liu Zhennan Yan Shaoting Zhang Dimitris N. Metaxas Thanks: Y. Gao, D. Liu and D. Metaxas are with the Computer Science Department, Rutgers University, Piscataway, NJ, USA. M. Zhou and Z. Yan are with SenseBrain Research, Princeton, NJ, USA.S. Zhang is with Shanghai Artificial Intelligence Laboratory, Shanghai, China. Thanks: Manuscript received on October 15, 2022. Major revision received on April 4, 2023.
Abstract
Transformers have demonstrated remarkable performance in natural language processing and computer vision. However, existing vision Transformers struggle to learn from limited medical data and are unable to generalize on diverse medical image tasks. To tackle these challenges, we present MedFormer, a data-scalable Transformer designed for generalizable 3D medical image segmentation. Our approach incorporates three key elements: a desirable inductive bias, hierarchical modeling with linear-complexity attention, and multi-scale feature fusion that integrates spatial and semantic information globally. MedFormer can learn across tiny- to large-scale data without pre-training. Comprehensive experiments demonstrate MedFormer’s potential as a versatile segmentation backbone, outperforming CNNs and vision Transformers on seven public datasets covering multiple modalities (e.g., CT and MRI) and various medical targets (e.g., healthy organs, diseased tissues, and tumors). We provide public access to our models and evaluation pipeline, offering solid baselines and unbiased comparisons to advance a wide range of downstream clinical applications.
原文 arXiv:2203.00131;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2203.00131v5