Learning what and where to segment : A new perspective on medical image few-shot segmentation

Copyright © 2023 The Author(s). Published by Elsevier B.V. All rights reserved..

Traditional medical image segmentation methods based on deep learning require experts to provide extensive manual delineations for model training. Few-shot learning aims to reduce the dependence on the scale of training data but usually shows poor generalizability to the new target. The trained model tends to favor the training classes rather than being absolutely class-agnostic. In this work, we propose a novel two-branch segmentation network based on unique medical prior knowledge to alleviate the above problem. Specifically, we explicitly introduce a spatial branch to provide the spatial information of the target. In addition, we build a segmentation branch based on the classical encoder-decoder structure in supervised learning and integrate prototype similarity and spatial information as prior knowledge. To achieve effective information integration, we propose an attention-based fusion module (AF) that enables the content interaction of decoder features and prior knowledge. Experiments on an echocardiography dataset and an abdominal MRI dataset show that the proposed model achieves substantial improvements over state-of-the-art methods. Moreover, some results are comparable to those of the fully supervised model. The source code is available at github.com/warmestwind/RAPNet.

Medienart:

E-Artikel

Erscheinungsjahr:

2023

Erschienen:

2023

Enthalten in:

Zur Gesamtaufnahme - volume:87

Enthalten in:

Medical image analysis - 87(2023) vom: 15. Juli, Seite 102834

Sprache:

Englisch

Beteiligte Personen:

Feng, Yong [VerfasserIn]
Wang, Yonghuai [VerfasserIn]
Li, Honghe [VerfasserIn]
Qu, Mingjun [VerfasserIn]
Yang, Jinzhu [VerfasserIn]

Links:

Volltext

Themen:

Echocardiography
Few-shot segmentation
Journal Article
Medical prior knowledge
Organ segmentation
Research Support, Non-U.S. Gov't

Anmerkungen:

Date Completed 06.06.2023

Date Revised 08.06.2023

published: Print-Electronic

Citation Status MEDLINE

doi:

10.1016/j.media.2023.102834

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM357100360