Learning Generalized Medical Image Representations through Image-Graph Contrastive Pretraining
Not on the map. Why?
Abstract (arXiv)
Medical image interpretation using deep learning has shown promise but often requires extensive expert-annotated datasets. To reduce this annotation burden, we develop an Image-Graph Contrastive Learning framework that pairs chest X-rays with structured report knowledge graphs automatically extracted from radiology notes. Our approach uniquely encodes the disconnected graph components via a relational graph convolution network and transformer attention. In experiments on the CheXpert dataset, this novel graph encoding strategy enabled the framework to outperform existing methods that use image-text contrastive learning in 1% linear evaluation and few-shot settings, while achieving comparable performance to radiologists. By exploiting unlabeled paired images and text, our framework demonstrates the potential of structured clinical insights to enhance contrastive learning for medical images. This work points toward reducing demands on medical experts for annotations, improving diagnostic precision, and advancing patient care through robust medical image understanding.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 21 of its 39 references that are in the map. First posted on arXiv 15 May 2024.
Builds on
21 of 39 references.
References · 21
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2017Graph Attention Networkscited by 8,923
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2017Modeling Relational Data with Graph Convolutional Networkscited by 1,694
- 2019CheXpert: A Large Chest Radiograph Dataset with Uncertainty Labels and Expert Comparisoncited by 1,423
- 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 898
- 2019Self-Supervised Learning of Pretext-Invariant Representationscited by 808
- 2018A Survey on Deep Transfer Learningcited by 419
9 more
- 2021RadGraph: Extracting Clinical Entities and Relations from Radiology Reportscited by 319
- 2020COVID-Net: A Tailored Deep Convolutional Neural Network Design for Detection of COVID-19 Cases from Chest X-Ray Imagescited by 264
- 2021Generalized Radiograph Representation Learning via Cross-supervision between Images and Free-text Radiology Reportscited by 80
- 2021CheXtransfer: Performance and Parameter Efficiency of ImageNet Models for Chest X-Ray Interpretationcited by 57
- 2021MedAug: Contrastive learning leveraging patient metadata improves representations for chest X-ray interpretationcited by 52
- 2019Alchemy: A Quantum Chemistry Dataset for Benchmarking AI Modelscited by 36
- 2021COVID-19 Prognosis via Self-Supervised Representation Learning and Multi-Image Predictioncited by 36
- 2022Contrastive Language-Image Pre-Training with Knowledge Graphscited by 34
- 2020CheXpedition: Investigating Generalization Challenges for Translation of Chest X-Ray Algorithms to the Clinical Settingcited by 18
Built on by
Citing papers per year: 2026: 1.
1 paper.