Anomaly Detection for Medical Images Using Self-Supervised and Translation-Consistent Features

He Zhao, Yuexiang Li*, Nanjun He, Kai Ma, Leyuan Fang, Huiqi Li*, Yefeng Zheng

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

69 Citations (Scopus)

Abstract

As the labeled anomalous medical images are usually difficult to acquire, especially for rare diseases, the deep learning based methods, which heavily rely on the large amount of labeled data, cannot yield a satisfactory performance. Compared to the anomalous data, the normal images without the need of lesion annotation are much easier to collect. In this paper, we propose an anomaly detection framework, namely SALAD , extracting S elf-supervised and tr A ns L ation-consistent features for A nomaly D etection. The proposed SALAD is a reconstruction-based method, which learns the manifold of normal data through an encode-and-reconstruct translation between image and latent spaces. In particular, two constraints (i.e., structure similarity loss and center constraint loss) are proposed to regulate the cross-space (i.e., image and feature) translation, which enforce the model to learn translation-consistent and representative features from the normal data. Furthermore, a self-supervised learning module is engaged into our framework to further boost the anomaly detection accuracy by deeply exploiting useful information from the raw normal data. An anomaly score, as a measure to separate the anomalous data from the healthy ones, is constructed based on the learned self-supervised-and-translation-consistent features. Extensive experiments are conducted on optical coherence tomography (OCT) and chest X-ray datasets. The experimental results demonstrate the effectiveness of our approach.

Original languageEnglish
Pages (from-to)3641-3651
Number of pages11
JournalIEEE Transactions on Medical Imaging
Volume40
Issue number12
DOIs
Publication statusPublished - 1 Dec 2021

Keywords

  • Medical image analysis
  • anomaly detection
  • feature space constraint
  • generative adversarial networks

Fingerprint

Dive into the research topics of 'Anomaly Detection for Medical Images Using Self-Supervised and Translation-Consistent Features'. Together they form a unique fingerprint.

Cite this