ALFREDO: Active Learning with FeatuRe disEntangelement and DOmain adaptation for medical image classification.

Details

Serval ID
serval:BIB_08F4A39D35A2
Type
Article: article from journal or magazin.
Collection
Publications
Institution
Title
ALFREDO: Active Learning with FeatuRe disEntangelement and DOmain adaptation for medical image classification.
Journal
Medical image analysis
Author(s)
Mahapatra D., Tennakoon R., George Y., Roy S., Bozorgtabar B., Ge Z., Reyes M.
ISSN
1361-8423 (Electronic)
ISSN-L
1361-8415
Publication state
In Press
Peer-reviewed
Oui
Language
english
Notes
Publication types: Journal Article
Publication Status: aheadofprint
Abstract
State-of-the-art deep learning models often fail to generalize in the presence of distribution shifts between training (source) data and test (target) data. Domain adaptation methods are designed to address this issue using labeled samples (supervised domain adaptation) or unlabeled samples (unsupervised domain adaptation). Active learning is a method to select informative samples to obtain maximum performance from minimum annotations. Selecting informative target domain samples can improve model performance and robustness, and reduce data demands. This paper proposes a novel pipeline called ALFREDO (Active Learning with FeatuRe disEntangelement and DOmain adaptation) that performs active learning under domain shift. We propose a novel feature disentanglement approach to decompose image features into domain specific and task specific components. Domain specific components refer to those features that provide source specific information, e.g., scanners, vendors or hospitals. Task specific components are discriminative features for classification, segmentation or other tasks. Thereafter we define multiple novel cost functions that identify informative samples under domain shift. We test our proposed method for medical image classification using one histopathology dataset and two chest X-ray datasets. Experiments show our method achieves state-of-the-art results compared to other domain adaptation methods, as well as state of the art active domain adaptation methods.
Keywords
Active learning, Domain adaptation, Feature disentanglement, Histopathology, X-ray
Pubmed
Web of science
Create date
19/07/2024 11:25
Last modification date
13/08/2024 7:48
Usage data