Deep learning for real-time colon polyp detection
A set of enhancements to further improve the performance of deep learning artificial intelligence algorithms trained to detect and localize colon polyps. The enhancements spanning training data mining efficiencies and automation, training data augmentation, early detection of polyps enable a more performant colon polyp detection solution for use on colonoscopy procedure recordings or live procedures in endoscopy centers.
1. A method of enhancing training of a colon polyp detector using a deep learning-based object detector, comprising:
gathering a polyp early appearance image training dataset that includes a plurality of video clips each including an initial image of a polyp that is just starting to appear in frame;
sampling the plurality of video clips in the initial portion including the initial image of the polyp at a first sampling rate to generate polyp early appearance image training data;
sampling a remainder portion of the plurality of video clips at a second sampling rate, the second sampling rate being slower than the first sampling rate, to generate polyp image training data; and
training the colon polyp detector with both the polyp early appearance image training data and with the polyp image training data.
2. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 1 , further comprising:
sampling the remainder portion of each of the plurality of video clips after the initial region at a lower image sampling rate than for the initial region.
3. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 1 , wherein
the initial portion has a predetermined fixed length of time.
4. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 1 , wherein:
the colon polyp detector training is further enhanced by additionally training the colon polyp detector with a complementary transformed image dataset including a transformation of the polyp early appearance image training data, in addition to training with the polyp early appearance image training data and the polyp image training data;
whereby accuracy of polyp detection is increased at time of inference by the addition of transformed images in the training.
5. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 4 , wherein:
the transformation of the polyp early appearance image training data is at least one of shifting and rotating an image within the polyp early appearance image training data.
6. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 4 , wherein:
the transformation of the polyp early appearance image training data is generated by shifting an image within the polyp image training data.
7. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 4 , wherein:
the transformation of the polyp early appearance image training data is generated by rotating an image within the polyp image training data.
8. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 7 , wherein:
the transformation rotates the image at a random angle.
9. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 4 , wherein:
the transformation of the polyp early appearance image training data is generated by altering at least one of hue, saturation, and exposure within a limited range.
10. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 1 , further comprising:
training the colon polyp detector to minimize a difference in time in the plurality of video clips between when the polyp first appears in frame, and when the colon polyp detector first detects that polyp.
11. The method of enhancing training of a colon polyp detector using a deep learning-based object detector according to claim 1 , further comprising:
further training the colon polyp detector with images of missed polyp detection.