Edit this page

NA-MIC Project Weeks

Back to Projects List

AI model development for lung ultrasound analysis

Key Investigators

Project Description

One of the primary indicators of acute heart failure is the presence of pulmonary congestion. To detect the fluid build up quickly the in emergency room, lung ultrasound exams are taken of the patient. Clinicians look for hyperechoic artifacts (B-lines) that appear in the image, where the more they see, the more congested the patient is.

Unfortunately, the manual detection of these B-lines is difficult due to the image quality, which depends on the type of transducer and the expertise of the clinician. Therefore, AI models have started to be developed to quickly detect these B-lines. Our group, over the past few years, has developed multiple methods [1], [2], [3]. One of the drawbacks though, is that the models do not compute the pleural percentage, which is the ratio of the B-line sectors to the pleural line sectors.

Therefore, we have started to investigate using object detection models for automatically finding the pleural lines and B-lines. Our goal for this project week is to train and validate detection models such as YOLO, [4] and talk to clinicians about the performance.

Objective

  1. Train and validate YOLO models for pleural line and B-line detection.
  2. Train and validate other state-of-the-art methods, such as RF-DETR [5]
  3. Show and discuss our results with clinicians.

Approach and Plan

  1. Develop models for pleural line and B-line detection.
  2. Make our model/code publicly available.
  3. Talk to clinicians

Progress and Next Steps

  1. We have trained some preliminary YOLO models. We trained one oriented bounding box YOLO model for pleural lines, and another axis-aligned box YOLO model for B-lines.
  2. We calculated the percent pleura and compared it to the experts.

Illustrations

Ground truth boxes in scanline space: oriented bounding box for the pleural line (left), and axis-aligned bounding box for the B-lines. Image

YOLO results after performing pleural line detection and B-line detection, and conversion back to the sector space, compared to the ground truth: Image

Percent pleura for one test lung zone - x = frame #, and y = percentage. We see that the AI predicted Image

Background and References