Domain-Adaptive Framework for ACL Injury Diagnosis Utilizing Contrastive Learning Techniques

Abstract

In sports medicine, anterior cruciate ligament (ACL) injuries are common and have a major effect on knee joint stability. For the sake of prognosis evaluation and treatment planning, an accurate clinical auxiliary diagnosis of ACL injuries is essential. Although existing deep learning techniques for ACL diagnosis work well on single datasets, research on cross-domain data transfer is still lacking. Building strong domain-adaptive diagnostic models requires addressing domain disparities in ACL magnetic resonance imaging (MRI) from different hospitals and making efficient use of multiple ACL datasets. This work uses the publicly available KneeMRI dataset from Croatian hospitals coupled with the publicly available MRnet dataset from Stanford University to investigate domain adaptation and transfer learning models. First, an optimized model efficiently screens training data in the source domain to find unusually misclassified occurrences. Subsequently, before being integrated into the contrastive learning module, a target domain feature extraction module processes features of target domain samples to improve extraction efficiency. By using contrastive learning between positive and negative sample pairs from source and target domains, this method makes domain adaptation easier and improves the efficacy of ACL auxiliary diagnostic models. Utilizing a spatially augmented ResNet-18 backbone network, the suggested approach produces notable enhancements in experimentation. To be more precise, the AUC for transfer learning improved by 3.5% from MRnet to KneeMRI and by 2.5% from KneeMRI to MRnet (from 0.845 to 0.870). This method shows how domain transfer can be used to improve diagnostic accuracy on a variety of datasets and effectively progresses the training of a strong ACL auxiliary diagnostic model.