GANDALF: Graph-based transformer and Data Augmentation Active Learning Framework with interpretable features for multi-label chest Xray classification.

Details

Serval ID
serval:BIB_793ED796E69F
Type
Article: article from journal or magazin.
Collection
Publications
Institution
Title
GANDALF: Graph-based transformer and Data Augmentation Active Learning Framework with interpretable features for multi-label chest Xray classification.
Journal
Medical image analysis
Author(s)
Mahapatra D., Bozorgtabar B., Ge Z., Reyes M.
ISSN
1361-8423 (Electronic)
ISSN-L
1361-8415
Publication state
Published
Issued date
04/2024
Peer-reviewed
Oui
Volume
93
Pages
103075
Language
english
Notes
Publication types: Journal Article
Publication Status: ppublish
Abstract
Informative sample selection in an active learning (AL) setting helps a machine learning system attain optimum performance with minimum labeled samples, thus reducing annotation costs and boosting performance of computer-aided diagnosis systems in the presence of limited labeled data. Another effective technique to enlarge datasets in a small labeled data regime is data augmentation. An intuitive active learning approach thus consists of combining informative sample selection and data augmentation to leverage their respective advantages and improve the performance of AL systems. In this paper, we propose a novel approach called GANDALF (Graph-based TrANsformer and Data Augmentation Active Learning Framework) to combine sample selection and data augmentation in a multi-label setting. Conventional sample selection approaches in AL have mostly focused on the single-label setting where a sample has only one disease label. These approaches do not perform optimally when a sample can have multiple disease labels (e.g., in chest X-ray images). We improve upon state-of-the-art multi-label active learning techniques by representing disease labels as graph nodes and use graph attention transformers (GAT) to learn more effective inter-label relationships. We identify the most informative samples by aggregating GAT representations. Subsequently, we generate transformations of these informative samples by sampling from a learned latent space. From these generated samples, we identify informative samples via a novel multi-label informativeness score, which beyond the state of the art, ensures that (i) generated samples are not redundant with respect to the training data and (ii) make important contributions to the training stage. We apply our method to two public chest X-ray datasets, as well as breast, dermatology, retina and kidney tissue microscopy MedMNIST datasets, and report improved results over state-of-the-art multi-label AL techniques in terms of model performance, learning rates, and robustness.
Keywords
Humans, X-Rays, Radiography, Thorax, Breast, Diagnosis, Computer-Assisted, Active learning, Data augmentation, Informative samples, Multi-label
Pubmed
Web of science
Create date
12/01/2024 12:20
Last modification date
12/03/2024 8:08
Usage data