Active Deep Learning with Fisher Information for Patch-wise Semantic Segmentation

Deep learning with convolutional neural networks (CNN) has achieved unprecedented success in segmentation, however it requires large training data, which is expensive to obtain. Active Learning (AL) frameworks can facilitate major improvements in CNN performance with intelligent selection of minimal data to be labeled. This paper proposes a novel diversified AL based on Fisher information (FI) for the first time for CNNs, where gradient computations from backpropagation are used for efficient computation of FI on the large CNN parameter space. We evaluated the proposed method in the context of newborn and adolescent brain extraction problem under two scenarios: (1) semi-automatic segmentation of a particular subject from a different age group or with a pathology not available in the original training data, where starting from an inaccurate pre-trained model, we iteratively label small number of voxels queried by AL until the model generates accurate segmentation for that subject, and (2) using AL to build a universal model generalizable to all images in a given data set. In both scenarios, FI-based AL improved performance after labeling a small percentage (less than 0.05%) of voxels. The results showed that FI-based AL significantly outperformed random sampling, and achieved accuracy higher than entropy-based querying in transfer learning, where the model learns to extract brains of newborn subjects given an initial model trained on adolescents.

Medienart:

E-Artikel

Erscheinungsjahr:

2018

Erschienen:

2018

Enthalten in:

Zur Gesamtaufnahme - volume:11045

Enthalten in:

Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support : 4th International Workshop, DLMIA 2018, and 8th International Workshop, ML-CDS 2018, held in conjunction with MICCAI 2018, Granada, Spain, S - 11045(2018) vom: 14. Sept., Seite 83-91

Sprache:

Englisch

Beteiligte Personen:

Sourati, Jamshid [VerfasserIn]
Gholipour, Ali [VerfasserIn]
Dy, Jennifer G [VerfasserIn]
Kurugol, Sila [VerfasserIn]
Warfield, Simon K [VerfasserIn]

Links:

Volltext

Themen:

Journal Article

Anmerkungen:

Date Revised 01.10.2020

published: Print-Electronic

Citation Status PubMed-not-MEDLINE

doi:

10.1007/978-3-030-00889-5_10

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM290825733