학술논문

Fine-tuned feature selection to improve prostate segmentation via a fully connected meta-learner architecture
Document Type
Conference
Source
2022 IEEE-EMBS International Conference on Biomedical and Health Informatics (BHI) Biomedical and Health Informatics (BHI), 2022 IEEE-EMBS International Conference on. :01-04 Sep, 2022
Subject
Bioengineering
Computing and Processing
Signal Processing and Analysis
Deep learning
Sensitivity
Magnetic resonance imaging
Glands
Feature extraction
Complexity theory
Prostate cancer
Prostate Segmentation
Deep Learning
Ensembling
Fine Tuning
Language
ISSN
2641-3604
Abstract
Precise delineation of the prostate gland on MRI is the cornerstone for accurate prostate cancer diagnosis, detection, characterization and treatment. The present work proposes a meta-learner deep learning (DL) network that combines the complexity of 3 well-established DL models and fine tune them in order to improve the segmentation of the prostate compared to the base learners. The backbone of the meta-learner consist the original U-net, Dense2U-net and Bridged U-net models. A model was added on top of the three base networks that has four convolutions with different receptor fields. The meta-learner outperformed the base-learners in 4 out of 5 performance metrics. The median Dice Score for the meta-learner was 89% while for the second best model it was 83%. Except for Hausdorff distance, where the meta-learner and Dense2U-net performed equally well, the improvement achieved in terms of average sensitivity, balanced accuracy, dice score and rand error, compared to the best performing base-learner, was 6%, 3%, 5% and 4%, respectively.