Enhancing representation in radiography-reports foundation model: a granular alignment algorithm using masked contrastive learning
Recently, multi-modal vision-language foundation models have gained significant attention in the medical field. While these models offer great opportunities, they still face crucial challenges, such as the requirement for fine-grained knowledge understanding in computer-aided diagnosis and the capability of utilizing very limited or even no task-specific labeled data in real-world clinical applications. In this study, we present MaCo, a masked contrastive chest X-ray foundation model that tackles these challenges. MaCo explores masked contrastive learning to simultaneously achieve fine-grained image understanding and zero-shot learning for a variety of medical imaging tasks. It designs a correlation weighting mechanism to adjust the correlation between masked chest X-ray image patches and their corresponding reports, thereby enhancing the model's representation learning capabilities. To evaluate the performance of MaCo, we conducted extensive experiments using 6 well-known open-source X-ray datasets. The experimental results demonstrate the superiority of MaCo over 10 state-of-the-art approaches across tasks such as classification, segmentation, detection, and phrase grounding. These findings highlight the significant potential of MaCo in advancing a wide range of medical image analysis tasks.
Authors
- Weijian Huang (ORCID: https://orcid.org/0000-0003-4835-8266)
- Jiarun Liu (ORCID: https://orcid.org/0000-0002-6921-1549)
- Hairong Zheng (ORCID: https://orcid.org/0000-0002-8558-5102)
- Shaoting Zhang (ORCID: https://orcid.org/0000-0002-8719-448X)
- Cheng Li (ORCID: https://orcid.org/0000-0001-5400-2093)
- Shanshan Wang (ORCID: https://orcid.org/0000-0002-0575-6523)
- Hao Yang (ORCID: https://orcid.org/0000-0002-4378-5755)
- Hong-Yu Zhou (ORCID: https://orcid.org/0000-0002-1256-7050)
- Yong Liang (ORCID: https://orcid.org/0000-0001-6021-2791)
Institutions
- Harvard University (US)
- Shanghai Jiao Tong University (CN)
- Chinese Academy of Sciences (CN)
- Peng Cheng Laboratory (CN)
- Shenzhen Institutes of Advanced Technology (CN)
- University of Chinese Academy of Sciences (CN)
Publication Details
- Journal
- Nature Communications
- Published
- 2024-09-02
- DOI
- https://doi.org/10.1038/s41467-024-51749-0
- Citations
- 49
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 19.81
Funders
- National Natural Science Foundation of China