论文标题

轻巧的编码器 - 脚溃疡细分架构

Lightweight Encoder-Decoder Architecture for Foot Ulcer Segmentation

论文作者

Ali, Shahzad, Mahmood, Arif, Jung, Soon Ki

论文摘要

需要连续监测足部溃疡愈合,以确保给定治疗的功效并避免任何恶化。足球溃疡分割是伤口诊断的重要一步。我们开发了一种模型,其精神与建立的编码器编码器和残留卷积神经网络相似。我们的模型包括剩余的连接以及在每个卷积块中集成的通道和空间注意力。用于模型培训,测试时间增加以及对所获得的预测的多数投票的简单基于补丁的方法导致了卓越的性能。我们的模型没有利用任何容易获得的骨干体系结构,在类似的外部数据集或任何转移学习技术上进行预训练。与用于足球溃疡细分任务的可用最新模型相比,网络参数的总数约为500万个模型,使其成为一个显着的轻巧模型。我们的实验以斑块级和图像级呈现结果。我们的模型适用于Miccai 2021的公开脚步溃疡细分(Fuseg)挑战数据集,就骰子相似性得分而言,最先进的图像级绩效为88.22%,在官方挑战排行榜中排名第二。我们还展示了一个非常简单的解决方案,可以将其与更先进的体系结构进行比较。

Continuous monitoring of foot ulcer healing is needed to ensure the efficacy of a given treatment and to avoid any possibility of deterioration. Foot ulcer segmentation is an essential step in wound diagnosis. We developed a model that is similar in spirit to the well-established encoder-decoder and residual convolution neural networks. Our model includes a residual connection along with a channel and spatial attention integrated within each convolution block. A simple patch-based approach for model training, test time augmentations, and majority voting on the obtained predictions resulted in superior performance. Our model did not leverage any readily available backbone architecture, pre-training on a similar external dataset, or any of the transfer learning techniques. The total number of network parameters being around 5 million made it a significantly lightweight model as compared with the available state-of-the-art models used for the foot ulcer segmentation task. Our experiments presented results at the patch-level and image-level. Applied on publicly available Foot Ulcer Segmentation (FUSeg) Challenge dataset from MICCAI 2021, our model achieved state-of-the-art image-level performance of 88.22% in terms of Dice similarity score and ranked second in the official challenge leaderboard. We also showed an extremely simple solution that could be compared against the more advanced architectures.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源