arXiv:2409.04952cs.CV2024-09中稿 · Medical Image Anal…被引 8

用贝叶斯主动学习选配对,降低溃疡性结肠炎严重程度评估的标注成本

Deep Bayesian Active Learning-to-Rank with Relative Annotation for Estimation of Ulcerative Colitis Severity

  • 基于模型不确定性自动选择最有效的图像配对进行相对标注
  • 在真实数据集上仅用少量标注即达到高精度,且对少数类样本表现优异
  • 适合医疗图像分析中标注成本高、类别不平衡的场景

基于图像的严重程度自动评估在计算机辅助诊断中至关重要。深度学习方法需要大量训练数据才能取得高性能。传统方法使用离散(量化)严重程度标签进行标注,但在严重程度模糊的图像上标注困难且成本高。相比之下,相对标注通过比较成对图像的严重程度,避免了量化问题,更易操作。然而,相对标注面临可标注配对数量庞大的挑战,因此配对选择至关重要。本文提出一种深度贝叶斯主动学习排序方法,可自动选择最具学习效率的未标注配对。该方法基于模型不确定性优先标注信息量高的样本。我们证明了将贝叶斯神经网络应用于成对学习排序的理论基础,并在私有和公开的溃疡性结肠炎内镜图像数据集上验证了方法的有效性。实验表明,该方法在显著类别不平衡条件下仍能保持高性能,因其能自动从少数类中选择样本。

原文摘要 · Abstract (English)

Automatic image-based severity estimation is an important task in computer-aided diagnosis. Severity estimation by deep learning requires a large amount of training data to achieve a high performance. In general, severity estimation uses training data annotated with discrete (i.e., quantized) severity labels. Annotating discrete labels is often difficult in images with ambiguous severity, and the annotation cost is high. In contrast, relative annotation, in which the severity between a pair of images is compared, can avoid quantizing severity and thus makes it easier. We can estimate relative disease severity using a learning-to-rank framework with relative annotations, but relative annotation has the problem of the enormous number of pairs that can be annotated. Therefore, the selection of appropriate pairs is essential for relative annotation. In this paper, we propose a deep Bayesian active learning-to-rank that automatically selects appropriate pairs for relative annotation. Our method preferentially annotates unlabeled pairs with high learning efficiency from the model uncertainty of the samples. We prove the theoretical basis for adapting Bayesian neural networks to pairwise learning-to-rank and demonstrate the efficiency of our method through experiments on endoscopic images of ulcerative colitis on both private and public datasets. We also show that our method achieves a high performance under conditions of significant class imbalance because it automatically selects samples from the minority classes.

医疗影像主动学习相对标注贝叶斯网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。