arXiv:2504.05207cs.CVcs.AI2025-04被引 2

用自训练缓解数据不平衡,提升CT中病灶检测与标记的全面性。

Correcting Class Imbalances with Self-Training for Improved Universal Lesion Detection and Tagging

  • 基于少量标注数据训练基础模型,通过多轮自训练引入新病灶样本。
  • 结合变阈值策略与上采样,使稀有病灶敏感度提升6.5%至78.5%。
  • 适用于医学影像中类别不均衡场景,尤其适合病灶检测任务研究者。

CT中通用病灶检测与标记(ULDT)对肿瘤负荷评估和病灶动态变化追踪至关重要。然而,完全标注数据的缺乏限制了有效方法的发展。先前工作使用DeepLesion数据集(4,427名患者,10,594项检查,32,120张切片,32,735个病灶,8个体部标签)进行算法开发,但该数据集未完全标注且存在类别不平衡。为此,本文构建了一个用于ULDT的自训练流程:在11.5%的DeepLesion子集(含边界框与标签)上训练VFNet模型,随后从更大未见数据子集中识别并纳入新病灶候选样本,进行多轮自训练。通过不同阈值策略筛选高质量预测结果,以覆盖类别不平衡问题。实验发现,直接自训练会提高高频类别的敏感度,但牺牲低频类别。而结合自训练样本上采样与可变阈值策略,相较无类别平衡的自训练(72%→78.5%),敏感度提升6.5%;相比无上采样的相同策略(66.8%→78.5%),提升11.7%。此外,所有8类病灶的4FP敏感度均得到保持或提升。

原文摘要 · Abstract (English)

Universal lesion detection and tagging (ULDT) in CT studies is critical for tumor burden assessment and tracking the progression of lesion status (growth/shrinkage) over time. However, a lack of fully annotated data hinders the development of effective ULDT approaches. Prior work used the DeepLesion dataset (4,427 patients, 10,594 studies, 32,120 CT slices, 32,735 lesions, 8 body part labels) for algorithmic development, but this dataset is not completely annotated and contains class imbalances. To address these issues, in this work, we developed a self-training pipeline for ULDT. A VFNet model was trained on a limited 11.5\% subset of DeepLesion (bounding boxes + tags) to detect and classify lesions in CT studies. Then, it identified and incorporated novel lesion candidates from a larger unseen data subset into its training set, and self-trained itself over multiple rounds. Multiple self-training experiments were conducted with different threshold policies to select predicted lesions with higher quality and cover the class imbalances. We discovered that direct self-training improved the sensitivities of over-represented lesion classes at the expense of under-represented classes. However, upsampling the lesions mined during self-training along with a variable threshold policy yielded a 6.5\% increase in sensitivity at 4 FP in contrast to self-training without class balancing (72\% vs 78.5\%) and a 11.7\% increase compared to the same self-training policy without upsampling (66.8\% vs 78.5\%). Furthermore, we show that our results either improved or maintained the sensitivity at 4FP for all 8 lesion classes.

病灶检测自训练数据不平衡医学影像

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