论文标题

plutonet:具有改进的部分解码器和解码器一致性训练的有效息肉分割网络

PlutoNet: An Efficient Polyp Segmentation Network with Modified Partial Decoder and Decoder Consistency Training

论文作者

Erol, Tugberk, Sarikaya, Duygu

论文摘要

深度学习模型用于最大程度地减少专家没有注意到的息肉的数量,并在干预过程中准确细分检测到的息肉。尽管提出了最先进的模型,但定义能够良好概括并在捕获低级功能和高级语义细节之间进行调解的表示表示仍然是一个挑战。这些模型的另一个挑战是它们需要太多参数,这可能会在实时应用程序中构成问题。为了解决这些问题,我们提出了用于息肉分割的plutonet,它仅需要2,626,537个参数,少于其对应物所需的参数的10 \%。使用Plutonet,我们提出了一种小说\ emph {解码器一致性培训}方法,该方法由共享编码器组成,该编码器是经过修改的部分解码器,它是部分解码器和全尺度连接的组合,可在不同尺度上捕获显着特征,而无需冗余,而无需冗余,并且辅助解码器侧重于更高的相关技巧。我们训练经过修改的部分解码器和辅助解码器,并具有结合损失以实施一致性,这有助于改善编码器表示。这样,我们能够降低不确定性和假阳性率。我们进行消融研究和广泛的实验,这些实验表明plutonet的性能明显优于最先进的模型,尤其是在不同领域的看不见的数据集和数据集上。

Deep learning models are used to minimize the number of polyps that goes unnoticed by the experts and to accurately segment the detected polyps during interventions. Although state-of-the-art models are proposed, it remains a challenge to define representations that are able to generalize well and that mediate between capturing low-level features and higher-level semantic details without being redundant. Another challenge with these models is that they require too many parameters, which can pose a problem with real-time applications. To address these problems, we propose PlutoNet for polyp segmentation which requires only 2,626,537 parameters, less than 10\% of the parameters required by its counterparts. With PlutoNet, we propose a novel \emph{decoder consistency training} approach that consists of a shared encoder, the modified partial decoder which is a combination of the partial decoder and full-scale connections that capture salient features at different scales without being redundant, and the auxiliary decoder which focuses on higher-level relevant semantic features. We train the modified partial decoder and the auxiliary decoder with a combined loss to enforce consistency, which helps improve the encoders representations. This way we are able to reduce uncertainty and false positive rates. We perform ablation studies and extensive experiments which show that PlutoNet performs significantly better than the state-of-the-art models, particularly on unseen datasets and datasets across different domains.

扫码加入交流群

加入微信交流群

微信交流群二维码

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