College of Information Engineering, Henan University of Science and Technology, Luoyang 471000, China.
College of Vehicle and Traffic Engineering, Henan University of Science and Technology, Luoyang 471000, China.
Comput Intell Neurosci. 2022 Jul 18;2022:1582624. doi: 10.1155/2022/1582624. eCollection 2022.
As a single-layer feedforward network (SLFN), extreme learning machine (ELM) has been successfully applied for classification and regression in machine learning due to its faster training speed and better generalization. However, it will perform poorly for domain adaptation in which the distributions between training data and testing data are inconsistent. In this article, we propose a novel ELM called two-stage transfer extreme learning machine (TSTELM) to solve this problem. At the statistical matching stage, we adopt maximum mean discrepancy (MMD) to narrow the distribution difference of the output layer between domains. In addition, at the subspace alignment stage, we align the source and target model parameters, design target cross-domain mean approximation, and add the output weight approximation to further promote the knowledge transferring across domains. Moreover, the prediction of test sample is jointly determined by the ELM parameters generated at the two stages. Finally, we investigate the proposed approach in classification task and conduct experiments on four public domain adaptation datasets. The result indicates that TSTELM could effectively enhance the knowledge transfer ability of ELM with higher accuracy than other existing transfer and non-transfer classifiers.
作为一种单层前馈神经网络(SLFN),极限学习机(ELM)由于其更快的训练速度和更好的泛化能力,已成功应用于机器学习中的分类和回归。然而,对于训练数据和测试数据分布不一致的领域自适应问题,它的表现会很差。在本文中,我们提出了一种新的极限学习机,称为两阶段迁移极限学习机(TSTELM),以解决这个问题。在统计匹配阶段,我们采用最大均值差异(MMD)来缩小输出层在不同领域之间的分布差异。此外,在子空间对齐阶段,我们对齐源和目标模型参数,设计目标跨域均值逼近,并添加输出权重逼近,以进一步促进跨域的知识转移。此外,测试样本的预测由两个阶段生成的 ELM 参数共同决定。最后,我们在分类任务中研究了所提出的方法,并在四个公共领域自适应数据集上进行了实验。结果表明,TSTELM 可以有效地增强 ELM 的知识转移能力,比其他现有的迁移和非迁移分类器具有更高的准确性。