论文标题
可以:最先进的视网膜船分割的小W-NET,具有简约的模型
The Little W-Net That Could: State-of-the-Art Retinal Vessel Segmentation with Minimalistic Models
论文作者
论文摘要
视网膜图像对视网膜脉管系统的分割是视网膜图像分析中最基本的任务之一。近年来,基于复杂的卷积神经网络体系结构越来越复杂的方法一直在缓慢地推动建立良好的基准数据集的性能。在本文中,我们退后一步,分析了这种复杂性的实际需求。具体而言,我们证明了标准U-NET的简约版本,其参数少几个数量级,经过精心训练和严格评估,非常接近当前最佳技术的性能。此外,我们提出了一个名为W-NET的简单扩展名,该扩展名在几个流行的数据集上达到了出色的性能,但仍比任何以前发表的方法都使用的数量级较小的订单。此外,我们提供迄今为止最全面的跨数据库性能分析,最多涉及10个不同的数据库。我们的分析表明,当考虑与训练数据有很大不同的测试图像时,视网膜血管分割问题远非解决,并且此任务代表了探索域适应技术的理想场景。在这种情况下,我们实验了一种简单的自我标记策略,使我们能够适度提高跨数据库的性能,这表明该领域仍然有很大的改进空间。最后,我们还测试了有关动脉/静脉分割问题的方法,在最近的文献中,我们再次在模型复杂性的一小部分中再次实现了与最先进的结果。在本文中重现结果的所有代码均已发布。
The segmentation of the retinal vasculature from eye fundus images represents one of the most fundamental tasks in retinal image analysis. Over recent years, increasingly complex approaches based on sophisticated Convolutional Neural Network architectures have been slowly pushing performance on well-established benchmark datasets. In this paper, we take a step back and analyze the real need of such complexity. Specifically, we demonstrate that a minimalistic version of a standard U-Net with several orders of magnitude less parameters, carefully trained and rigorously evaluated, closely approximates the performance of current best techniques. In addition, we propose a simple extension, dubbed W-Net, which reaches outstanding performance on several popular datasets, still using orders of magnitude less learnable weights than any previously published approach. Furthermore, we provide the most comprehensive cross-dataset performance analysis to date, involving up to 10 different databases. Our analysis demonstrates that the retinal vessel segmentation problem is far from solved when considering test images that differ substantially from the training data, and that this task represents an ideal scenario for the exploration of domain adaptation techniques. In this context, we experiment with a simple self-labeling strategy that allows us to moderately enhance cross-dataset performance, indicating that there is still much room for improvement in this area. Finally, we also test our approach on the Artery/Vein segmentation problem, where we again achieve results well-aligned with the state-of-the-art, at a fraction of the model complexity in recent literature. All the code to reproduce the results in this paper is released.