arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.00528cs.CVcs.CL
收藏

PET/CT中用于阳性发现视觉定位的视觉-语言建模

Vision-Language Modeling in PET/CT for Visual Grounding of Positive Findings

University of Wisconsin-Madison(威斯康星大学麦迪逊分校);University of Wisconsin Health(威斯康星大学医疗中心);Carbone Cancer Center(卡博癌症中心)

Zachary Huemann, Samuel Church, Joshua D. Warner, Daniel Tran, Xin Tie, Alan B McMillan, Junjie Hu, Steve Y. Cho, Meghan Lubner, Tyler J. Bradshaw

AI总结:

本研究开发了一种自动化弱标签流程生成PET/CT图像-文本对数据集,并训练了3D视觉定位模型ConTEXTual Net 3D,其性能显著优于其他模型但不及核医学医师,表明需要更大数据集以缩小差距。

摘要:

视觉-语言模型可以通过视觉定位将物体的文本描述与其在图像中的具体位置联系起来。这在增强放射学报告中具有潜在应用。然而,这些模型需要大型标注图像-文本数据集,而PET/CT缺乏此类数据。我们开发了一个自动化流程,用于生成将PET/CT报告描述与其图像位置关联的弱标签,并利用它训练了一个3D视觉-语言视觉定位模型。我们的流程通过识别SUVmax和轴向切片编号的提及,在PET/CT报告中寻找阳性发现。从25,578例PET/CT检查中,我们提取了11,356个句子-标签对。使用这些数据,我们训练了ConTEXTual Net 3D,它通过token级别的交叉注意力将来自大型语言模型的文本嵌入与3D nnU-Net集成在一起。该模型的性能与LLMSeg、ConTEXTual Net的2.5D版本以及两名核医学医师进行了比较。弱标签流程在98%的病例(246/251)中准确识别了病灶位置,其中7.5%需要边界调整。ConTEXTual Net 3D的F1分数为0.80,优于LLMSeg(F1=0.22)和2.5D模型(F1=0.53),尽管其表现不及两名医师(F1分别为0.94和0.91)。该模型在FDG(F1=0.78)和DCFPyL(F1=0.75)检查中表现更好,而在DOTATE(F1=0.58)和Fluciclovine(F1=0.66)上表现下降。该模型在不同病灶大小上表现一致,但对低摄取病灶的准确性降低。我们新颖的弱标签流程准确生成了PET/CT图像-文本对的标注数据集,促进了3D视觉定位模型的发展。ConTEXTual Net 3D显著优于其他模型,但未达到核医学医师的表现水平。我们的研究表明,可能需要更大的数据集来缩小这一性能差距。

英文摘要:

Vision-language models can connect the text description of an object to its specific location in an image through visual grounding. This has potential applications in enhanced radiology reporting. However, these models require large annotated image-text datasets, which are lacking for PET/CT. We developed an automated pipeline to generate weak labels linking PET/CT report descriptions to their image locations and used it to train a 3D vision-language visual grounding model. Our pipeline finds positive findings in PET/CT reports by identifying mentions of SUVmax and axial slice numbers. From 25,578 PET/CT exams, we extracted 11,356 sentence-label pairs. Using this data, we trained ConTEXTual Net 3D, which integrates text embeddings from a large language model with a 3D nnU-Net via token-level cross-attention. The model's performance was compared against LLMSeg, a 2.5D version of ConTEXTual Net, and two nuclear medicine physicians. The weak-labeling pipeline accurately identified lesion locations in 98% of cases (246/251), with 7.5% requiring boundary adjustments. ConTEXTual Net 3D achieved an F1 score of 0.80, outperforming LLMSeg (F1=0.22) and the 2.5D model (F1=0.53), though it underperformed both physicians (F1=0.94 and 0.91). The model achieved better performance on FDG (F1=0.78) and DCFPyL (F1=0.75) exams, while performance dropped on DOTATE (F1=0.58) and Fluciclovine (F1=0.66). The model performed consistently across lesion sizes but showed reduced accuracy on lesions with low uptake. Our novel weak labeling pipeline accurately produced an annotated dataset of PET/CT image-text pairs, facilitating the development of 3D visual grounding models. ConTEXTual Net 3D significantly outperformed other models but fell short of the performance of nuclear medicine physicians. Our study suggests that even larger datasets may be needed to close this performance gap.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑