Uncertainty-aware transformer model for anatomical landmark detection in paraspinal muscle MRIs

Soorena Salari; Amirhossein Rasoulian; Michele Battie; Maryse Fortin; Hassan Rivaz; Yiming Xiao

doi:10.1117/12.2653155

3 April 2023 Uncertainty-aware transformer model for anatomical landmark detection in paraspinal muscle MRIs

Soorena Salari, Amirhossein Rasoulian, Michele Battie, Maryse Fortin, Hassan Rivaz, Yiming Xiao

Proceedings Volume 12464, Medical Imaging 2023: Image Processing; 1246414 (2023) https://doi.org/10.1117/12.2653155
Event: SPIE Medical Imaging, 2023, San Diego, California, United States

Abstract

Anatomical landmark identification is crucial in the registration of medical images in a wide range of clinical applications. Various machine and deep learning (DL) techniques have been proposed to annotate anatomical landmarks automatically. However, very few have taken advantage of the more recent Transformer models that do not suffer from inductive bias of convolutional neural networks as well as incorporating uncertainty assessments. This paper proposes a novel technique based on the Swin Transformer V2 (Swin-V2) with an uncertainty quantification module for robust anatomical landmark detection in paraspinal muscle MRIs to facilitate muscle morphometric analysis for low back pain (LBP). Specifically, we employ statistical measurements from stochastic sampling using Monte Carlo (MC) dropouts as uncertainty metrics for automatic landmark detection. The proposed method was trained and validated using 350 axial MRI slices of paraspinal muscles at the mid-disc spinal levels commonly associated with LBP. We demonstrate that the selected uncertainty metrics are correlated with automatic landmark detection errors and, in addition, can be used as inputs to grade the quality of the identified landmarks with a lightweight random forest algorithm for a more straightforward interpretation.

Conference Presentation

Citation Download Citation

Soorena Salari, Amirhossein Rasoulian, Michele Battie, Maryse Fortin, Hassan Rivaz, and Yiming Xiao "Uncertainty-aware transformer model for anatomical landmark detection in paraspinal muscle MRIs", Proc. SPIE 12464, Medical Imaging 2023: Image Processing, 1246414 (3 April 2023); https://doi.org/10.1117/12.2653155

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available