arrow
返回

Multi-Task Fusion for Improving Mammography Screening Data Classification

delete2022-04-01
delete10
delete
OA
AI
M
M Wimmer *
G
Gert Sluiter
D
David Major
D
D. Lenis
A
Astrid Berg
T
Theresa Neubauer
K
Katja Bühler
DOI:10.1109/TMI.2021.3129068delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Machine learning and deep learning methods have become essential for computer-assisted prediction in medicine, with a growing number of applications also in the field of mammography. Typically these algorithms are trained for a specific task, e.g., the classification of lesions or the prediction of a mammogram's pathology status. To obtain a comprehensive view of a patient, models which were all trained for the same task(s) are subsequently ensembled or combined. In this work, we propose a pipeline approach, where we first train a set of individual, task-specific models and subsequently investigate the fusion thereof, which is in contrast to the standard model ensembling strategy. We fuse model predictions and high-level features from deep learning models with hybrid patient models to build stronger predictors on patient level. To this end, we propose a multi-branch deep learning model which efficiently fuses features across different tasks and mammograms to obtain a comprehensive patient-level prediction. We train and evaluate our full pipeline on public mammography data, i.e., DDSM and its curated version CBIS-DDSM, and report an AUC score of 0.962 for predicting the presence of any lesion and 0.791 for predicting the presence of malignant lesions on patient level. Overall, our fusion approaches improve AUC scores significantly by up to 0.04 compared to standard model ensembling. Moreover, by providing not only global patient-level predictions but also task-specific model results that are related to radiological features, our pipeline aims to closely support the reading workflow of radiologists.
Keyword:
Cancer
Feature extraction
Lesions
Task analysis
Breast
Predictive models
Standards
Mammography
DDSM
CBIS-DDSM
deep learning
model fusion

期刊

IEEE Transactions on Medical Imaging 封面图
IEEE Transactions on Medical Imaging
IF:
9.8
论文数:
6.2K
被引数:
3.7W

机构

暂无机构信息
引用论文

引用论文

Degradation of reactive dyes I. A comparative study of ozonation, enzymatic and photochemical processes活性染料的降解I.臭氧、酶和光化学过程的比较研究
err1999-02-01
err0
PREAI
errPatricio Peralta-Zamora; Airton Kunz; Sandra Games de Moraes; Ronaldo Pelegrini; Patricia de Campos Moleiro; Juan Reyes; Nelson Duran
err分享
err收藏
Trp-dependent auxin biosynthesis in Arabidopsis: involvement of cytochrome P450s CYP79B2 and CYP79B3
err2002-12-01
err0
errOAAI
errYunde Zhao; Anna K. Hull; Neeru R. Gupta; Kendrick A. Goss; José Alonso; Joseph R. Ecker; Jennifer Normanly; Joanne Chory; John L. Celenza
err分享
err收藏
err分享
err收藏
Robust breast cancer detection in mammography and digital breast tomosynthesis using an annotation-efficient deep learning approach使用注释有效的深度学习方法在乳房x线照相术和数字乳房断层合成中进行稳健的乳腺癌检测
err2021-01-11
err202
errOAAI
errLotter, William; Diab, Abdul Rahman; Haslam, Bryan; Kim, Jiye G.; Grisot, Giorgia; Wu, Eric; Wu, Kevin; Onieva, Jorge Onieva; Boyer, Yun; Boxerman, Jerrold L.; Wang, Meiyun; Bandler, Mack; Vijayaraghavan, Gopal R.; Gregory Sorensen, A.
err分享
err收藏
学者 查看更多内容