Ensemble of Pruned Networks for Reliable Classifiers

Zhen Gao,Han Zhang,Xiaohui Wei,Jiajun Xiao,Shulin Zeng,Guangjun Ge,Yu Wang,Pedro Reviriego
DOI: https://doi.org/10.1109/AICAS51828.2021.9458545
2021-01-01
Abstract:Convolutional Neural Networks (CNNs) have been widely used for image recognition or natural language processing. When CNNs are used in safety-critical applications, their reliability becomes a priority and in particular their tolerance to soft errors. Unfortunately, traditional fault tolerant techniques based on modular redundancy introduce a large overhead, which is not acceptable in many resource limited embedded systems given the complexity of CNNs. To reduce the cost of protecting CNNs, the use of an ensemble of smaller CNNs has been recently proposed. The idea is that the ensemble cannot only improve the classification accuracy but also protect against errors. To reduce the cost, pruning the CNNs is widely used. In this paper, the combination of both ideas is explored, namely ensembles of CNNs that are built using different pruned versions of the same CNN are analyzed. To evaluate the effectiveness of the idea, VGG16 is used as a case study. Small networks with different sizes are generated by pruning the original VGG16, and a combiner was designed with the ability to detect a faulty base learner. The reliability of different ensembles of the pruned networks is evaluated in PyTorch on Cifar-10. The reliability evaluation results reveal that the ensemble of pruned networks could achieve higher reliability than the Triple Modular Redundancy (TMR) protected VGG16 networks with an overhead lower than double modular redundancy. Complementarity analysis between different pruned VGG nets has been performed to explain the reliability improvement.
What problem does this paper attempt to address?