轻量级混合模型精准分割视盘与视杯,适合临床快速筛查
OptiModNet: A UNet-Transformer Hybrid with Grouped-Query and Channel Attention for Optic Disc and Cup Segmentation

- 融合分组查询与通道注意力的混合架构,兼顾局部与全局特征
- 在REFUGE2数据集上性能超现有方法2.5%以上,仅需3.73 GFLOPs
- 低计算开销,适合资源受限的医疗场景部署
精确分割视盘与视杯对青光眼的早期检测和诊断至关重要。然而,在跨数据集保持高性能的同时维持低计算成本仍具挑战。在青光眼检测中,低计算量方法对于实现快速、大规模筛查及在资源有限的临床环境中部署尤为关键。尽管深度学习模型如UNet、视觉变压器(ViTs)和扩散模型表现出色,但通常伴随显著计算开销。UNet擅长捕捉局部特征,但难以建模全局上下文;而ViT虽能有效处理长距离依赖,却计算密集。混合架构如UNetR结合了基于Transformer的编码器与UNet风格解码器,性能提升但复杂度增加。为此,本文提出OptiModNet,一种专为视盘与视杯分割设计的轻量级混合架构。模型在多阶段集成多种注意力机制,增强局部与全局特征表示。引入聚合金字塔损失,在解码器多个层级监督预测,促进梯度传播与结构一致性。在REFUGE2数据集上评估,该方法在视盘与视杯分割任务中达到当前最优性能,超越现有方法超过2.5%,同时保持高效率,仅需3.73 GFLOPs和1.93M参数。代码已开源。
原文摘要 · Abstract (English)
Precise segmentation of the optic disc and cup is critical for the early detection and diagnosis of glaucoma. However, achieving consistently high performance across datasets while maintaining low computational requirements remains a significant challenge. In glaucoma detection, low-computation methods are crucial for enabling rapid, large-scale screening and facilitating deployment in resource-limited clinical environments. While deep learning models such as UNets, Vision Transformers (ViTs), and Diffusion models have demonstrated strong segmentation performance but these methods often come with substantial computational overhead. UNets are efficient at capturing local features but are limited in modeling global contextual information. Conversely, ViTs excel at long-range dependency modeling but are computationally intensive. Hybrid architectures, such as UNetR, which combine transformer-based encoders with UNet-style decoders, have shown improved performance but while incurring additional complexity. Considering these, in this work, we propose OptiModNet, a light weight novel hybrid architecture tailored for optic disc and cup segmentation. The model integrates diverse attention mechanisms at multiple stages of the network to enhance both local and global feature representation. We include an Aggregated Pyramid Loss that supervises predictions at multiple decoder depths, to promote better gradient flow and structural consistency. We evaluate OptiModNet on the REFUGE2 dataset for both optic disc and cup segmentation tasks. Our method achieves state-of-the-art performance, exceeding existing approaches by over 2.5\%, while maintaining high efficiency with only 3.73 GFLOPs and 1.93M parameters. The code is available at https://github.com/SG1947/OptiModNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。