English

Generalized Radiograph Representation Learning via Cross-supervision between Images and Free-text Radiology Reports

Image and Video Processing 2022-01-28 v2 Computer Vision and Pattern Recognition Machine Learning

Abstract

Pre-training lays the foundation for recent successes in radiograph analysis supported by deep learning. It learns transferable image representations by conducting large-scale fully-supervised or self-supervised learning on a source domain. However, supervised pre-training requires a complex and labor intensive two-stage human-assisted annotation process while self-supervised learning cannot compete with the supervised paradigm. To tackle these issues, we propose a cross-supervised methodology named REviewing FreE-text Reports for Supervision (REFERS), which acquires free supervision signals from original radiology reports accompanying the radiographs. The proposed approach employs a vision transformer and is designed to learn joint representations from multiple views within every patient study. REFERS outperforms its transfer learning and self-supervised learning counterparts on 4 well-known X-ray datasets under extremely limited supervision. Moreover, REFERS even surpasses methods based on a source domain of radiographs with human-assisted structured labels. Thus REFERS has the potential to replace canonical pre-training methodologies.

Keywords

Cite

@article{arxiv.2111.03452,
  title  = {Generalized Radiograph Representation Learning via Cross-supervision between Images and Free-text Radiology Reports},
  author = {Hong-Yu Zhou and Xiaoyu Chen and Yinghao Zhang and Ruibang Luo and Liansheng Wang and Yizhou Yu},
  journal= {arXiv preprint arXiv:2111.03452},
  year   = {2022}
}

Comments

Accepted by Nature Machine Intelligence. The official version is at https://www.nature.com/articles/s42256-021-00425-9. Codes are available at https://github.com/funnyzhou/REFERS