arXiv:2605.20760cs.CV2026-05

轻量级3D U-Net实现脊柱CT快速精准分割,适合边缘设备部署。

SpineContextResUNet: A Computationally Efficient Residual UNet for Spine CT Segmentation

论文配图:SpineContextResUNet: A Computationally Efficient Residual UNet for Spine CT Segmentation
图 1 · 摘自论文原文
  • 用并行多膨胀卷积设计轻量化上下文模块,捕捉长程解剖关系。
  • 在VerSe2020和CTSpine1K上分别取得88.17%和88.13%的Dice分数。
  • 仅需1.7M参数,在低配硬件上稳定运行,适合临床边缘部署。

CT扫描中椎体自动分割是病理评估与手术规划的前提。现有先进方法如基于Transformer或大规模集成模型,需大量GPU资源,难以在资源受限环境或边缘设备上应用。为此,本文提出SpineContextResUNet,一种计算高效的3D残差U-Net,用于快速脊柱定位。其架构引入轻量级上下文模块,采用并行多膨胀卷积捕获长程解剖依赖,避免RNN的高延迟或自注意力机制的内存开销。在两个公开基准数据集VerSe2020与CTSpine1K上,模型分别达到88.17%与88.13%的Dice分数。为评估硬件限制下的表现,将瓶颈版SwinUNETR缩放到约1.7M参数以匹配本模型规模;在有限数据下,受缺乏空间归纳偏置影响,该变压器模型性能严重下降。而本模型基于CNN,仍保持高精度。值得注意的是,总分割器(TotalSegmentator)在消费级硬件(Intel Core i5,8GB RAM)上因内存溢出无法运行,而本模型可在类似设备上实现稳定推理,适用于床旁诊断及Nvidia Jetson Orin Nano等边缘平台部署。

原文摘要 · Abstract (English)

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.

医学图像3D分割边缘计算轻量模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。