LLM2D
面向医学图像中任意分割的无真值评估方法
Towards Ground-truth-free Evaluation of Any Segmentation in Medical Images
发布日期: 9/24/2024
arXiv ID: oai:arXiv.org:2409.14874v1

摘要

我们致力于构建一个无需真实标签的评估模型,用于评估SAM(分割一切模型)及其变体在医学图像中生成的分割质量。该模型通过比较输入图像与其对应的分割图来估计分割质量分数。基于先前的研究,我们将此问题框架化为监督学习框架内的回归问题,使用Dice分数(以及其他可选指标)来计算训练损失。该模型利用大量公开的医学图像数据集进行训练,这些数据集包含了SAM及其变体的分割预测。我们将此模型命名为EvanySeg(医学图像中任意分割的评估)。我们对基于卷积的模型(如ResNet)和基于Transformer的模型(如ViT)进行了探索,结果表明ViT在EvanySeg中表现更优。该模型可应用于多种任务,包括:(1)通过检测低百分位分割质量分数来识别分割质量较差的样本;(2)在没有真实标签的情况下,通过测试样本的平均分数来基准化分割模型;(3)在人机协作过程中,通过在分数空间内应用阈值来提醒人类专家;(4)在测试时,当存在多个分割模型时,通过选择得分最高的预测来为每个测试样本选择最佳分割预测。模型和代码将在https://github.com/ahjolsenbics/EvanySeg上提供。