Planar 3D Transfer Learning for End to End Unimodal MRI Unbalanced Data Segmentation

Result code in IS VaVaI

<a href="https://www.isvavai.cz/riv?ss=detail&h=RIV%2F00216305%3A26220%2F21%3APU141428" target="_blank" >RIV/00216305:26220/21:PU141428 - isvavai.cz</a>

Result on the web

<a href="https://link.springer.com/chapter/10.1007/978-3-030-76423-4_10" target="_blank" >https://link.springer.com/chapter/10.1007/978-3-030-76423-4_10</a>

DOI - Digital Object Identifier

<a href="http://dx.doi.org/10.1109/ICPR48806.2021.9412150" target="_blank" >10.1109/ICPR48806.2021.9412150</a>

Result language

angličtina

Original language name

Planar 3D Transfer Learning for End to End Unimodal MRI Unbalanced Data Segmentation

Original language description

We present a novel approach of 2D to 3D transfer learning based on mapping pre-trained 2D convolutional neural network weights into planar 3D kernels. The method is validated by the proposed planar 3D res-u-net network with encoder transferred from the 2D VGG-16, which is applied for a single-stage unbalanced 3D image data segmentation. In particular, we evaluate the method on the MICCAI 2016 MS lesion segmentation challenge dataset utilizing solely fluid-attenuated inversion recovery (FLAIR) sequence without brain extraction for training and inference to simulate real medical praxis. The planar 3D res-u-net network performed the best both in sensitivity and Dice score amongst end to end methods processing raw MRI scans and achieved comparable Dice score to a state-of-the-art unimodal not end to end approach. Complete source code was released under the open-source license, and this paper complies with the Machine learning reproducibility checklist. By implementing practical transfer learning for 3D data representation, we could segment heavily unbalanced data without selective sampling and achieved more reliable results using less training data in a single modality. From a medical perspective, the unimodal approach gives an advantage in real praxis as it does not require co-registration nor additional scanning time during an examination. Although modern medical imaging methods capture high-resolution 3D anatomy scans suitable for computer-aided detection system processing, deployment of automatic systems for interpretation of radiology imaging is still rather theoretical in many medical areas. Our work aims to bridge the gap by offering a solution for partial research questions.

Czech name

—

Czech description

—

Type

D - Article in proceedings

CEP classification

—

OECD FORD branch

10201 - Computer sciences, information science, bioinformathics (hardware development to be 2.2, social aspect to be 5.8)

Project

—

Continuities

S - Specificky vyzkum na vysokych skolach

Publication year

2021

Confidentiality

S - Úplné a pravdivé údaje o projektu nepodléhají ochraně podle zvláštních právních předpisů

Article name in the collection

2020 25th International Conference on Pattern Recognition (ICPR)

ISBN

978-1-7281-8808-9

ISSN

—

e-ISSN

—

Number of pages

8

Pages from-to

1-8

Publisher name

IEEE

Place of publication

Online

Event location

Milano

Event date

Jan 10, 2021

Type of event by nationality

WRD - Celosvětová akce

UT code for WoS article

000678409206025