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...

Full description

Saved in:
Bibliographic Details
Published in:2022 IEEE 2nd International Conference on Data Science and Computer Application (ICDSCA) pp. 348 - 351
Main Authors: Zhang, Ce, Cui, Qingshan, Ren, Shaolong
Format: Conference Proceeding
Language:English
Published: IEEE 28.10.2022
Subjects:
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
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary: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