Self-Supervised Alignment Learning for Medical Image Segmentation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Haofeng, Ouyang, Yiming, Wan, Xiang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917701617713152
author Li, Haofeng
Ouyang, Yiming
Wan, Xiang
author_facet Li, Haofeng
Ouyang, Yiming
Wan, Xiang
contents Recently, self-supervised learning (SSL) methods have been used in pre-training the segmentation models for 2D and 3D medical images. Most of these methods are based on reconstruction, contrastive learning and consistency regularization. However, the spatial correspondence of 2D slices from a 3D medical image has not been fully exploited. In this paper, we propose a novel self-supervised alignment learning framework to pre-train the neural network for medical image segmentation. The proposed framework consists of a new local alignment loss and a global positional loss. We observe that in the same 3D scan, two close 2D slices usually contain similar anatomic structures. Thus, the local alignment loss is proposed to make the pixel-level features of matched structures close to each other. Experimental results show that the proposed alignment learning is competitive with existing self-supervised pre-training approaches on CT and MRI datasets, under the setting of limited annotations.
format Preprint
id arxiv_https___arxiv_org_abs_2406_15699
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Self-Supervised Alignment Learning for Medical Image Segmentation
Li, Haofeng
Ouyang, Yiming
Wan, Xiang
Computer Vision and Pattern Recognition
Recently, self-supervised learning (SSL) methods have been used in pre-training the segmentation models for 2D and 3D medical images. Most of these methods are based on reconstruction, contrastive learning and consistency regularization. However, the spatial correspondence of 2D slices from a 3D medical image has not been fully exploited. In this paper, we propose a novel self-supervised alignment learning framework to pre-train the neural network for medical image segmentation. The proposed framework consists of a new local alignment loss and a global positional loss. We observe that in the same 3D scan, two close 2D slices usually contain similar anatomic structures. Thus, the local alignment loss is proposed to make the pixel-level features of matched structures close to each other. Experimental results show that the proposed alignment learning is competitive with existing self-supervised pre-training approaches on CT and MRI datasets, under the setting of limited annotations.
title Self-Supervised Alignment Learning for Medical Image Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.15699