提出新框架CurvNet,用隐空间轮廓表示与迭代数据生成提升脊柱弯曲角测量精度。
CurvNet: Latent Contour Representation and Iterative Data Engine for Curvature Angle Estimation
- 在隐空间建模脊柱轮廓,实现轮廓重建与分解,解决分割不连通问题。
- 通过自生成图像与自动标注构建大规模清洁数据集Spinal-AI2024,含超3万张影像。
- 在多个数据集上达到最优性能,适合医学影像分析与智能筛查应用。
弯曲角是曲线的定量度量,其中柯布角专用于脊柱弯曲评估。基于X光片的自动柯布角测量对脊柱侧弯筛查与诊断至关重要。然而,现有回归与分割方法常面临脊柱表征不准确或掩码连接性与碎片化问题;基于关键点的方法则受限于训练数据与标注不足。为此,本文提出新型弯曲角估计框架CurvNet,包含基于隐空间轮廓表示的轮廓检测与基于迭代数据引擎的图像自生成。具体而言,我们在隐空间提出参数化脊柱轮廓表示,实现特征脊柱分解与轮廓重建;结合隐空间系数回归与锚框分类,解决预测不准与掩码连接问题。此外,我们构建了包含图像自生成、自动标注与自动筛选的迭代数据引擎。通过该引擎,我们生成了名为Spinal-AI2024的无隐私泄露清洁数据集,据我们所知为目前最大公开脊柱侧弯X光数据集。在AASCE2019、私有Spinal2023及生成的Spinal-AI2024数据集上的大量实验表明,本方法在柯布角估计任务上达到当前最佳表现。代码与数据集分别开源于https://github.com/Ernestchenchen/CurvNet 和 https://github.com/Ernestchenchen/Spinal-AI2024。
原文摘要 · Abstract (English)
Curvature angle is a quantitative measurement of a curve, in which Cobb angle is customized for spinal curvature. Automatic Cobb angle measurement from X-ray images is crucial for scoliosis screening and diagnosis. However, most existing regression-based and segmentation-based methods struggle with inaccurate spine representations or mask connectivity and fragmentation issues. Besides, landmark-based methods suffer from insufficient training data and annotations. To address these challenges, we propose a novel curvature angle estimation framework named CurvNet including latent contour representation based contour detection and iterative data engine based image self-generation. Specifically, we propose a parameterized spine contour representation in latent space, which enables eigen-spine decomposition and spine contour reconstruction. Latent contour coefficient regression is combined with anchor box classification to solve inaccurate predictions and mask connectivity issues. Moreover, we develop a data engine with image self-generation, automatic annotation, and automatic selection in an iterative manner. By our data engine, we generate a clean dataset named Spinal-AI2024 without privacy leaks, which is the largest released scoliosis X-ray dataset to our knowledge. Extensive experiments on public AASCE2019, our private Spinal2023, and our generated Spinal-AI2024 datasets demonstrate that our method achieves state-of-the-art Cobb angle estimation performance. Our code and Spinal-AI2024 dataset are available at https://github.com/Ernestchenchen/CurvNet and https://github.com/Ernestchenchen/Spinal-AI2024, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。