Saved in:
Bibliographic Details
Main Authors: Du, Yuexi, Hooley, Regina J., Lewin, John, Dvornek, Nicha C.
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2403.13148
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929282627928064
author Du, Yuexi
Hooley, Regina J.
Lewin, John
Dvornek, Nicha C.
author_facet Du, Yuexi
Hooley, Regina J.
Lewin, John
Dvornek, Nicha C.
contents Digital Breast Tomosynthesis (DBT) is a widely used medical imaging modality for breast cancer screening and diagnosis, offering higher spatial resolution and greater detail through its 3D-like breast volume imaging capability. However, the increased data volume also introduces pronounced data imbalance challenges, where only a small fraction of the volume contains suspicious tissue. This further exacerbates the data imbalance due to the case-level distribution in real-world data and leads to learning a trivial classification model that only predicts the majority class. To address this, we propose a novel method using view-level contrastive Self-supervised Initialization and Fine-Tuning for identifying abnormal DBT images, namely SIFT-DBT. We further introduce a patch-level multi-instance learning method to preserve spatial resolution. The proposed method achieves 92.69% volume-wise AUC on an evaluation of 970 unique studies.
format Preprint
id arxiv_https___arxiv_org_abs_2403_13148
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SIFT-DBT: Self-supervised Initialization and Fine-Tuning for Imbalanced Digital Breast Tomosynthesis Image Classification
Du, Yuexi
Hooley, Regina J.
Lewin, John
Dvornek, Nicha C.
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
Digital Breast Tomosynthesis (DBT) is a widely used medical imaging modality for breast cancer screening and diagnosis, offering higher spatial resolution and greater detail through its 3D-like breast volume imaging capability. However, the increased data volume also introduces pronounced data imbalance challenges, where only a small fraction of the volume contains suspicious tissue. This further exacerbates the data imbalance due to the case-level distribution in real-world data and leads to learning a trivial classification model that only predicts the majority class. To address this, we propose a novel method using view-level contrastive Self-supervised Initialization and Fine-Tuning for identifying abnormal DBT images, namely SIFT-DBT. We further introduce a patch-level multi-instance learning method to preserve spatial resolution. The proposed method achieves 92.69% volume-wise AUC on an evaluation of 970 unique studies.
title SIFT-DBT: Self-supervised Initialization and Fine-Tuning for Imbalanced Digital Breast Tomosynthesis Image Classification
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2403.13148