Abstract
This paper presents a novel approach to enhance the detection and segmentation of small liver lesions in computed tomography (CT) scans using a size-focused multi-model framework. Current state-of-the-art segmentation models, primarily based on the UNet architecture, often exhibit inferior performance on small lesions due to severe class and size imbalances. We introduce a model architecture incorporating a configurable attention mechanism within the model’s skip connections and a lesion selection algorithm that compares predictions from multiple models, including a general lesion segmentation model and a small lesion-focused model, selecting the most suitable prediction. The approach was evaluated on a clinical 3-phase CT dataset and the public LiTS dataset. Results show improvements in overall lesion segmentation performance by 1.5% and 1.9% for the clinical and LiTS datasets, respectively. Additionally, the detection of small lesions improved by 4.4% and 1.8% for both datasets, respectively.
| Original language | English |
|---|---|
| Title of host publication | Machine Learning in Medical Imaging - 15th International Workshop, MLMI 2024, Held in Conjunction with MICCAI 2024, Proceedings |
| Editors | Xuanang Xu, Zhiming Cui, Kaicong Sun, Islem Rekik, Xi Ouyang |
| Publisher | Springer Science and Business Media Deutschland GmbH |
| Pages | 320-330 |
| Number of pages | 11 |
| ISBN (Print) | 9783031732836 |
| DOIs | |
| State | Published - 2025 |
| Event | 15th International Workshop on Machine Learning in Medical Imaging, MLMI 2024 was held in conjunction with the 27th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2024 - Marrakesh, Morocco Duration: 6 Oct 2024 → 6 Oct 2024 |
Publication series
| Name | Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics) |
|---|---|
| Volume | 15241 LNCS |
| ISSN (Print) | 0302-9743 |
| ISSN (Electronic) | 1611-3349 |
Conference
| Conference | 15th International Workshop on Machine Learning in Medical Imaging, MLMI 2024 was held in conjunction with the 27th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2024 |
|---|---|
| Country/Territory | Morocco |
| City | Marrakesh |
| Period | 6/10/24 → 6/10/24 |
Bibliographical note
Publisher Copyright:© The Author(s), under exclusive license to Springer Nature Switzerland AG 2025.
Keywords
- Convolutional neural networks
- Liver lesion segmentation
- Medical image segmentation
- Small lesion segmentation
ASJC Scopus subject areas
- Theoretical Computer Science
- General Computer Science