LViT: Language meets Vision Transformer in Medical Image SegmentationThanks: Zihan Li is with Xiamen University and the Department of Computer Science, University of Illinois at Urbana-Champaign, Urbana, IL 61801, USA (e-mail: zl111@illinois.edu).Thanks: Yunxiang Li and You Zhang are with the Department of Radiation Oncology, UT Southwestern Medical Center, Dallas, TX 75235, USA.Thanks: Qingde Li is with the Department of Computer Science, University of Hull, Hull, HU6 7RX, UK.Thanks: Puyang Wang is with DAMO Academy, Alibaba Group, Hangzhou 310024, China.Thanks: Dazhou Guo, Le Lu, and Dakai Jin are with DAMO Academy, Alibaba Group, New York, NY 10014, USA.Thanks: Qingqi Hong is with Xiamen University, Xiamen 361005, China. (e-mail: hongqq@xmu.edu.cn).Thanks: Corresponding author: Qingqi Hong
Zihan Li Yunxiang Li Qingde Li Puyang Wang Dazhou Guo Le Lu Affiliation: Dakai Jin, , You Zhang, Qingqi Hong,
Abstract
Deep learning has been widely used in medical image segmentation and other aspects. However, the performance of existing medical image segmentation models has been limited by the challenge of obtaining sufficient high-quality labeled data due to the prohibitive data annotation cost. To alleviate this limitation, we propose a new text-augmented medical image segmentation model LViT (Language meets Vision Transformer). In our LViT model, medical text annotation is incorporated to compensate for the quality deficiency in image data. In addition, the text information can guide to generate pseudo labels of improved quality in the semi-supervised learning. We also propose an Exponential Pseudo label Iteration mechanism (EPI) to help the Pixel-Level Attention Module (PLAM) preserve local image features in semi-supervised LViT setting. In our model, LV (Language-Vision) loss is designed to supervise the training of unlabeled images using text information directly. For evaluation, we construct three multimodal medical segmentation datasets (image + text) containing X-rays and CT images. Experimental results show that our proposed LViT has superior segmentation performance in both fully-supe
原文 arXiv:2206.14718;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2206.14718v4