The Effect of Class Distribution on Multi-Modal Medical Images Classification in Meta-Learning
2025 (English)In: Lect. Notes Comput. Sci., Springer Nature , 2025, p. 102-111Conference paper, Published paper (Refereed)
Abstract [en]
Medical images are available in small datasets with different modalities and for various organs. Although Transfer learning is a promising approach for training models on small datasets, further studies are required on using them with several image modalities and body parts. This paper explores two dominant meta-learning algorithms: a metric-based algorithm, namely Prototypical Network, and an optimization-based algorithm, namely MAML. The algorithms trained on a small multi-modal medical dataset (i.e., Slake), and a general dataset of the same size (i.e., Tiny-imagenet) with different class distribution methods: Random, Logical-based, and Statistical-based class distribution. The aim is to apply diversity among the classes in the meta-training set and similarity between the classes in the meta-training, and meta-testing or meta-validation sets. The results validate the importance of class distribution on the accuracy of the algorithms. MAML is hard to be trained on Tiny-imagenet but it shows good accuracy on Slake in specific cases. In statistical distribution, the Prototypical Network shows high accuracy on the datasets, especially when the similarity between the meta-training and meta-validation sets is considered for Tiny-imagenet.
Place, publisher, year, edition, pages
Springer Nature , 2025. p. 102-111
Series
Lecture Notes in Computer Science, ISSN 0302-9743 ; 15444 LNCS
Keywords [en]
Class distribution, Medical images, Meta-learning, Adversarial machine learning, Federated learning, Class distributions, Medical image, Medical image classification, Meta-training, Metalearning, Multi-modal, Small data set, Training model, Transfer learning, Validation sets, Contrastive Learning
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:mdh:diva-70423DOI: 10.1007/978-3-031-80507-3_11ISI: 001677662400011Scopus ID: 2-s2.0-85219189276ISBN: 9783031805066 (print)OAI: oai:DiVA.org:mdh-70423DiVA, id: diva2:1943959
Conference
Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
2025-03-122025-03-122026-03-04Bibliographically approved