Few-shot Medical Image Classification with MAML Based on Dice Loss

In medical image processing based on deep learning, medical image datasets are usually few-shot and unbalanced between classes. This will cause the model to overfit and be biased towards the class with more samples during training and the accuracy of the class with fewer samples will decrease. As a...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:2022 IEEE 2nd International Conference on Data Science and Computer Application (ICDSCA) s. 348 - 351
Hlavní autoři: Zhang, Ce, Cui, Qingshan, Ren, Shaolong
Médium: Konferenční příspěvek
Jazyk:angličtina
Vydáno: IEEE 28.10.2022
Témata:
Biomedical image processing
Costs
Deep learning
dice loss
Entropy
few-shot learning
imbalanced learning
Linear programming
Machine learning
model-agnostic meta-learning
Neural networks
Training
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:In medical image processing based on deep learning, medical image datasets are usually few-shot and unbalanced between classes. This will cause the model to overfit and be biased towards the class with more samples during training and the accuracy of the class with fewer samples will decrease. As a result, the overall accuracy will also decrease. To address these issues, the model-agnostic meta-learning based on dice loss is proposed. The algorithm avoids the need for a large amount of data to be driven and can quickly adapt to new tasks with only a small amount of labeled data. Meanwhile, the objective function can be optimized autonomously in the direction of a small number of classes during training. Finally, good results are achieved on two few-shot datasets of medical images. Our method can provide a solution to achieve good results with less data cost in medical image processing.
DOI:10.1109/ICDSCA56264.2022.9988390