MV2SwimNet: A lightweight transformer-based hybrid model for knee meniscus tears detection.

Journal: PloS one
Published Date:

Abstract

Knee Ailments, such as meniscus injuries, bother millions globally, with research showing that more than 14% of the population above 40 years lives with meniscus-related conditions. Conventional diagnosis techniques, like manual MRI interpretation, are labour-intensive, error-prone, and dependent on skilled radiologists, making an automatic and more accurate alternative indispensable. Current deep-learning solutions heavily depend on CNNs, which perform poorly in long-range dependencies and global contextual info. We proposed MV2SwimNet, a hybrid of MobileNetV2 and Swin Transformer, integrating Window Multi-Head Self-Attention (W-MSA) and Multi-Stage Hierarchical Representation (MSHR), efficiently incorporating both local and global features towards enhanced diagnostic capability. Our strategy utilizes the efficiency of lightweight MobileNetV2 coupled with a hierarchical architecture and self-attention-based Swin Transformer, enabling better spatial representation and advanced feature extraction. W-MSA allows our model to process MRI scans effectively by attending to the corresponding regions of images. In contrast, MSHR adjusts feature representations across different levels in a way that allows for progressive and robust learning in stages. We tested MV2SwimNet on two sets using 3-fold cross-validation and achieved 99.94% and 96.04% accuracy on dataset1 and dataset2, which beats state-of-the-art techniques. These results confirm MV2SwimNet efficiency, robustness, and real-world application potential in medicine, providing a highly accurate, automated medical diagnosis tool for knee disease detection. The code of the proposed method can be accessed through the URL: https://github.com/Visheshtanwar/MV2SwimNet.

Authors

  • Vishesh Tanwar
    Chitkara University Institute of Engineering and Technology, Chitkara University, Rajpura 140401, Punjab, India.
  • Bhisham Sharma
    Chitkara University School of Engineering and Technology, Chitkara University, Himachal Pradesh, India.
  • Dhirendra Prasad Yadav
    Department of Computer Engineering and Applications, GLA University, Mathura 281406, Uttar Pradesh, India.
  • Julian L Webber
    Graduate School of Engineering Science, Osaka University, Osaka, Japan.
  • Abolfazl Mehbodniya
    Department of Electronics and Communication Engineering, Kuwait College of Science and Technology, Kuwait City, Kuwait.