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

Détails

ID Serval
serval:BIB_793ED796E69F
Type
Article: article d'un périodique ou d'un magazine.
Collection
Publications
Institution
Titre
GANDALF: Graph-based transformer and Data Augmentation Active Learning Framework with interpretable features for multi-label chest Xray classification.
Périodique
Medical image analysis
Auteur⸱e⸱s
Mahapatra D., Bozorgtabar B., Ge Z., Reyes M.
ISSN
1361-8423 (Electronic)
ISSN-L
1361-8415
Statut éditorial
Publié
Date de publication
04/2024
Peer-reviewed
Oui
Volume
93
Pages
103075
Langue
anglais
Notes
Publication types: Journal Article
Publication Status: ppublish
Résumé
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.
Mots-clé
Humans, X-Rays, Radiography, Thorax, Breast, Diagnosis, Computer-Assisted, Active learning, Data augmentation, Informative samples, Multi-label
Pubmed
Web of science
Création de la notice
12/01/2024 12:20
Dernière modification de la notice
12/03/2024 8:08
Données d'usage