arXiv:2412.18589eess.IVcs.CV2024-12被引 17

用文字控制生成肿瘤图像,提升AI在难病例上的表现

Text-Driven Tumor Synthesis

  • 通过文本描述控制肿瘤纹理、边界等特征,实现精准生成
  • 在早期检测中提升敏感性8.5%,分割任务提升DSC 6.3%
  • 仅用141对图像报告数据,借助海量报告实现高效合成

肿瘤合成可生成人工智能常误检或漏检的案例,从而提升模型性能。然而现有方法多为无条件生成或仅基于肿瘤形状,难以控制纹理、异质性、边界和病理类型等具体特征,导致生成结果相似或重复,无法有效解决AI弱点。本文提出文本驱动的肿瘤合成方法TextoMorph,通过引入放射科报告中的文本信息,实现对肿瘤特征的精细控制。该方法特别适用于人工智能最易出错的场景:早期肿瘤检测(敏感性提升+8.5%)、精确放疗分割(DSC提升+6.3%)以及良恶性分类(敏感性提升+8.2%)。TextoMorph结合跨文本与CT扫描的对比学习,仅需141对图像-报告配对即可训练,同时利用34,035份放射科报告增强泛化能力。我们设计了严格的评估方法,包括文本驱动视觉图灵测试和影像组学模式分析,验证了合成肿瘤在纹理、异质性、边界和病理特征上的真实性和多样性。

原文摘要 · Abstract (English)

Tumor synthesis can generate examples that AI often misses or over-detects, improving AI performance by training on these challenging cases. However, existing synthesis methods, which are typically unconditional -- generating images from random variables -- or conditioned only by tumor shapes, lack controllability over specific tumor characteristics such as texture, heterogeneity, boundaries, and pathology type. As a result, the generated tumors may be overly similar or duplicates of existing training data, failing to effectively address AI's weaknesses. We propose a new text-driven tumor synthesis approach, termed TextoMorph, that provides textual control over tumor characteristics. This is particularly beneficial for examples that confuse the AI the most, such as early tumor detection (increasing Sensitivity by +8.5%), tumor segmentation for precise radiotherapy (increasing DSC by +6.3%), and classification between benign and malignant tumors (improving Sensitivity by +8.2%). By incorporating text mined from radiology reports into the synthesis process, we increase the variability and controllability of the synthetic tumors to target AI's failure cases more precisely. Moreover, TextoMorph uses contrastive learning across different texts and CT scans, significantly reducing dependence on scarce image-report pairs (only 141 pairs used in this study) by leveraging a large corpus of 34,035 radiology reports. Finally, we have developed rigorous tests to evaluate synthetic tumors, including Text-Driven Visual Turing Test and Radiomics Pattern Analysis, showing that our synthetic tumors is realistic and diverse in texture, heterogeneity, boundaries, and pathology.

肿瘤生成文本控制医学AI扩散模型

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