Enhanced local feature extraction of lite network with scale-invariant CNN for precise segmentation of small brain tumors in MRI

Wei Yuan and Han Kang

PLOS ONE, 2025, vol. 20, issue 10, 1-17

Abstract: Deep learning has emerged as the preeminent technique for semantic segmentation of brain MRI tumors. However, existing methods often rely on hierarchical downsampling to generate multi-scale feature maps, effectively capturing fine-grained global features but struggling with large-scale local features due to insufficient network depth. This limitation is particularly detrimental for segmenting diminutive targets such as brain tumors, where local feature extraction is crucial. Augmenting network depth to address this issue leads to excessive parameter counts, incompatible with resource-constrained devices. To tackle this challenge, we propose that object recognition should exhibit scale invariance, so we introduce a shared CNN network architecture for image encoding. The input MRI image is directly downsampled into three scales, with a shared 10-layer convolutional network employed across all scales to extract features. This approach enhances the network’s ability to capture large-scale local features without increasing the total parameter count. Further, we utilize a Transformer on the smallest scale to extract global features. The decoding stage follows the UNet structure, incorporating incremental upsampling and feature fusion from previous scales. Comparative experiments on the LGG Segmentation Dataset and BraTS21 dataset demonstrate that our proposed LiteMRINet achieves higher segmentation accuracy while significantly reducing parameter count. This makes our approach particularly advantageous for devices with limited memory resources. Our code is available at https://github.com/chinaericy/MRINet.

Date: 2025
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0334447 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 34447&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0334447

DOI: 10.1371/journal.pone.0334447

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().