Neural network model training method and device, storage medium and equipment
A technology of neural network model and training method, applied in the direction of biological neural network model, neural learning method, etc., which can solve the problems of inaccurate noise label processing and limited improvement of model anti-interference effect, achieving high accuracy and good explainability performance, good anti-jamming effect
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0041] see figure 1 , which shows the training method of the neural network model in Embodiment 1 of the present invention, the method specifically includes steps S01-step S03:
[0042] Step S01, acquiring an original data set, and training an original neural network model according to the original data set.
[0043] Wherein, the original data set may be an image sample set, such as a medical image sample set. After the original data set is obtained, the neural network training can be performed based on the original data set to obtain the original neural network model. For example, the image segmentation neural network training can be performed on the image sample set to obtain the image segmentation network model. As another example, the original data set can also be a text data set, and the text data set is trained to obtain a semi-supervised multi-label learning model.
[0044] Step S02, identifying noise labels from the original neural network model.
[0045] During spe...
Embodiment 2
[0056] see figure 2 , shows the training method of the neural network model in the second embodiment of the present invention. The difference between the detection method in this embodiment and the detection method in the first embodiment is that the training method of the neural network model in this embodiment Also further include step S11-step S14:
[0057] Step S11, obtaining an original image sample set, and training an original image segmentation network model according to the original image sample set.
[0058] In this embodiment, the method adopts a teacher-student framework when training the neural network model, and the original image segmentation network model is a teacher model.
[0059] Step S12, using confidence learning technology to identify noise labels from the original image segmentation network model.
[0060] Specifically, step S12 specifically includes the following refinement steps:
[0061] Calculating the predicted probability of the original neura...
Embodiment 3
[0084] On the other hand, the present invention also proposes a training device for a neural network model, please refer to Figure 4 , which shows the training device of the neural network model provided by the third embodiment of the present invention, the device includes:
[0085] Data acquisition module 11, is used for obtaining original data set, and trains original neural network model according to described original data set;
[0086] Noise identification module 12, for identifying noise label from described original neural network model;
[0087] The model training module 13 is used to modify the original data corresponding to the noise label, and train a new neural network model according to the modified data set.
[0088] Wherein, the original data set may be an image sample set, such as a medical image sample set. After the original data set is obtained, the neural network training can be performed based on the original data set to obtain the original neural netwo...
PUM
Login to View More Abstract
Description
Claims
Application Information
Login to View More 


