arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.03750eess.IVcs.CV
收藏

广义细胞核检测的开关损耗

Switching Loss for Generalized Nucleus Detection in Histopathology

Indian Institute of Technology Bombay(印度理工学院孟买分校);National Institute of Technology Raipur(印度国立技术学院赖布尔分校);University of Illinois at Chicago(伊利诺伊大学芝加哥分校)

Deepak Anand, Gaurav Patel, Yaman Dang, Amit Sethi

摘要:

深度学习方法在医学图像分析中的两个基本任务(检测和分割)的准确性可能会受到类别不平衡的影响。我们提出了一个“切换损失”的功能,自适应地转移前景和背景类之间的重点。虽然现有的损失函数来解决这个问题是由分类任务的动机,开关损耗是基于骰子损失,这是更好地适合分割和检测。此外,为了最大限度地利用训练样本,我们对每个小批量进行了损失调整,而不是像以前的建议那样对整个训练集进行一次调整。在源数据集上使用所提出的损失函数训练的核检测器优于使用交叉熵、骰子或焦点损失训练的核检测器。值得注意的是,在没有对目标数据集进行再训练的情况下,我们预先训练的核检测器的性能也优于现有的核检测器,这些核检测器至少在目标数据集的一些图像上进行了训练。为了建立所提出的损失的广泛实用性,我们还证实,与其他损失函数相比,它在MRI中导致更准确的心室分割。我们的GPU支持的预训练细胞核检测软件也可以立即处理整个载玻片图像,并且速度非常快。

英文摘要:

The accuracy of deep learning methods for two foundational tasks in medical image analysis -- detection and segmentation -- can suffer from class imbalance. We propose a `switching loss' function that adaptively shifts the emphasis between foreground and background classes. While the existing loss functions to address this problem were motivated by the classification task, the switching loss is based on Dice loss, which is better suited for segmentation and detection. Furthermore, to get the most out the training samples, we adapt the loss with each mini-batch, unlike previous proposals that adapt once for the entire training set. A nucleus detector trained using the proposed loss function on a source dataset outperformed those trained using cross-entropy, Dice, or focal losses. Remarkably, without retraining on target datasets, our pre-trained nucleus detector also outperformed existing nucleus detectors that were trained on at least some of the images from the target datasets. To establish a broad utility of the proposed loss, we also confirmed that it led to more accurate ventricle segmentation in MRI as compared to the other loss functions. Our GPU-enabled pre-trained nucleus detection software is also ready to process whole slide images right out-of-the-box and is usably fast.

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

↑