arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.15744cs.CVcs.AI
收藏

通过掩码-属性对齐释放视觉-语言预训练在 3D 零样本病灶分割中的潜力

Unleashing the Potential of Vision-Language Pre-Training for 3D Zero-Shot Lesion Segmentation via Mask-Attribute Alignment

Shanghai AI Laboratory(上海人工智能实验室);Shanghai Jiao Tong University(上海交通大学)

Yankai Jiang, Wenhui Lei, Xiaofan Zhang, Shaoting Zhang

AI总结:

本文提出 Malenia 框架,通过多尺度掩码-属性对齐和跨模态知识注入模块,将已见病灶的可扩展知识与未见病灶视觉特征链接,有效解决 3D CT 零样本病灶分割任务。

摘要:

医学视觉-语言预训练模型的最新进展推动了零样本疾病识别的显著进步。然而,将图像级知识迁移至像素级任务(如 3D CT 扫描中的病灶分割)仍是一项关键挑战。由于病理视觉特征的复杂性与多变性,现有方法难以将训练中未遇到的细粒度病灶特征与疾病相关的文本表示进行对齐。本文提出 Malenia,一种新颖的多尺度病灶级掩码-属性对齐框架,专为 3D 零样本病灶分割设计。Malenia 提升了掩码表示与其关联元素属性间的兼容性,明确将未见病灶的视觉特征与从已见病灶学到的可扩展知识相链接。此外,我们设计了 Cross-Modal Knowledge Injection 模块,利用互益信息增强视觉与文本特征,有效引导分割结果的生成。跨三个数据集和 12 个病灶类别的综合实验验证了 Malenia 的优越性能。

英文摘要:

Recent advancements in medical vision-language pre-training models have driven significant progress in zero-shot disease recognition. However, transferring image-level knowledge to pixel-level tasks, such as lesion segmentation in 3D CT scans, remains a critical challenge. Due to the complexity and variability of pathological visual characteristics, existing methods struggle to align fine-grained lesion features not encountered during training with disease-related textual representations. In this paper, we present Malenia, a novel multi-scale lesion-level mask-attribute alignment framework, specifically designed for 3D zero-shot lesion segmentation. Malenia improves the compatibility between mask representations and their associated elemental attributes, explicitly linking the visual features of unseen lesions with the extensible knowledge learned from previously seen ones. Furthermore, we design a Cross-Modal Knowledge Injection module to enhance both visual and textual features with mutually beneficial information, effectively guiding the generation of segmentation results. Comprehensive experiments across three datasets and 12 lesion categories validate the superior performance of Malenia.

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

↑