Cross-Modal Conditioned Reconstruction for Language-Guided Medical Image Segmentation
Recent developments underscore the potential of textual information in enhancing learning models for a deeper understanding of medical visual semantics. However, language-guided medical image segmentation still faces a challenging issue. Previous works employ implicit architectures to embed textual information. This leads to segmentation results that are inconsistent with the semantics represented by the language, sometimes even diverging significantly. To this end, we propose a novel cross-modal conditioned Reconstruction for Language-guided Medical Image Segmentation (RecLMIS) to explicitly capture cross-modal interactions, which assumes that well-aligned medical visual features and medical notes can effectively reconstruct each other. We introduce conditioned interaction to adaptively predict patches and words of interest. Subsequently, they are utilized as conditioning factors for mutual reconstruction to align with regions described in the medical notes. Extensive experiments demonstrate the superiority of our RecLMIS, surpassing LViT by 3.74% mIoU on the MosMedData+ dataset and 1.89% mIoU on the QATA-CoV19 dataset. More importantly, we achieve a relative reduction of 20.2% in parameter count and a 55.5% decrease in computational load. The code will be available at https://github.com/ShawnHuang497/RecLMIS.
Authors
- Hongxiang Li (ORCID: https://orcid.org/0009-0000-7710-8835)
- Meng Cao (ORCID: https://orcid.org/0000-0002-1008-5509)
- Chenyu You (ORCID: https://orcid.org/0000-0001-8365-7822)
- Long Chen (ORCID: https://orcid.org/0000-0001-6148-9709)
- Xiaoshuang Huang (ORCID: https://orcid.org/0009-0002-9787-3643)
- Dong An (ORCID: https://orcid.org/0000-0002-8389-987X)
Institutions
- Hong Kong University of Science and Technology (HK)
- Peking University (CN)
- Yale University (US)
- Mohamed bin Zayed University of Artificial Intelligence (AE)
- China Agricultural University (CN)
- University of Hong Kong (HK)
Publication Details
- Journal
- IEEE Transactions on Medical Imaging
- Published
- 2024-12-26
- DOI
- https://doi.org/10.1109/tmi.2024.3523333
- Citations
- 15
- Primary Topic
- Medical Image Segmentation Techniques
- Type
- article
- Field-Weighted Citation Impact
- 3.74
Funders
- National Key Research and Development Program of China