A robust and lightweight deep attention multiple instance learning algorithm for predicting genetic alterations

Copyright © 2023 Elsevier Ltd. All rights reserved..

Self-attention mechanism-based algorithms are attractive in digital pathology due to their interpretability, but suffer from computation complexity. This paper presents a novel, lightweight Attention-based Multiple Instance Mutation Learning (AMIML) model to allow small-scale attention operations for predicting gene mutations. Compared to the standard self-attention model, AMIML reduces the number of model parameters by approximately 70%. Using data for 24 clinically relevant genes from four cancer cohorts in TCGA studies (UCEC, BRCA, GBM, and KIRC), we compare AMIML with a standard self-attention model, five other deep learning models, and four traditional machine learning models. The results show that AMIML has excellent robustness and outperforms all the baseline algorithms in the vast majority of the tested genes. Conversely, the performance of the reference deep learning and machine learning models vary across different genes, and produce suboptimal prediction for certain genes. Furthermore, with the flexible and interpretable attention-based pooling mechanism, AMIML can further zero in and detect predictive image patches.

Medienart:

E-Artikel

Erscheinungsjahr:

2023

Erschienen:

2023

Enthalten in:

Zur Gesamtaufnahme - volume:105

Enthalten in:

Computerized medical imaging and graphics : the official journal of the Computerized Medical Imaging Society - 105(2023) vom: 05. Apr., Seite 102189

Sprache:

Englisch

Beteiligte Personen:

Guo, Bangwei [VerfasserIn]
Li, Xingyu [VerfasserIn]
Yang, Miaomiao [VerfasserIn]
Zhang, Hong [VerfasserIn]
Xu, Xu Steven [VerfasserIn]

Links:

Volltext

Themen:

Attention mechanism
Deep learning
Gene Mutation
Journal Article
Mutiple Instance Learning
Research Support, Non-U.S. Gov't
Whole slide images

Anmerkungen:

Date Completed 06.03.2023

Date Revised 07.04.2023

published: Print-Electronic

Citation Status MEDLINE

doi:

10.1016/j.compmedimag.2023.102189

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM352497610