UA-DRN: Unbiased Aggregation of Deep Neural Networks for Regression Ensemble

Jiawei Li,Yiming Li,Jiacheng Yang,Taian Guo,Shutao Xia
2019-01-01
Abstract:. Current research has managed to train multiple Deep Neural Networks (DNNs) in affordable computing time. Then, finding a practical method to aggregate these DNNs becomes a fundamental problem. To address this, we present an unbiased combination scheme to guide the aggregation of the diverse DNNs models, by leveraging the Negative Correlation Learning (NCL) and the James-Stein’s Estimator (JSE). The proposed framework, called the Unbiased Aggregation of Deep Regression Network (UA-DRN), contributes both in the training phase and the deploy phase. To fast diversify every individual DNNs, we improve the NCL technique with a cyclic annealing learning rate strategy, instead of training them from scratch. In addition, we adopt the JSE for model aggregation at the test time. Specifically, the James-Stein’s Estimator (JSE) is non-iteration and can make an unbiased estimate of regression ensemble, especially the input data corrupted by noise. We evaluate the proposed UA-DRN framework on several regression tasks and the image denoising benchmarks. The extensive experiments demonstrate that UA-DRN can consistently improve the regression performance in term of different quantitative metrics.
Computer Science
What problem does this paper attempt to address?