Semi-supervised medical image classification via distance correlation minimization and graph attention regularization

Onderzoeksoutput: Articlepeer review

Samenvatting

We propose a novel semi-supervised learning method to leverage unlabeled data alongside minimal annotated data and improve medical imaging classification performance in realistic scenarios with limited labeling budgets to afford data annotations. Our method introduces distance correlation to minimize correlations between feature representations from different views of the same image encoded with non-coupled deep neural networks architectures. In addition, it incorporates a data-driven graph-attention based regularization strategy to model affinities among images within the unlabeled data by exploiting their inherent relational information in the feature space. We conduct extensive experiments on four medical imaging benchmark data sets involving X-ray, dermoscopic, magnetic resonance, and computer tomography imaging on single and multi-label medical imaging classification scenarios. Our experiments demonstrate the effectiveness of our method in achieving very competitive performance and outperforming several state-of-the-art semi-supervised learning methods. Furthermore, they confirm the suitability of distance correlation as a versatile dependence measure and the benefits of the proposed graph-attention based regularization for semi-supervised learning in medical imaging analysis.
Originele taal-2English
Artikelnummer103107
Pagina's (van-tot)1-16
Aantal pagina's16
TijdschriftMedical Image Analysis
Volume94
DOI's
StatusPublished - mei 2024

Bibliografische nota

Publisher Copyright:
© 2024

Vingerafdruk

Duik in de onderzoeksthema's van 'Semi-supervised medical image classification via distance correlation minimization and graph attention regularization'. Samen vormen ze een unieke vingerafdruk.

Citeer dit