通过双瓶颈监督压缩特征,实现高效医学图像分割。
Tree-NET: Enhancing 2D Medical Image Segmentation Through Efficient Low-Level Feature Training
- 在输入和输出端同时使用瓶颈特征压缩,降低计算负担。
- 相比基线模型,减少4至13倍的计算量,内存占用更低。
- 可无缝集成到现有模型中,适合医疗影像实时处理场景。
本文提出Tree-NET,一种新型医学图像分割框架,通过双瓶颈监督提升分割精度与计算效率。以往研究仅在训练阶段使用瓶颈特征监督,无法带来推理时的加速。Tree-NET首次在输入与输出阶段均引入潜在空间特征监督,以极低参数开销压缩输入与标签维度,同时保持精度。其三组件结构包括:编码器-网(Encoder-Net)与解码器-网(Decoder-Net),用于自动编码压缩数据;桥接-网(Bridge-Net)则在压缩表示上训练分割模型。整个系统基于密集低维特征运行,显著提升推理效率,并可无修改接入现有分割模型。在皮肤病变与息肉分割任务上评估,使用U-NET、U-NET++与Polyp-PVT等主干网络,实验表明,Tree-NET使FLOPs降低4至13倍,内存消耗减少,且保持与基线相当的精度。例如,在ISIC 2018数据集上,未训练的U-NET++基线Dice分数从0.829提升至0.862,IoU从0.736升至0.787;在CVC-ClinicDB上,使用Polyp-PVT主干,达到Dice 0.946,IoU 0.901,性能匹配或超越基线。结果表明,Tree-NET是医学图像分割的高效可靠方案。
原文摘要 · Abstract (English)
This paper introduces Tree-NET, a novel framework for medical image segmentation that leverages bottleneck supervision to enhance both segmentation accuracy and computational efficiency. While previous studies have applied bottleneck feature supervision to segmentation tasks, it has typically been limited to the training phase, offering no computational benefits during inference. To the best of our knowledge, this is the first framework to employ dual bottleneck supervision for segmentation, leveraging latent space features at both the input and output stages. This approach reduces input and label dimensions with minimal parameter overhead while preserving accuracy. Tree-NET features a three-component architecture: Encoder-Net and Decoder-Net, which compress input and label data via autoencoding, and Bridge-Net, a segmentation model trained on these compressed representations. By operating entirely on dense, low-dimensional features, Tree-NET improves runtime efficiency and can be integrated into existing segmentation models without modifying their internal structures or increasing model size. We evaluate Tree-NET on two key segmentation tasks: skin lesion and polyp segmentation using various backbone models, including U-NET, U-NET++, and Polyp-PVT. Experimental results show that Tree-NET reduces FLOPs by a factor of 4 to 13 and decreases memory usage while maintaining segmentation accuracy comparable to baseline models. For example, with an untrained U-NET++ backbone, Tree-NET improves the Dice score on ISIC 2018 from 0.829 to 0.862 and the IoU from 0.736 to 0.787. On CVC-ClinicDB, it achieves a Dice score of 0.946 and an IoU of 0.901 using a Polyp-PVT backbone, matching or surpassing baseline performance. These findings underscore Tree-NET's potential as a robust and efficient solution for medical image segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。