Eye fundus image classification method and device based on multi-task curriculum type learning, equipment and medium
A fundus image and classification method technology, applied in neural learning methods, image analysis, image enhancement, etc., can solve problems such as unbalanced distribution, lack of interpretability, difficulty in overcoming training samples, etc., to increase the range of receptive fields and enhance features Encoding ability, effect of improving accuracy
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0066] Embodiment 1 provides a fundus image classification method based on multi-task curriculum learning, such as figure 1 shown, proceed as follows:
[0067] Step A, design a teacher network based on the self-attention mechanism; take the fundus image sample as input and the glaucoma classification label as output, and supervise the training of the teacher network;
[0068] 1) Design a teacher network based on the self-attention mechanism
[0069] The structure of the teacher network is as follows figure 2As shown, it includes in turn: the ResNet-34 backbone structure with the fully connected layer removed, the convolutional layer, the GC self-attention mechanism module, the global average pooling layer, and the fully connected layer. In the ResNet-34 backbone structure with the fully connected layer removed, a set of feature maps after each pooling layer is named as the 1st to 5th set of feature maps in sequence, and the 2nd to 5th set of feature maps are reduced to Sam...
Embodiment 2
[0120] This embodiment provides a fundus image classification device based on multi-task curriculum learning, including: a teacher network module and a multi-task student network module; wherein,
[0121] The structure of the teacher network module is designed based on a self-attention mechanism, and the fundus image sample is used as an input, and the glaucoma classification label is used as an output to perform supervised training, and after the training is completed, it is used to generate a label evidence map corresponding to each fundus image sample;
[0122] The structure of the multi-task student network module includes an evidence map prediction branch and a glaucoma prediction branch; and the loss function for training the multi-task student network is designed according to the sample prior weighting coefficient θ and the sample feedback loss coefficient α of the fundus image sample Obtained; among them, the sample prior weighting coefficient θ is designed according to...
Embodiment 3
[0127] This embodiment provides an electronic device, including a memory and a processor, where a computer program is stored in the memory, and when the computer program is executed by the processor, the processor implements the method described in Embodiment 1.
PUM
Abstract
Description
Claims
Application Information
- R&D Engineer
- R&D Manager
- IP Professional
- Industry Leading Data Capabilities
- Powerful AI technology
- Patent DNA Extraction
Browse by: Latest US Patents, China's latest patents, Technical Efficacy Thesaurus, Application Domain, Technology Topic, Popular Technical Reports.
© 2024 PatSnap. All rights reserved.Legal|Privacy policy|Modern Slavery Act Transparency Statement|Sitemap|About US| Contact US: help@patsnap.com