Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

Yankai Jiang, Wenhui Lei, Xiaofan Zhang, Shaoting Zhang• 2024

Related benchmarks

TaskDatasetResultRank
Tumor SegmentationKiTS Kidney Tumor 23
DSC55.37
12
Tumor SegmentationMSD Pancreas Tumor
DSC40.26
9
Tumor SegmentationMSD Lung Tumor
DSC32.75
9
Tumor SegmentationMSD Liver Tumor
DSC59.83
9
Tumor SegmentationMSD Colon Tumor
DSC34.72
9
Tumor SegmentationMSD Hepatic Vessel Tumor
DSC59.71
9
Tumor SegmentationIn-house MRI Liver Tumor
DSC0.2903
9
Tumor SegmentationMSD Brain
DSC19.83
9
Tumor SegmentationBraTS 23
DSC17.94
9
Showing 9 of 9 rows

Other info

Follow for update