arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.23903cs.CVcs.LG

SegMate: 基于非对称注意力机制的轻量级架构用于高效的多器官分割

SegMate: Asymmetric Attention-Based Lightweight Architecture for Efficient Multi-Organ Segmentation

POLITEHNICA Bucharest(巴奇利大学);University of Bucharest(布加勒斯特大学)

Andrei-Alexandru Bunea, Dan-Matei Popovici, Radu Tudor Ionescu

AI总结:

SegMate 通过非对称架构和注意力机制,实现高效的多器官分割,减少计算资源需求并提升分割性能。

摘要:

当前最先进的医学图像分割模型在准确性方面表现优异,但需要大量的计算资源,限制了在资源有限的临床环境中的部署。我们提出了SegMate,一种高效的2.5D框架,实现了最先进的准确性,同时显著降低了计算需求。我们的高效设计是通过精心整合非对称架构、注意力机制、多尺度特征融合、基于切片的位置条件和多任务优化而实现的。我们展示了该框架在三个现代骨干网络(EfficientNetV2-M、MambaOut-Tiny、FastViT-T12)上的效率-准确性权衡。我们在三个数据集上进行了实验:TotalSegmentator、SegTHOR和AMOS22。与 vanilla 模型相比,SegMate 将计算(GFLOPs)减少高达 2.5 倍,内存足迹(VRAM)减少高达 2.1 倍,同时通常实现了约 1% 的性能提升。在 TotalSegmentator 上,我们仅使用 295MB 峰值 GPU 内存就实现了 93.51% 的 Dice 分数。在 SegTHOR 和 AMOS22 上的零样本跨数据集评估显示了强大的泛化能力,分别达到 86.85% 和 89.35% 的 Dice 分数。我们开源代码在 https://github.com/andreibunea99/SegMate 上。

英文摘要:

State-of-the-art models for medical image segmentation achieve excellent accuracy but require substantial computational resources, limiting deployment in resource-constrained clinical settings. We present SegMate, an efficient 2.5D framework that achieves state-of-the-art accuracy, while considerably reducing computational requirements. Our efficient design is the result of meticulously integrating asymmetric architectures, attention mechanisms, multi-scale feature fusion, slice-based positional conditioning, and multi-task optimization. We demonstrate the efficiency-accuracy trade-off of our framework across three modern backbones (EfficientNetV2-M, MambaOut-Tiny, FastViT-T12). We perform experiments on three datasets: TotalSegmentator, SegTHOR and AMOS22. Compared with the vanilla models, SegMate reduces computation (GFLOPs) by up to 2.5x and memory footprint (VRAM) by up to 2.1x, while generally registering performance gains of around 1%. On TotalSegmentator, we achieve a Dice score of 93.51% with only 295MB peak GPU memory. Zero-shot cross-dataset evaluations on SegTHOR and AMOS22 demonstrate strong generalization, with Dice scores of up to 86.85% and 89.35%, respectively. We release our open-source code at https://github.com/andreibunea99/SegMate.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑