Parameter-Efficient 3D Segmentation of Liver and Liver tumors: Depthwise factorization Scales Better Than Dense Convolution with Spatial Dimensionality
Not on the map. Why?
Abstract (arXiv)
Three-dimensional dense convolutional networks are the strongest performers on volumetric medical image segmentation, but their parameter counts scale poorly: moving a dense k x k convolution to k x k x k multiplies its weights by k. We observe that depthwise separable factorization does not share this penalty. Because the cubic kernel term applies only to the depthwise stage while the pointwise projection, which dominates the parameter count, is unchanged, the same architecture grows by 5 % from 2D to 3D where a dense convolutional U-Net grows by 200 %. We exploit this asymmetry to build a 3D U-Net with 536,990 parameters, 24x fewer than an identical dense 3D U-Net. On MSD Task03 Liver (the Medical Segmentation Decathlon liver task, derived from LiTS), evaluated per case under five-fold cross-validation over all 131 public volumes, the model reaches a tumor Dice of 0.577 (95% CI [0.518, 0.633]) and a liver Dice of 0.947 (95% CI [0.941, 0.952]). Its liver Dice exceeds previously reported performance. Its tumor Dice exceeds their low-resolution configuration (0.4701) by 0.107 and their 2D configuration (0.5394), at approximately one twenty-fourth of the parameters and roughly half the in-plane resolution. Trained under identical conditions on a common held-out split, it exceeds a dense 3D U-Net on liver by +0.031 Dice (paired p = 0.006) and on tumor by +0.041 (95 % CI [+0.005, +0.081], paired p = 0.056), suggesting the factorization also acts as a regularizer in the small-data regime characteristic of medical imaging. We further show, on both LiTS and a 2D endoscopy benchmark, that a large fraction of the network's learnable spatial filters can be replaced by fixed shifts at no cost in accuracy, but that replacing all of them is measurably worse, the placement of spatial capacity matters more than its total amount.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 30 references that are in the map. First posted on arXiv 27 Sep 2026.
Builds on
25 of 30 references.
References · 25
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
- 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,332
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
- 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,476
- 2017ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devicescited by 1,200
- 2016Instance Normalization: The Missing Ingredient for Fast Stylizationcited by 1,198
- 2021The Medical Segmentation Decathloncited by 628
- 2019Kvasir-SEG: A Segmented Polyp Datasetcited by 589
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 583
13 more
- 2019The Liver Tumor Segmentation Benchmark (LiTS)cited by 509
- 2018nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentationcited by 291
- 2022UNeXt: MLP-based Rapid Medical Image Segmentation Networkcited by 185
- 2017Shift: A Zero FLOP, Zero Parameter Alternative to Spatial Convolutionscited by 93
- 2016Automatic Liver and Lesion Segmentation in CT Using Cascaded Fully Convolutional Neural Networks and 3D Conditional Random Fieldscited by 43
- 2019All You Need is a Few Shifts: Designing Efficient Convolutional Neural Networks for Image Classificationcited by 38
- 2023CMUNeXt: An Efficient Medical Image Segmentation Network based on Large Kernel and Skip Fusioncited by 31
- 2019One Network to Segment Them All: A General, Lightweight System for Accurate 3D Medical Image Segmentationcited by 25
- 2018Constructing Fast Network through Deconstruction of Convolutioncited by 24
- 2021S2-MLPv2: Improved Spatial-Shift MLP Architecture for Visioncited by 24
- 20231M parameters are enough? A lightweight CNN-based model for medical image segmentationcited by 15
- 2024LATUP-Net: A Lightweight 3D Attention U-Net with Parallel Convolutions for Brain Tumor Segmentationcited by 7
- 2025LSU-Net: Lightweight Automatic Organs Segmentation Network For Medical Imagescited by 1
Built on by
Not cited yet.