Skip to content

samleoqh/machine-ln

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

30 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep learning for automatic polyp detection


Abstract


Polyps are known as possible colorectal cancer precursors, and their early detection is of great importance, but highly challenging from an image processing stand- point. In this work, we evaluate several state-of-the-art machine learning techniques and deep learning methods in the medical image processing domain and research solutions about how they can be more efficiently utilized for automatic detection of polyps in endoscopy and colonoscopy images.

This work proposes an effective transfer learning (TL) framework relying on pre- trained DCNNs using a large collection of natural ImageNet images. This has been achieved by evaluating various kinds of cutting edge techniques including both traditional machine learning methods by training feature-based classifiers from scratch and modern DCNNs algorithms with (TL) and fine tuning pre-trained models. We transfer learned ImageNet weights as initial weights, and then fine-tune this model combined with a new deep classifier called fully connected networks (FCNs) with data augmentation and patch-extraction of colonoscopy images to automatically detect polyps. In case of insufficient colonoscopy images, patch-based data augmentation and deep features extracted using TL strategy can provide sufficient and balanced classification in- formation.

With the proposed TL framework with our optimized hyper-parameters, the sys- tem achieved overall 96.00% polyp detection precision and sensitivity, which outperformed the traditional machine learning classification methods in each defined performance metric. Moreover, the TL framework proposed is scalable and flexible so that it can easily be extended to include other types of disease detection in the future and also be able to integrate one more DCNNs model to boost its generalizing capabilities.

About

some old stuff

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published