Unsupervised Domain Adaptation for Automated Knee Osteoarthritis Phenotype Classification

TL;DR

This study demonstrates that unsupervised domain adaptation significantly enhances automated knee osteoarthritis phenotype classification accuracy on small datasets by leveraging large source datasets, outperforming traditional training methods.

Contribution

The paper introduces a novel UDA pipeline that improves knee OA phenotype classification on small datasets by adapting models trained on large source datasets, with comprehensive validation.

Findings

UDA improves classification metrics over non-UDA methods.

The approach effectively utilizes large source datasets for small target datasets.

Enhanced AUROC, sensitivity, specificity, and accuracy with UDA.

Abstract

Purpose: The aim of this study was to demonstrate the utility of unsupervised domain adaptation (UDA) in automated knee osteoarthritis (OA) phenotype classification using a small dataset (n=50). Materials and Methods: For this retrospective study, we collected 3,166 three-dimensional (3D) double-echo steady-state magnetic resonance (MR) images from the Osteoarthritis Initiative dataset and 50 3D turbo/fast spin-echo MR images from our institute (in 2020 and 2021) as the source and target datasets, respectively. For each patient, the degree of knee OA was initially graded according to the MRI Osteoarthritis Knee Score (MOAKS) before being converted to binary OA phenotype labels. The proposed UDA pipeline included (a) pre-processing, which involved automatic segmentation and region-of-interest cropping; (b) source classifier training, which involved pre-training phenotype classifiers on the source dataset; (c) target encoder adaptation, which involved unsupervised adaption of the source encoder to the target encoder and (d) target classifier validation, which involved statistical analysis of the target classification performance evaluated by the area under the receiver operating characteristic curve (AUROC), sensitivity, specificity and accuracy. Additionally, a classifier was trained without UDA for comparison. Results: The target classifier trained with UDA achieved improved AUROC, sensitivity, specificity and accuracy for both knee OA phenotypes compared with the classifier trained without UDA. Conclusion: The proposed UDA approach improves the performance of automated knee OA phenotype classification for small target datasets by utilising a large, high-quality source dataset for training. The results successfully demonstrated the advantages of the UDA approach in classification on small datasets.