Asymmetric Source-Free Unsupervised Domain Adaptation for Medical Image Diagnosis

  • Yajie Zhang
  • , Zhi An Huang*
  • , Jibin Wu
  • , Kay Chen Tan*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Existing source-free unsupervised domain adaptation (SFUDA) methods primarily focus on addressing the domain gap issue for single-modal data, overlooking two crucial aspects: 1) In medical scenarios, clinicians often rely on multi-modal information for disease diagnosis. Consequently, emphasizing single-modal (symmetric modality) SFUDA algorithms neglect the complementary information from other modalities (asymmetric modalities). 2) Restricting SFUDA to a single modality limits downstream institutions's ability to handle diverse modalities beyond that singular modality. To tackle these challenges, we propose an Asymmetric Source-Free Unsupervised Domain Adaptation (A-SFUDA) algorithm. This method leverages source model and unlabeled data from both symmetric and asymmetric modalities in the target domain for disease diagnosis. A-SFUDA adopts a two-stage training approach. In the first stage, A-SFUDA employs knowledge distillation (KD) to obtain two models capable of handling symmetric and asymmetric data in the target domain, facilitating preliminary diagnosis ability. In the second stage, A-SFUDA optimizes the target models through a pseudo-label correction mechanism based on multi-modal prediction correction and class-centered distance correction. Incorporating the two pseudo-label correction modules effectively mitigates noise within the training data, thereby facilitating the learning of the target models. We validate the performance of the proposed A-SFUDA algorithm on a large chest X-ray dataset, demonstrating its excellent performance for disease diagnosis in the target domain.

Original languageEnglish
Title of host publicationProceedings - 2024 IEEE Conference on Artificial Intelligence, CAI 2024
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages234-239
Number of pages6
ISBN (Electronic)9798350354096
DOIs
StatePublished - 2024
Event2nd IEEE Conference on Artificial Intelligence, CAI 2024 - Singapore, Singapore
Duration: 25 Jun 202427 Jun 2024

Publication series

NameProceedings - 2024 IEEE Conference on Artificial Intelligence, CAI 2024

Conference

Conference2nd IEEE Conference on Artificial Intelligence, CAI 2024
Country/TerritorySingapore
CitySingapore
Period25/06/2427/06/24

Keywords

  • asymmetric modality
  • pseudo-labeling
  • source-free
  • unsupervised domain adaptation

Fingerprint

Dive into the research topics of 'Asymmetric Source-Free Unsupervised Domain Adaptation for Medical Image Diagnosis'. Together they form a unique fingerprint.

Cite this