IMSAHLO: Integrating Multi-Scale Attention and Hybrid Loss Optimization Framework for Robust Neuronal Brain Cell Segmentation
Not on the map. Why?
Abstract (arXiv)
Accurate segmentation of neuronal cells in fluorescence microscopy is a fundamental task for quantitative analysis in computational neuroscience. However, it is significantly impeded by challenges such as the coexistence of densely packed and sparsely distributed cells, complex overlapping morphologies, and severe class imbalance. Conventional deep learning models often fail to preserve fine topological details or accurately delineate boundaries under these conditions. To address these limitations, we propose a novel deep learning framework, IMSAHLO (Integrating Multi-Scale Attention and Hybrid Loss Optimization), for robust and adaptive neuronal segmentation. The core of our model features Multi-Scale Dense Blocks (MSDBs) to capture features at various receptive fields, effectively handling variations in cell density, and a Hierarchical Attention (HA) mechanism that adaptively focuses on salient morphological features to preserve Region of Interest (ROI) boundary details. Furthermore, we introduce a novel hybrid loss function synergistically combining Tversky and Focal loss to combat class imbalance, alongside a topology-aware Centerline Dice (clDice) loss and a Contour-Weighted Boundary loss to ensure topological continuity and precise separation of adjacent cells. Large-scale experiments on the public Fluorescent Neuronal Cells (FNC) dataset demonstrate that our framework outperforms state-of-the-art architectures, achieving precision of 81.4%, macro F1 score of 82.7%, micro F1 score of 83.3%, and balanced accuracy of 99.5% on difficult dense and sparse cases. Ablation studies validate the synergistic benefits of multi-scale attention and hybrid loss terms. This work establishes a foundation for generalizable segmentation models applicable to a wide range of biomedical imaging modalities, pushing AI-assisted analysis toward high-throughput neurobiological pipelines.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 28 of its 62 references that are in the map. First posted on arXiv 14 Jan 2026.
Builds on
28 of 62 references.
References · 28
- 2017Attention Is All You Needcited by 59,932
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2017Squeeze-and-Excitation Networkscited by 4,763
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,210
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
- 2021MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformercited by 636
- 2017Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentationscited by 614
- 2019ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networkscited by 572
- 2017Tversky loss function for image segmentation using 3D fully convolutional deep networkscited by 228
- 2020A survey of loss functions for semantic segmentationcited by 216
16 more
- 2018Boundary loss for highly unbalanced segmentationcited by 187
- 2022UNeXt: MLP-based Rapid Medical Image Segmentation Networkcited by 185
- 2018A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentationcited by 171
- 2021Unified Focal loss: Generalising Dice and cross entropy-based losses to handle class imbalanced medical image segmentationcited by 88
- 2018Recalibrating Fully Convolutional Networks with Spatial and Channel 'Squeeze & Excitation' Blockscited by 75
- 2023ResiDual: Transformer with Dual Residual Connectionscited by 30
- 2022Attention Mechanism in Neural Networks: Where it Comes and Where it Goescited by 29
- 2024Centerline Boundary Dice Loss for Vascular Segmentationcited by 27
- 2023U-Net v2: Rethinking the Skip Connections of U-Net for Medical Image Segmentationcited by 23
- 2024Understanding Test-Time Augmentationcited by 23
- 2017Multi-Scale Dense Networks for Resource Efficient Image Classificationcited by 15
- 2023Channel prior convolutional attention for medical image segmentationcited by 8
- 2019Hierarchical Attention Networks for Medical Image Segmentationcited by 5
- 2024Contour-weighted loss for class-imbalanced image segmentationcited by 3
- 2021Contour Loss for Instance Segmentation via k-step Distance Transformation Imagecited by 1
- 2025Steerable Pyramid Weighted Loss: Multi-Scale Adaptive Weighting for Semantic Segmentationcited by 1
Built on by
Not cited yet.