SVNC-Net: An optimized U-Net variant with 2D convolutions for lightweight 3D spleen segmentation

TL;DR

SVNC-Net is a lightweight 3D spleen segmentation model using 2D convolutions, designed for efficient and accurate spleen volume measurement in CT scans.

Contribution

SVNC-Net introduces an optimized U-Net variant using 2D convolutions and depthwise separable convolutions for efficient 3D spleen segmentation.

Findings

SVNC-Net achieves high segmentation accuracy while reducing computational and memory demands.

The model was evaluated against other CNN-based models on two publicly available datasets.

Post-training compression techniques like pruning and quantization further improve model efficiency.

Abstract

Accurate measurement of spleen volume is essential for the diagnosis of splenomegaly. While Computed Tomography (CT) is among the most reliable imaging modalities for this task, manual segmentation of the spleen is labor-intensive and impractical for routine clinical workflows. Automatic segmentation methods provide a more viable alternative for clinical deployment. In recent years, 3D Convolutional Neural Network (CNN) models have been widely used for this purpose due to their high segmentation accuracy. However, their computational and memory demands make them less suitable for real-time applications on edge devices with limited processing capabilities. To address these limitations, we introduce SVNC-Net (Spleen Volume and Neighborhood Convolutional Network) for efficient 3D spleen segmentation from CT scans. Rather than developing an entirely new architecture from scratch, SVNC-Net builds upon the U-Net framework with targeted architectural optimizations for efficiency. In SVNC-Net, each CT slice is processed independently using 2D convolutions. In its architecture, depthwise separable convolution is used to significantly reduce computational complexity and memory usage. To evaluate its performance and efficiency, a comparative analysis was conducted against well-known CNN-based models, including UPerNet, EMANet, CCNet, SegNet, and ShuffleNet. This evaluation was performed on two publicly available datasets used together for the first time in the literature. The promising results achieved from the comparative analysis verified that SVNC-Net is highly suitable for real-time applications and resource-constrained environments. Additionally, we explore post-training compression techniques such as pruning and quantization, which further enhance the model’s compactness and inference speed. These findings contribute to the ongoing efforts to develop efficient 2D deep learning models for 3D organ segmentation, particularly in resource-constrained clinical scenarios.

Figures