Improving lesion detection algorithm in digital breast tomosynthesis leveraging ensemble cross-validation models with multi-depth levels

Belayat Hossain; Robert M. Nishikawa; Juhun Lee

doi:10.1117/12.2611007

4 April 2022 Improving lesion detection algorithm in digital breast tomosynthesis leveraging ensemble cross-validation models with multi-depth levels

Belayat Hossain, Robert M. Nishikawa, Juhun Lee

Author Affiliations +

Proceedings Volume 12033, Medical Imaging 2022: Computer-Aided Diagnosis; 120330D (2022) https://doi.org/10.1117/12.2611007
Event: SPIE Medical Imaging, 2022, San Diego, California, United States

Abstract

We report an improved algorithm for detecting biopsy-proven breast lesions on digital breast tomosynthesis (DBT) where the given positive samples in the training set were limited. Instead of using a large scale inhouse dataset, our original algorithm used false positive findings (FPs) from non-biopsied (actionable) images to tackle the problem of a limited number of trainable samples. In this study, we further improved our algorithm by fusing multiple weak lesion detection models by using an ensemble approach. We used cross-validation (CV) to develop multiple lesion detection models. We first constructed baseline detection algorithms by varying the depth levels (medium and large) of the convolutional layers in the YOLOv5 algorithm using biopsied samples. We detected actionable FPs in non-biopsied images using a medium baseline model. We fine-tuned the baseline algorithms using the identified actionable FPs and the biopsied samples. For lesion detection, we processed the DBT volume slice-by-slice, then combined the estimated lesion of each slice along the depth of the DBT volume using volumetric morphological closing. Using 5-fold CV, we developed different multi-depth detection models for each depth level. Finally, we developed an ensemble algorithm by combining CV models with different depth levels. Our new algorithm achieved a mean sensitivity of 0.84 per DBT volume in the independent validation set from the DBTex challenge set, close to one of the top performing algorithms utilizing large inhouse data. These results show that our ensemble approach on different CV models is useful for improving the performance of the lesion detection algorithms.

Conference Presentation

Citation Download Citation

Belayat Hossain, Robert M. Nishikawa, and Juhun Lee "Improving lesion detection algorithm in digital breast tomosynthesis leveraging ensemble cross-validation models with multi-depth levels", Proc. SPIE 12033, Medical Imaging 2022: Computer-Aided Diagnosis, 120330D (4 April 2022); https://doi.org/10.1117/12.2611007

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