Literature DB >> 33750847

DropConnect is effective in modeling uncertainty of Bayesian deep networks.

Aryan Mobiny1, Pengyu Yuan2, Supratik K Moulik3, Naveen Garg4, Carol C Wu4, Hien Van Nguyen2.   

Abstract

Deep neural networks (DNNs) have achieved state-of-the-art performance in many important domains, including medical diagnosis, security, and autonomous driving. In domains where safety is highly critical, an erroneous decision can result in serious consequences. While a perfect prediction accuracy is not always achievable, recent work on Bayesian deep networks shows that it is possible to know when DNNs are more likely to make mistakes. Knowing what DNNs do not know is desirable to increase the safety of deep learning technology in sensitive applications; Bayesian neural networks attempt to address this challenge. Traditional approaches are computationally intractable and do not scale well to large, complex neural network architectures. In this paper, we develop a theoretical framework to approximate Bayesian inference for DNNs by imposing a Bernoulli distribution on the model weights. This method called Monte Carlo DropConnect (MC-DropConnect) gives us a tool to represent the model uncertainty with little change in the overall model structure or computational cost. We extensively validate the proposed algorithm on multiple network architectures and datasets for classification and semantic segmentation tasks. We also propose new metrics to quantify uncertainty estimates. This enables an objective comparison between MC-DropConnect and prior approaches. Our empirical results demonstrate that the proposed framework yields significant improvement in both prediction accuracy and uncertainty estimation quality compared to the state of the art.

Entities:  

Year:  2021        PMID: 33750847      PMCID: PMC7943811          DOI: 10.1038/s41598-021-84854-x

Source DB:  PubMed          Journal:  Sci Rep        ISSN: 2045-2322            Impact factor:   4.379


  4 in total

1.  Bacterial species identification using MALDI-TOF mass spectrometry and machine learning techniques: A large-scale benchmarking study.

Authors:  Thomas Mortier; Anneleen D Wieme; Peter Vandamme; Willem Waegeman
Journal:  Comput Struct Biotechnol J       Date:  2021-11-09       Impact factor: 7.271

2.  Automatic brain segmentation in preterm infants with post-hemorrhagic hydrocephalus using 3D Bayesian U-Net.

Authors:  Axel Largent; Josepheen De Asis-Cruz; Kushal Kapse; Scott D Barnett; Jonathan Murnick; Sudeepta Basu; Nicole Andersen; Stephanie Norman; Nickie Andescavage; Catherine Limperopoulos
Journal:  Hum Brain Mapp       Date:  2022-01-13       Impact factor: 5.038

3.  An Efficient and Uncertainty-Aware Decision Support System for Disaster Response Using Aerial Imagery.

Authors:  Junchi Bin; Ran Zhang; Rui Wang; Yue Cao; Yufeng Zheng; Erik Blasch; Zheng Liu
Journal:  Sensors (Basel)       Date:  2022-09-21       Impact factor: 3.847

4.  Bayesian Fully Convolutional Networks for Brain Image Registration.

Authors:  Kunpeng Cui; Panpan Fu; Yinghao Li; Yusong Lin
Journal:  J Healthc Eng       Date:  2021-07-26       Impact factor: 2.682

  4 in total

北京卡尤迪生物科技股份有限公司 © 2022-2023.