mlsanityData to 7 Oct 2026

SpineContextResUNet: A Computationally Efficient Residual UNet for Spine CT Segmentation

arXiv 2605.20760 ↗First posted 20 May 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Automated segmentation of the vertebral column in Computed Tomography (CT) scans is a prerequisite for pathological assessment and surgical planning. However, state-of-the-art methods, particularly those based on Transformers or large-scale ensembles, demand substantial GPU resources, creating a barrier for clinical adoption in resource-constrained environments or on edge devices. To address this, we introduce SpineContextResUNet, a computationally efficient 3D Residual U-Net designed for rapid spinal localization. Our architecture integrates a lightweight Context Block that employs parallel multi-dilated convolutions to capture long-range anatomical dependencies without the high latency of Recurrent Neural Networks (RNNs) or the memory overhead of Self-Attention mechanisms. Extensive validation on two public benchmarks, VerSe2020 and CTSpine1K, demonstrates that our model achieves a Dice score of 88.17% and 88.13% respectively. To evaluate performance under strict hardware constraints, we compared our model against a bottlenecked SwinUNETR scaled to match our ~1.7M hardware footprint. While the constrained Transformer suffers severe performance degradation due to a lack of spatial inductive biases in a limited-data regime, our CNN-based approach successfully maintains high accuracy. Crucially, heavy baselines like TotalSegmentator fail due to memory exhaustion on commodity hardware (Intel Core i5, 8GB RAM), our model performs robust inference, making it a viable solution for point-of-care diagnostics and deployment on edge platforms like the Nvidia Jetson Orin Nano.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 16 of its 21 references that are in the map. First posted on arXiv 20 May 2026.

Builds on

16 of 21 references.

References · 16

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  2. 2023Segment Anythingcited by 8,158
  3. 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
  4. 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,239
  5. 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,284
  6. 2017Rethinking Atrous Convolution for Semantic Image Segmentationcited by 2,154
  7. 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,471
  8. 2015Multi-Scale Context Aggregation by Dilated Convolutionscited by 1,128
  9. 2022TotalSegmentator: robust segmentation of 104 anatomical structures in CT imagescited by 597
  10. 2019ResUNet-a: a deep learning framework for semantic segmentation of remotely sensed datacited by 260
  11. 2020VerSe: A Vertebrae Labelling and Segmentation Benchmark for Multi-detector CT Imagescited by 111
  12. 2018Combo Loss: Handling Input and Output Imbalance in Multi-Organ Segmentationcited by 76
4 more
  1. 2021CTSpine1K: A Large-Scale Dataset for Spinal Vertebrae Segmentation in Computed Tomographycited by 45
  2. 2021A Computed Tomography Vertebral Segmentation Dataset with Anatomical Variations and Multi-Vendor Scanner Datacited by 28
  3. 2017A 3D Coarse-to-Fine Framework for Volumetric Medical Image Segmentationcited by 20
  4. 2018Joint Vertebrae Identification and Localization in Spinal CT Images by Combining Short- and Long-Range Contextual Informationcited by 7

Built on by

Not cited yet.