Unsupervised Domain Adaptation for Low-Dose CT Reconstruction via Bayesian Uncertainty Alignment

Kecheng Chen, Jie Liu, Renjie Wan*, Victor Ho-Fun Lee, Varut Vardhanabhuti, Hong Yan, Haoliang Li*

*Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

Abstract

Low-dose computed tomography (LDCT) image reconstruction techniques can reduce patient radiation exposure while maintaining acceptable imaging quality. Deep learning (DL) is widely used in this problem, but the performance of testing data (also known as target domain) is often degraded in clinical scenarios due to the variations that were not encountered in training data (also known as source domain). Unsupervised domain adaptation (UDA) of LDCT reconstruction has been proposed to solve this problem through distribution alignment. However, existing UDA methods fail to explore the usage of uncertainty quantification, which is crucial for reliable intelligent medical systems in clinical scenarios with unexpected variations. Moreover, existing direct alignment for different patients would lead to content mismatch issues. To address these issues, we propose to leverage a probabilistic reconstruction framework to conduct a joint discrepancy minimization between source and target domains in both the latent and image spaces. In the latent space, we devise a Bayesian uncertainty alignment to reduce the epistemic gap between the two domains. This approach reduces the uncertainty level of target domain data, making it more likely to render well-reconstructed results on target domains. In the image space, we propose a sharpness-aware distribution alignment (SDA) to achieve a match of second-order information, which can ensure that the reconstructed images from the target domain have similar sharpness to normal-dose CT (NDCT) images from the source domain. Experimental results on two simulated datasets and one clinical low-dose imaging dataset show that our proposed method outperforms other methods in quantitative and visualized performance. © 2024 IEEE.
Original languageEnglish
Pages (from-to)8525-8539
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume36
Issue number5
Online published10 Jul 2024
DOIs
Publication statusPublished - May 2025

Funding

This work was supported in part by the Hong Kong Innovation and Technology Commission (InnoHK Project ClMDA), in part by the Hong Kong Research Grants Council under Project 21200522 and Project 11204821, in part by the City University of Hong Kong (CityU) New Research Initiatives/Infrastructure Support from Central under Grant APRC 9610528, in part by the National Natural Science Foundation of China under Grant 62302415, in part by the Guangdong Basic and Applied Basic Research Foundation under Grant 2022A1515110692 and Grant 2024A1515012822, in part by the Blue Sky Research Fund of Hong Kong Baptist University (HKBU) under Grant BSRF/21-22/16, in part by the Seed Fund for Collaborative Research under Project 2207101536, and in part by The University of Hong Kong

Research Keywords

  • Adversarial learning
  • Noise modeling
  • Probabilistic model
  • Robust CT reconstruction

Fingerprint

Dive into the research topics of 'Unsupervised Domain Adaptation for Low-Dose CT Reconstruction via Bayesian Uncertainty Alignment'. Together they form a unique fingerprint.

Cite this