Contrastive Cross-Modal Pre-Training: A General Strategy for Small Sample Medical Imaging.

Journal: IEEE journal of biomedical and health informatics
Published Date:

Abstract

A key challenge in training neural networks for a given medical imaging task is the difficulty of obtaining a sufficient number of manually labeled examples. In contrast, textual imaging reports are often readily available in medical records and contain rich but unstructured interpretations written by experts as part of standard clinical practice. We propose using these textual reports as a form of weak supervision to improve the image interpretation performance of a neural network without requiring additional manually labeled examples. We use an image-text matching task to train a feature extractor and then fine-tune it in a transfer learning setting for a supervised task using a small labeled dataset. The end result is a neural network that automatically interprets imagery without requiring textual reports during inference. We evaluate our method on three classification tasks and find consistent performance improvements, reducing the need for labeled data by 67%-98%.

Authors

  • Gongbo Liang
    Department of Computer Science, University of Kentucky, Lexington, Kentucky.
  • Connor Greenwell
  • Yu Zhang
    College of Marine Electrical Engineering, Dalian Maritime University, Dalian, China.
  • Xin Xing
  • Xiaoqin Wang
  • Ramakanth Kavuluru
    Div. of Biomedical Informatics, Dept. of Internal Medicine, Dept. of Computer Science, University of Kentucky, Lexington, KY.
  • Nathan Jacobs