arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2208.01485eess.IVcs.CVcs.LG

IterMiUnet:一种用于自动血管分割的轻量级架构

IterMiUnet: A lightweight architecture for automatic blood vessel segmentation

Indian Institute of Technology Roorkee(鲁尔基印度理工学院);Jawaharlal Nehru University(贾瓦哈拉尔·尼赫鲁大学);All India Institute of Medical Sciences(全印度医学科学学院)

Ashish Kumar, R. K. Agrawal, Leve Joseph

AI总结:

针对视网膜血管分割模型参数量大、难落地的问题,本文融合 Iternet 与 MiUnet 提出轻量级卷积模型 IterMiUnet,在保持深度与性能的同时减少参数、加快训练推理,并在 DRIVE、STARE、CHASE-DB1 上验证有效。

摘要:

眼底图像中血管的自动分割有助于分析视网膜血管状况,这对于识别高血压、糖尿病等多种全身性疾病至关重要。尽管基于深度学习的模型在这一分割任务上取得了成功,但其中大多数模型参数量庞大,因此在实际应用中的使用受到限制。本文提出了 IterMiUnet,这是一种新型的基于卷积的轻量级分割模型,所需参数显著更少,却能提供与现有模型相近的性能。该模型利用了 Iternet 架构出色的分割能力,同时通过在其中融入 MiUnet 模型的编码器-解码器结构,克服了其参数量庞大的缺陷。因此,新模型在不牺牲网络深度的情况下减少了参数,而网络深度对于深度模型学习抽象的层次化概念是必要的。这种轻量级分割模型加快了训练和推理时间,并可能对数据稀缺、因而参数量庞大的模型容易过拟合的医疗领域有所帮助。所提出模型在三个公开可用数据集上进行了评估:DRIVE、STARE 和 CHASE-DB1。此外,还进行了交叉训练和评定者间差异评估。所提出模型具有作为多种疾病早期诊断工具的巨大潜力。

英文摘要:

The automatic segmentation of blood vessels in fundus images can help analyze the condition of retinal vasculature, which is crucial for identifying various systemic diseases like hypertension, diabetes, etc. Despite the success of Deep Learning-based models in this segmentation task, most of them are heavily parametrized and thus have limited use in practical applications. This paper proposes IterMiUnet, a new lightweight convolution-based segmentation model that requires significantly fewer parameters and yet delivers performance similar to existing models. The model makes use of the excellent segmentation capabilities of Iternet architecture but overcomes its heavily parametrized nature by incorporating the encoder-decoder structure of MiUnet model within it. Thus, the new model reduces parameters without any compromise with the network's depth, which is necessary to learn abstract hierarchical concepts in deep models. This lightweight segmentation model speeds up training and inference time and is potentially helpful in the medical domain where data is scarce and, therefore, heavily parametrized models tend to overfit. The proposed model was evaluated on three publicly available datasets: DRIVE, STARE, and CHASE-DB1. Further cross-training and inter-rater variability evaluations have also been performed. The proposed model has a lot of potential to be utilized as a tool for the early diagnosis of many diseases.

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

↑