中文

COBRA:仅 CPU 的腹部器官分割

图像与视频处理 2022-07-22 v1 计算机视觉与模式识别

摘要

腹部器官分割是一项困难且耗时的任务。为减轻临床专家的负担,全自动化方法极为必要。当前方法以卷积神经网络(CNN)为主,但其计算需求及大数据集依赖限制了实际中的应用。通过实现一种小型高效的定制 3D CNN、编译训练好的模型并优化计算图,我们的方法以每幅图像 1.6 秒的速率生成高精度分割结果(Dice 相似系数(%):肝脏 97.3±\pm1.3,肾脏 94.8±\pm3.6,脾脏 96.4±\pm3.0,胰腺 80.9±\pm10.1)。关键在于,我们能够在仅 CPU 上执行分割推理(无需 GPU),从而便于在无专用硬件的情况下轻松广泛部署该模型。

关键词

引用

@article{arxiv.2207.10446,
  title  = {COBRA: Cpu-Only aBdominal oRgan segmentAtion},
  author = {Edward G. A. Henderson and Dónal M. McSweeney and Andrew F. Green},
  journal= {arXiv preprint arXiv:2207.10446},
  year   = {2022}
}

备注

MCR-RRR submission for the Fast and Low GPU memory Abdominal oRgan sEgmentation Challenge (FLARE) at MICCAI 2021