Mr Truong Dang t.dang1@rgu.ac.uk
Research Assistant
Two layer ensemble of deep learning models for medical image segmentation. [Preprint]
Dang, Truong; Nguyen, Tien Thanh; McCall, John; Elyan, Eyad; Moreno-García, Carlos Francisco
Authors
Dr Thanh Nguyen t.nguyen11@rgu.ac.uk
Senior Research Fellow
Professor John McCall j.mccall@rgu.ac.uk
Professorial Lead
Professor Eyad Elyan e.elyan@rgu.ac.uk
Professor
Dr Carlos Moreno-Garcia c.moreno-garcia@rgu.ac.uk
Associate Professor
Abstract
In recent years, deep learning has rapidly become a method of choice for the segmentation of medical images. Deep Neural Network (DNN) architectures such as UNet have achieved state-of-the-art results on many medical datasets. To further improve the performance in the segmentation task, we develop an ensemble system which combines various deep learning architectures. We propose a two-layer ensemble of deep learning models for the segmentation of medical images. The prediction for each training image pixel made by each model in the first layer is used as the augmented data of the training image for the second layer of the ensemble. The prediction of the second layer is then combined by using a weights-based scheme in which each model contributes differently to the combined result. The weights are found by solving linear regression problems. Experiments conducted on two popular medical datasets namely CAMUS and Kvasir-SEG show that the proposed method achieves better results concerning two performance metrics (Dice Coefficient and Hausdorff distance) compared to some well-known benchmark algorithms.
Citation
DANG, T., NGUYEN, T.T., MCCALL, J., ELYAN, E. and MORENO-GARCÍA, C.F. 2021. Two layer ensemble of deep learning models for medical image segmentation. arXiv [online]. Available from: https://doi.org/10.48550/arXiv.2104.04809
Working Paper Type | Preprint |
---|---|
Deposit Date | Jun 17, 2021 |
Publicly Available Date | Jan 30, 2023 |
DOI | https://doi.org/10.48550/arXiv.2104.04809 |
Keywords | Image analysis; Computers in medical imaging; Deep neural networks; Machine learning |
Public URL | https://rgu-repository.worktribe.com/output/1364443 |
Files
DANG 2021 Two layer ensemble of deep (PREPRINT)
(1.4 Mb)
PDF
Publisher Licence URL
https://creativecommons.org/licenses/by/4.0/
You might also like
A multimodel-based screening framework for C-19 using deep learning-inspired data fusion.
(2024)
Journal Article
Downloadable Citations
About OpenAIR@RGU
Administrator e-mail: publications@rgu.ac.uk
This application uses the following open-source libraries:
SheetJS Community Edition
Apache License Version 2.0 (http://www.apache.org/licenses/)
PDF.js
Apache License Version 2.0 (http://www.apache.org/licenses/)
Font Awesome
SIL OFL 1.1 (http://scripts.sil.org/OFL)
MIT License (http://opensource.org/licenses/mit-license.html)
CC BY 3.0 ( http://creativecommons.org/licenses/by/3.0/)
Powered by Worktribe © 2024
Advanced Search