遥感影像场景识别的贝叶斯共轭批次归一化方法

Scene Recognition of Remotely Sensed Images Based on Bayes Adjoint Batch Normalization

  • 摘要: 归一化方法作为特征预处理的关键部分,在浅学习和深度学习中都是至关重要的。针对批次归一化方法中存在对批次样本容量依赖较大的问题,当前的优化思路主要是从样本信息的其他维度(如通道、层、时间等)来弥补批次样本容量较小的不足。从贝叶斯理论的角度出发,通过将总体信息、先验信息和样本信息以科学、合理的融合方式来弥补批次样本容量不足的缺陷,从而可以更加准确地估计样本均值和样本方差,使得归一化后的特征落入最佳的非饱和区域,以便更好地反映整个特征空间的原始表征,进而深度学习模型可以达到最佳的特征表达能力。实验与分析表明,所提的贝叶斯共轭批次归一化方法可行且有效,在NWPU-RESISC45数据集上,其分类精度比批次归一化方法高5.64%。得益于总体信息和先验信息,所提方法受样本容量的影响较小。

     

    Abstract:
    Objective Normalization method plays an important role in feature preprocessing phase not only in conventional machine learning domain but also in contemporary deep learning domain. Batch normalization (BN) is very successful, but its performance very depends on the sample size. Therefore, many researchers try to improve it when the sample size is inadequate through adding the sample size merely in the sample information space.
    Methods This paper utilizes Bayes theory to integrate general information, prior information and sample information, and to offset the inadequate sample information. In this way, the mean value and the variance of sample can be estimated more precisely and more robust especially when the sample size is small, and the normalized feature will better fall into non-saturating region of activation function, which enables deep learning model to better describe original feature space.
    Results The Top-1 test classification accuracy in NWPU-RESISC45 dataset is improved by 5.64% than that of BN method. Moreover, with the help of general information and prior information, the Bayes adjoint batch normalization (BABN) method is not sensitive to the sample size.
    Conclusions The experiment results show that the proposed BABN method is feasible and effective, and performs better than BN method and other variants in the scene recognition of remote sensing image.

     

/

返回文章
返回