Two-layer ensemble of deep learning models for medical image segmentation.

Dang, Truong; Nguyen, Tien Thanh; McCall, John; Elyan, Eyad; Moreno-García, Carlos Francisco

Abstract

One of the most important areas in medical image analysis is segmentation, in which raw image data is partitioned into structured and meaningful regions to gain further insights. By using Deep Neural Networks (DNN), AI-based automated segmentation algorithms can potentially assist physicians with more effective imaging-based diagnoses. However, since it is difficult to acquire high-quality ground truths for medical images and DNN hyperparameters require significant manual tuning, the results by DNN-based medical models might be limited. A potential solution is to combine multiple DNN models using ensemble learning. We propose a two-layer ensemble of deep learning models in which the prediction of 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 weight-based scheme which is found by solving linear regression problems. To the best of our knowledge, our paper is the first work which proposes a two-layer ensemble of deep learning models with an augmented data technique in medical image segmentation. Experiments conducted on five different medical image datasets for diverse segmentation tasks show that proposed method achieves better results in terms of several performance metrics compared to some well-known benchmark algorithms. Our proposed two-layer ensemble of deep learning models for segmentation of medical images shows effectiveness compared to several benchmark algorithms. The research can be expanded in several directions like image classification.

Citation

DANG, T., NGUYEN, T.T., MCCALL, J., ELYAN, E. and MORENO-GARCÍA, C.F. 2024. Two-layer ensemble of deep learning models for medical image segmentation. Cognitive computation [online], 16(3), pages 1141-1160. Available from: https://doi.org/10.1007/s12559-024-10257-5

• Journal Article Type: Article
• Acceptance Date: Jan 11, 2024
• Online Publication Date: Jan 31, 2024
• Publication Date: May 31, 2024
• Deposit Date: Feb 1, 2024
• Publicly Available Date: Feb 13, 2024
• Journal: Cognitive computation
• Print ISSN: 1866-9956
• Electronic ISSN: 1866-9964
• Publisher: Springer
• Peer Reviewed: Peer Reviewed
• Volume: 16
• Issue: 3
• Pages: 1141-1160
• DOI: https://doi.org/10.1007/s12559-024-10257-5
• Keywords: Image segmentation; Ensemble method; Ensemble learning; Deep learning; Medical image
• Public URL: https://rgu-repository.worktribe.com/output/2217607