Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Johnson-Lindenstrauss Lemma Guided Network for Efficient 3D Medical Segmentation

About

Lightweight 3D medical image segmentation remains constrained by a fundamental \textit{``efficiency / robustness conflict''}, particularly when processing complex anatomical structures and heterogeneous modalities. In this paper, we study how to redesign the framework based on the characteristics of high-dimensional 3D images, and explore data synergy to overcome the fragile representation of lightweight methods. Our approach, VeloxSeg, begins with a deployable and extensible dual-stream CNN-Transformer architecture composed of Paired Window Attention (PWA) and Johnson-Lindenstrauss lemma-guided convolution (JLC). For each 3D image, we invoke a ``glance-and-focus'' principle, where PWA rapidly retrieves multi-scale information, and JLC ensures robust local feature extraction with minimal parameters, significantly enhancing the model's ability to operate with low computational budget. Followed by an extension of the dual-stream architecture that incorporates modal interaction into the multi-scale image-retrieval process, VeloxSeg efficiently models heterogeneous modalities. Finally, Spatially Decoupled Knowledge Transfer (SDKT) via Gram matrices injects the texture prior extracted by a self-supervised network into the segmentation network, yielding stronger representations than baselines at no extra inference cost. Experimental results on multimodal benchmarks show that VeloxSeg achieves a 26\% Dice improvement, alongside increasing GPU throughput by 11$\times$, CPU by 48$\times$, and reducing training peak GPU memory usage by $1/20$, inference by $1/24$. Code is available at https://github.com/JinPLu/VeloxSeg.

Jinpeng Lu, Linghan Cai, Yinda Chen, Guo Tang, Songhan Jiang, Haoyuan Shi, Zhiwei Xiong• 2025

Related benchmarks

TaskDatasetResultRank
Brain Tumor SegmentationBraTS 2021
DSC Avg91.44
24
SegmentationAutoPET II
Dice62.51
21
SegmentationHecktor 2022
Dice Score56.48
21
Medical Image SegmentationAutoPET II
ThrG Score390.9
17
Medical Image SegmentationHecktor 2022
ThrG319.8
17
Brain Tumor SegmentationBraTS 2021
Parameters (M)1.46
17
Medical Image SegmentationAutoPET II
Peak GPU Memory Usage (Training)842
17
Medical Image SegmentationHecktor 2022
Peak GPU Memory Usage (Training)1.01e+3
17
Showing 8 of 8 rows

Other info

Follow for update