vix.ing · top · new · best · stats · spec

Weakly Supervised Contrastive Learning for Better Severity Scoring of Lung Ultrasound

2022/01/18 by Gautam Rajendrakumar Gare, Hai V. Tran, Gare, Gautam Rajendrakumar +7
Medicine · #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #FOS: Electrical engineering #Image and Video Processing (eess.IV) #Lung Cancer Diagnosis and Treatment #Machine Learning (cs.LG) #Radiology practices and education #Ultrasound in Clinical Applications #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2201.07357

openalex publication_date 2022/01/18 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

With the onset of the COVID-19 pandemic, ultrasound has emerged as an effective tool for bedside monitoring of patients. Due to this, a large amount of lung ultrasound scans have been made available which can be used for AI based diagnosis and analysis. Several AI-based patient severity scoring models have been proposed that rely on scoring the appearance of the ultrasound scans. AI models are trained using ultrasound-appearance severity scores that are manually labeled based on standardized visual features. We address the challenge of labeling every ultrasound frame in the video clips. Our contrastive learning method treats the video clip severity labels as noisy weak severity labels for individual frames, thus requiring only video-level labels. We show that it performs better than the conventional cross-entropy loss based training. We combine frame severity predictions to come up with video severity predictions and show that the frame based model achieves comparable performance to a video based TSM model, on a large dataset combining public and private sources.

Related