Memristor-based Bayesian spiking neural network for IBD diagnosis

Abstract

Traditional deep learning methods, such as deep neural networks, have achieved remarkable success in image processing, natural language processing, graph computation, and other fields. However, in the diagnosis of Inflammatory Bowel Disease (IBD), conventional deep learning methods are highly susceptible to generating overconfident results, which can lead to great loss of life and property if it results in misdiagnosis of the patient. In this work, we propose a memristor-based IBD diagnostic system inspired by the idea of variational inference based on Bayes' law. The memristors in our system are used for both synaptic weight storage and entropy sources so that an efficient Bayesian inference system can be realized. Our memristor-based true random number generator can produce probability-tunable bitstreams required for Bayesian inference with a mean-square error that is reduced by 10.9 × than that of a conventional pseudo-random generator. The proposed system consists of a coarse-grained and a fined-grained spiking neural network, and an optimized strategy that trims the neuron's threshold is proposed for maintaining the appropriate model sparsity and reducing information loss. With Bayesian inference, our work achieves lower diagnostic error rates, which are reduced by 10.79 % and 0.49 % for the coarse-grained and fine-grained networks, respectively. The proposed system is easy to hardware implement and presents a novel solution for community-wide IBD diagnosis.