HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SKID: Self-Supervised Learning for Knee Injury Diagnosis from MRI Data

Siladittya Manna Saumik Bhattacharya Umapada Pal

SKID: Self-Supervised Learning for Knee Injury Diagnosis from MRI Data

Abstract

In medical image analysis, the cost of acquiring high-quality data and their annotation by experts is a barrier in many medical applications. Most of the techniques used are based on supervised learning framework and need a large amount of annotated data to achieve satisfactory performance. As an alternative, in this paper, we propose a self-supervised learning (SSL) approach to learn the spatial anatomical representations from the frames of magnetic resonance (MR) video clips for the diagnosis of knee medical conditions. The pretext model learns meaningful spatial context-invariant representations. The downstream task in our paper is a class imbalanced multi-label classification. Different experiments show that the features learnt by the pretext model provide competitive performance in the downstream task. Moreover, the efficiency and reliability of the proposed pretext model in learning representations of minority classes without applying any strategy towards imbalance in the dataset can be seen from the results. To the best of our knowledge, this work is the first work of its kind in showing the effectiveness and reliability of self-supervised learning algorithms in class imbalanced multi-label classification tasks on MR videos. The code for evaluation of the proposed work is available at https://github.com/sadimanna/skid.

Code Repositories

sadimanna/sslm
Official
tf
sadimanna/skid
Official
tf

Benchmarks

BenchmarkMethodologyMetrics
multi-label-classification-on-mrnetSKIDv3
AUC on ACL Tear (ACL): 0.893
AUC on Abnormality (ABN): 0.904
AUC on Meniscus Tear (MEN): 0.810
Accuracy on ACL Tear (ACL): 0.800
Accuracy on Abnormality (ABN): 0.874
Accuracy on Meniscus Tear (MEN): 0.725
Average AUC: 0.869
Average Accuracy: 0.799

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp