Task-specific self-supervision for CT image denoising

Ayaan Haque, Adam Wang, Abdullah Al Zubaer Imran

Research output: Contribution to journalArticlepeer-review

Abstract

CT image quality is largely reliant on radiation dose, which causes a trade-off between image quality and dose, affecting the subsequent image-based diagnostic and treatment performances. Deep learning approaches for low-dose CT image denoising require access to large training sets and specifically the reference full-dose images, which can be difficult to obtain. Self-supervised learning enables learning with reduced reference data burden. Currently available self-supervised CT denoising works are either dependent on foreign domains or pretexts that are not very task-relevant, requiring lots of adjustments (architectural, loss function, training parameters, etc.) and additional skills to perform the downstream denoising task. To tackle the aforementioned challenges, we propose a novel self-supervised pretraining approach, namely Self-Supervised Window-Leveling for Image DeNoising (SSWL-IDN), leveraging an innovative, task-relevant, simple yet effective surrogate–prediction of the window-leveled equivalent. SSWL-IDN leverages residual learning and a hybrid loss combining perceptual loss and MSE, all incorporated in a VAE framework. Our extensive (in- and cross-domain) experimentation demonstrates the effectiveness of SSWL-IDN in aggressive denoising of CT (abdomen and chest) images acquired at two different dose levels (5% and 25%), without requiring any kind of architectural or downstream training adjustments. We have made our code publicly available at https://github.com/ayaanzhaque/SSWL-IDN.

Original languageEnglish
Pages (from-to)2196-2208
Number of pages13
JournalComputer Methods in Biomechanics and Biomedical Engineering: Imaging and Visualization
Volume11
Issue number6
DOIs
StatePublished - 2023

Bibliographical note

Publisher Copyright:
© 2023 Informa UK Limited, trading as Taylor & Francis Group.

Keywords

  • Computed tomography
  • image denoising
  • self-supervised learning

ASJC Scopus subject areas

  • Computational Mechanics
  • Biomedical Engineering
  • Radiology Nuclear Medicine and imaging
  • Computer Science Applications

Fingerprint

Dive into the research topics of 'Task-specific self-supervision for CT image denoising'. Together they form a unique fingerprint.

Cite this