Initialization Method of Batch Uniformization Auto Encoder by Principal Component Analysis
Initialization Method of Batch Uniformization Auto Encoder by Principal Component Analysis
复制标题
基于主成分分析的批量均匀化自动编码器初始化方法
DOI:
10.1109/ssci50451.2021.9660064
复制
发表时间:
2021
期刊:
影响因子:
--
通讯作者:
Shogo Takaoka; Takuya Kitamura; Aiga Suzuki; Masahiro Murakawa
中科院分区:
文献类型:
--
作者:
Shogo Takaoka; Takuya Kitamura; Aiga Suzuki; Masahiro Murakawa
A batch uniformization autoencoder (BU-AE) often performs poorly in anomaly detection problems if the initial weights of the network are determined by a random number-based method. This paper proposes a principal component initialization method with anomalous data (PCI-AD) for BU-AE, where the initial weights are obtained by principal component analysis; the PCI-AD is trained to reduce the reconstruction error of normal data and increase that of the anomalous data. Since these weights are equivalent in concept to the objective function of BU-AE, BU-AE with PCI-AD provides stable performance. To demonstrate the effectiveness of PCI-AD, the proposed method was compared through several computational experiments with He's initialization, which is a random number-based initialization method. In the experiment using CIFAR-10, it was confirmed that the use of PCI-AD improved the value of AUC by up to 17% compared to He's initialization.