Model and Feature Diversity for Bayesian Neural Networks in Mutual Learning
Van Cuong Pham, Cuong C. Nguyen, Trung Le, Dinh Phung, Gustavo Carneiro, Thanh-Toan Do
Abstract
Bayesian Neural Networks (BNNs) offer probability distributions for model parameters, enabling uncertainty quantification in predictions. However, they often underperform compared to deterministic neural networks. Utilizing mutual learning can effectively enhance the performance of peer BNNs. In this paper, we propose a novel approach to improve BNNs performance through deep mutual learning. The proposed approaches aim to increase diversity in both network parameter distributions and feature distributions, promoting peer networks to acquire distinct features that capture different characteristics of the input, which enhances the effectiveness of mutual learning. Experimental results demonstrate significant improvements in the classification accuracy, negative log-likelihood, and expected calibration error when compared to traditional mutual learning for BNNs.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Flat Seeking Bayesian Neural NetworksVan-Anh Nguyen, Tung-Long Vuong, Hoang Phan, Thanh-Toan Do et al.NeurIPS 2023 · 14 citations
- Optimal Transport Model Distributional RobustnessVan-Anh Nguyen, Trung Le, Anh Tuan Bui, Thanh-Toan Do et al.NeurIPS 2023 · 8 citations
Builds on5
- A Comprehensive Overhaul of Feature DistillationByeongho Heo, Jeesoo Kim, Sangdoo Yun, Hyojin Park et al.ICCV 2019 · 727 citations
- Feature-map-level Online Adversarial Knowledge DistillationInseop Chung, Seonguk Park, Jangho Kim, Nojun KwakICML 2020 · 151 citations
- The k-tied Normal Distribution: A Compact Parameterization of Gaussian Mean Field Posteriors in Bayesian Neural NetworksJakub Swiatkowski, Kevin Roth, Bastiaan S. Veeling, Linh Tran et al.ICML 2020 · 52 citations
- Walsh-Hadamard Variational Inference for Bayesian Deep LearningSimone Rossi, Sébastien Marmin, Maurizio FilipponeNeurIPS 2020 · 17 citations
- Online Knowledge Distillation via Collaborative LearningQiushan Guo, Xinjiang Wang, Yichao Wu, Zhipeng Yu et al.CVPR 2020
Related papers
- Sampling-Free Learning of Bayesian Quantized Neural NetworksJiahao Su, Milan Cvitkovic, Furong HuangICLR 2020 · 7 citations
- Improving Accuracy and Calibration via Differentiated Deep Mutual LearningHan Liu, Peng Cui, Bingning Wang, Weipeng Chen et al.CVPR 2025
- Credal Wrapper of Model Averaging for Uncertainty Estimation in ClassificationKaizheng Wang, Fabio Cuzzolin, Keivan Shariatmadar, David Moens et al.ICLR 2025
- Bayesian Deep Learning and a Probabilistic Perspective of GeneralizationAndrew Gordon Wilson, Pavel IzmailovNeurIPS 2020 · 845 citations
- DIBS: Diversity Inducing Information Bottleneck in Model EnsemblesSamarth Sinha, Homanga Bharadhwaj, Anirudh Goyal, Hugo Larochelle et al.AAAI 2021 · 43 citations
