CONFIDENCE CLASSIFIERS FOR DIAGNOSTIC TRAINING DATA
Automated assignment of confidence levels to medical diagnoses in a machine learning training data with respect to annotations made upon review of medical records such as x-ray films and test results. Confidence levels support machine learning for computer-aided diagnostic activity.
1 . A computer-implemented method for assigning confidence levels to training data for machine learning models comprising:
receiving a set of medical records for a medical exam including a corresponding diagnostic annotation for each medical record;
identifying a set of diagnostic activity data associated with a first diagnostic annotation of a first medical record;
determining a first confidence level for the first diagnostic annotation with reference to the diagnostic activity data; and
generating a set of training data from the set of medical records with diagnostic annotations and corresponding confidence level assignments including the first medical record, the first diagnostic annotation, and the first confidence level.
2 . The computer-implemented method of claim 1 further comprising:
weighting the diagnostic annotations of the medical diagnosis according to the determined confidence levels; and
wherein:
the set of training data includes the weighted diagnostic annotations.
3 . The computer-implemented method of claim 1 further comprising:
training a convolutional neural network with the weighted diagnoses as part of the modified training data.
4 . The computer-implemented method of claim 1 , wherein:
the set of medical records includes images generated during the medical exam; and
the set of diagnostic activity data for a medical record includes how long an annotator viewed an image of the medical record when annotating the image.
5 . The computer-implemented method of claim 1 , wherein:
the medical exam is a breast cancer screening;
the image is a mammogram; and
the annotation is a member selected from the group consisting of:
a) no finding,
b) benign,
c) malignant, and
d) suspicious.
6 . The computer-implemented method of claim 1 further comprising:
using the set of training data to train CAD (computer-aided diagnosis) algorithms to:
use a training loss function to weight examples by confidence, giving a higher weight to classification errors on cases with high confidence scores and vice versa;
adapt an adjusted learning schedule by scheduling batches and adjusting batch size so that the algorithm starts off by learning the high confidence cases and then proceed to the more difficult, low confidence cases;
train the algorithm without confidence adjustment; and
adjusting an output of the algorithm based on the confidence estimates.
7 . A computer program product comprising:
a set of storage device(s); and
computer code stored collectively in the set of storage device(s), with the computer code including data and instructions to cause a processor(s) set to perform at least the following operations:
receiving a set of medical records for a medical exam including a corresponding diagnostic annotation for each medical record,
identifying a set of diagnostic activity data associated with a first diagnostic annotation of a first medical record,
determining a first confidence level for the first diagnostic annotation with reference to the diagnostic activity data, and
generating a set of training data from the set of medical records with diagnostic annotations and corresponding confidence level assignments including the first medical record, the first diagnostic annotation, and the first confidence level.
8 . The computer program product of claim 7 wherein the computer code further includes instructions for causing the processor(s) set to perform the following operation(s):
weighting the diagnostic annotations of the medical diagnosis according to the determined confidence levels; and
wherein:
the set of training data includes the weighted diagnostic annotations.
9 . The computer program product of claim 7 further comprising:
training a convolutional neural network with the weighted diagnoses as part of the modified training data.
10 . The computer program product of claim 7 , wherein:
the set of medical records includes images generated during the medical exam; and
the set of diagnostic activity data for a medical record includes how long an annotator viewed an image of the medical record when annotating the image.
11 . The computer program product of claim 7 , wherein:
the medical exam is a breast cancer screening;
the image is a mammogram; and
the annotation is a member selected from the group consisting of:
a) no finding,
b) benign,
c) malignant, and
d) suspicious.
12 . The computer program product of claim 7 wherein the computer code further includes instructions for causing the processor(s) set to perform the following operation(s):
using the set of training data to train CAD (computer-aided diagnosis) algorithms to:
use a training loss function to weight examples by confidence, giving a higher weight to classification errors on cases with high confidence scores and vice versa;
adapt an adjusted learning schedule by scheduling batches and adjusting batch size so that the algorithm starts off by learning the high confidence cases and then proceed to the more difficult, low confidence cases;
train the algorithm without confidence adjustment; and
adjusting an output of the algorithm based on the confidence estimates.
13 . A computer system comprising:
a processor(s) set;
a set of storage device(s); and
computer code stored collectively in the set of storage device(s), with the computer code including data and instructions to cause the processor(s) set to perform at least the following operations:
receiving a set of medical records for a medical exam including a corresponding diagnostic annotation for each medical record,
identifying a set of diagnostic activity data associated with a first diagnostic annotation of a first medical record,
determining a first confidence level for the first diagnostic annotation with reference to the diagnostic activity data, and
generating a set of training data from the set of medical records with diagnostic annotations and corresponding confidence level assignments including the first medical record, the first diagnostic annotation, and the first confidence level.
14 . The computer system of claim 13 wherein the computer code further includes instructions for causing the processor(s) set to perform the following operation(s):
weighting the diagnostic annotations of the medical diagnosis according to the determined confidence levels; and
wherein:
the set of training data includes the weighted diagnostic annotations.
15 . The computer system of claim 13 further comprising:
training a convolutional neural network with the weighted diagnoses as part of the modified training data.
16 . The computer system of claim 13 , wherein:
the set of medical records includes images generated during the medical exam; and
the set of diagnostic activity data for a medical record includes how long an annotator viewed an image of the medical record when annotating the image.
17 . The computer system of claim 13 , wherein:
the medical exam is a breast cancer screening;
the image is a mammogram; and
the annotation is a member selected from the group consisting of:
a) no finding,
b) benign,
c) malignant, and
d) suspicious.
18 . The computer system of claim 13 wherein the computer code further includes instructions for causing the processor(s) set to perform the following operation(s):
using the set of training data to train CAD (computer-aided diagnosis) algorithms to:
use a training loss function to weight examples by confidence, giving a higher weight to classification errors on cases with high confidence scores and vice versa;
adapt an adjusted learning schedule by scheduling batches and adjusting batch size so that the algorithm starts off by learning the high confidence cases and then proceed to the more difficult, low confidence cases;
train the algorithm without confidence adjustment; and
adjusting an output of the algorithm based on the confidence estimates.