NA-MIC / ProjectWeek

Website for NA-MIC Project Weeks
https://projectweek.na-mic.org
84 stars 284 forks source link

Project: Deep learning model for B-line detection in lung ultrasound videos using crowdsourced labels #732

Closed goldenfire6 closed 1 year ago

goldenfire6 commented 1 year ago

Category

Segmentation / Classification / Landmarking

Presenter Location

In-person

Key Investigators

Project Description

Automated B-line detection in lung ultrasound videos has been demonstrated before, most recently by Lucassen 2023. However, acquiring the many labels necessary can be a resource-intensive process, limited by the availability of expert clinicians capable of producing high-quality labels. Recently, gamified crowdsourcing with a new quality control mechanism and built-in learning for labelers has been demonstrated to be capable of producing annotations on lung ultrasound videos comparable in quality to expert clinicians (as well as analogous results for EEG and skin lesion classification tasks), which can greatly shorten the time required to acquire high-quality labels for model training. Though these crowd labels have been shown to have expert-level quality, it has yet to be demonstrated whether crowd-produced labels are capable of training high-performance models.

Objective

  1. Train a deep learning model to classify lung ultrasound videos as having B-lines or having no B-lines.

Approach and Plan

  1. Create a data file associating all 3000+ clips with filepath, crowd label, and expert labels (for those that have expert labels).
  2. Adapt the model (ResNet(2+1)D-18 or similar pretrained model) and training procedure used in Lucassen 2023 to train a new model on a new crowd-labeled dataset of 3000+ lung ultrasound videos from 500 patients.
  3. Evaluate the model performance and compare to previously reported model performance for ultrasound video classification of B-line presence.

Progress and Next Steps

  1. De-identified and masked 3000+ lung ultrasound clips
  2. Uploaded 3000+ clips with standard filename format to a GPU cluster.
  3. Crowd-labeled all 3000+ lung ultrasound clips using 193 clips from ~70 patients for crowd training.

Illustrations

No response

Background and References

https://pubmed.ncbi.nlm.nih.gov/37276107/

github-actions[bot] commented 1 year ago

Project Page Pull Request Creation

:white_check_mark: COMPLETED: See https://github.com/NA-MIC/ProjectWeek/pull/733