Skip and Skip: Segmenting Medical Images with Prompts

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chen, Jiawei, Yang, Dingkang, Lei, Yuxuan, Zhang, Lihua
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913399920656384
author Chen, Jiawei
Yang, Dingkang
Lei, Yuxuan
Zhang, Lihua
author_facet Chen, Jiawei
Yang, Dingkang
Lei, Yuxuan
Zhang, Lihua
contents Most medical image lesion segmentation methods rely on hand-crafted accurate annotations of the original image for supervised learning. Recently, a series of weakly supervised or unsupervised methods have been proposed to reduce the dependence on pixel-level annotations. However, these methods are essentially based on pixel-level annotation, ignoring the image-level diagnostic results of the current massive medical images. In this paper, we propose a dual U-shaped two-stage framework that utilizes image-level labels to prompt the segmentation. In the first stage, we pre-train a classification network with image-level labels, which is used to obtain the hierarchical pyramid features and guide the learning of downstream branches. In the second stage, we feed the hierarchical features obtained from the classification branch into the downstream branch through short-skip and long-skip and get the lesion masks under the supervised learning of pixel-level labels. Experiments show that our framework achieves better results than networks simply using pixel-level annotations.
format Preprint
id arxiv_https___arxiv_org_abs_2406_14958
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Skip and Skip: Segmenting Medical Images with Prompts
Chen, Jiawei
Yang, Dingkang
Lei, Yuxuan
Zhang, Lihua
Computer Vision and Pattern Recognition
Most medical image lesion segmentation methods rely on hand-crafted accurate annotations of the original image for supervised learning. Recently, a series of weakly supervised or unsupervised methods have been proposed to reduce the dependence on pixel-level annotations. However, these methods are essentially based on pixel-level annotation, ignoring the image-level diagnostic results of the current massive medical images. In this paper, we propose a dual U-shaped two-stage framework that utilizes image-level labels to prompt the segmentation. In the first stage, we pre-train a classification network with image-level labels, which is used to obtain the hierarchical pyramid features and guide the learning of downstream branches. In the second stage, we feed the hierarchical features obtained from the classification branch into the downstream branch through short-skip and long-skip and get the lesion masks under the supervised learning of pixel-level labels. Experiments show that our framework achieves better results than networks simply using pixel-level annotations.
title Skip and Skip: Segmenting Medical Images with Prompts
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.14958