Vision Transformers (ViTs) have revolutionized medical imaging analysis, showcasing superior efficacy compared to conventional Convolutional Neural Networks (CNNs) in vital tasks such as polyp classification, detection, and segmentation. Leveraging attention mechanisms to focus on specific image regions, ViTs exhibit contextual awareness in processing visual data, culminating in robust and precise predictions, even for intricate medical images. Moreover, the inherent self-attention mechanism in Transformers accommodates varying input sizes and resolutions, granting an unprecedented flexibility absent in traditional CNNs. However, Transformers grapple with challenges like excessive memory usage and limited training parallelism due to self-attention, rendering them impractical for real-time disease detection on resource-constrained devices. In this study, we address these hurdles by investigating the integration of the recently introduced retention mechanism into polyp segmentation, introducing RetSeg, an encoder-decoder network featuring multi-head retention blocks. Drawing inspiration from Retentive Networks (RetNet), RetSeg is designed to bridge the gap between precise polyp segmentation and resource utilization, particularly tailored for colonoscopy images. We train and validate RetSeg for polyp segmentation employing two publicly available datasets: Kvasir-SEG and CVC-ClinicDB. Additionally, we showcase RetSeg's promising performance across diverse public datasets, including CVC-ColonDB, ETIS-LaribPolypDB, CVC-300, and BKAI-IGH NeoPolyp. While our work represents an early-stage exploration, further in-depth studies are imperative to advance these promising findings.
翻译:视觉Transformer在医学图像分析领域引发了革命性变革,在息肉分类、检测和分割等关键任务中展现出优于传统卷积神经网络的卓越性能。通过利用注意力机制聚焦特定图像区域,ViT在处理视觉数据时具备上下文感知能力,即使在复杂医学图像上也能生成稳健精确的预测。此外,Transformer固有的自注意力机制能够适应不同尺寸和分辨率的输入,提供了传统CNN所不具备的灵活性。然而,由于自注意力机制导致内存占用过高和训练并行性受限,Transformer难以在资源受限设备上实现实时疾病检测。本研究通过探索将最新提出的保持机制整合到息肉分割中来解决这些问题,提出了RetSeg——一个包含多头保持块的编码器-解码器网络。受保持网络启发,RetSeg旨在弥合精确息肉分割与资源利用之间的鸿沟,特别针对结肠镜图像进行定制。我们使用两个公开数据集Kvasir-SEG和CVC-ClinicDB对RetSeg进行息肉分割训练和验证。此外,我们展示了RetSeg在多个不同公开数据集(包括CVC-ColonDB、ETIS-LaribPolypDB、CVC-300和BKAI-IGH NeoPolyp)上的优异性能。尽管本研究仍处于早期探索阶段,但需要进一步深入的研究来推进这些有前景的发现。