用旋转对称卷积改进血管分割,更准且更轻量。
Improved Vessel Segmentation with Symmetric Rotation-Equivariant U-Net
- 在U-Net中引入对称旋转等变卷积,自动适应图像方向变化。
- 在视网膜血管图像上达到更高分割精度,优于标准U-Net和现有等变方法。
- 模型参数和内存占用大幅降低,适合医疗设备部署。
自动化分割在医学图像分析与计算机辅助干预中至关重要。尽管基于卷积神经网络(CNN)的方法表现优异,但它们忽略了图像中重要的等变性质,如旋转和反射等变性,这会导致性能下降和预测不一致,尤其在无明确方向的血管分割任务中更为明显。现有等变学习方法虽能缓解此问题,但显著增加训练成本或模型规模。为此,我们提出一种高效的对称旋转等变(SRE)卷积核(SRE-Conv),应用于U-Net架构,以学习旋转与反射等变特征,同时大幅减少模型尺寸。我们在视网膜血管眼底成像数据集上验证了该方法的有效性。结果表明,所提出的SRE U-Net不仅在处理旋转图像时显著优于标准U-Net,还超越了现有等变学习方法,且具有更少的可训练参数和更低的内存开销。代码已开源:https://github.com/OnofreyLab/sre_conv_segm_isbi2025。
原文摘要 · Abstract (English)
Automated segmentation plays a pivotal role in medical image analysis and computer-assisted interventions. Despite the promising performance of existing methods based on convolutional neural networks (CNNs), they neglect useful equivariant properties for images, such as rotational and reflection equivariance. This limitation can decrease performance and lead to inconsistent predictions, especially in applications like vessel segmentation where explicit orientation is absent. While existing equivariant learning approaches attempt to mitigate these issues, they substantially increase learning cost, model size, or both. To overcome these challenges, we propose a novel application of an efficient symmetric rotation-equivariant (SRE) convolutional (SRE-Conv) kernel implementation to the U-Net architecture, to learn rotation and reflection-equivariant features, while also reducing the model size dramatically. We validate the effectiveness of our method through improved segmentation performance on retina vessel fundus imaging. Our proposed SRE U-Net not only significantly surpasses standard U-Net in handling rotated images, but also outperforms existing equivariant learning methods and does so with a reduced number of trainable parameters and smaller memory cost. The code is available at https://github.com/OnofreyLab/sre_conv_segm_isbi2025.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。