Back

Bionnica: A Deep Neural Network Architecture for Colorectal Polyps' Premalignancy Risk Evaluation

Babuc, D.; Ivascu, T.; Ardelean, M.; Onchis, D.

2024-06-20 health informatics
10.1101/2024.06.19.24309153 medRxiv
Show abstract

The third most prevalent cancer nowadays is colorectal cancer. Colonoscopy is an important procedure in the stage of detection of polyps malignancy because it helps in early identification and establishes effective therapy. This paper explores specific deep-learning architectures for the binary classification of colorectal polyps and considers the evaluation of their premalignancy risk. The main scope is to create a custom-based deep learning architecture that classifies adenomatous, hyperplastic, and serrated polyps samples into benign and premalignant based on images from the colonoscopic dataset. Each images output is modified through masked autoencoders which enhance the classification performance of the proposed model, called Bionnica. From the four evaluated state-of-the-art deep learning models (ZF NET, VGG-16, AlexNet, and ResNet-50), our experiments showed that ResNet-50 and ZF NET are most accurate (above 84%), with ResNet-50 excelling at indicating patients with premalignant colorectal polyps (above 92%). ZF NET is the fastest at handling 700 images. Our proposed deep learning model, Bionnica, is more performant than ZF NET and provides an efficient classification of colorectal polyps given its simple structure. The advantage of our model comes from the custom enhancement interpretability with a rule-based layer that guides the learning process and supports medical personnel in their decisions.

Published in Multimedia Tools and Applications · not in our set (fewer than 10 published preprints to learn from) · training set

Matching journals

The top 6 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.