AM-RP Stacking PILers: Random projection stacking pseudoinverse learning algorithm based on attention mechanism

被引:1
|
作者
Cai, Zhenjiao [1 ]
Zhang, Sulan [1 ]
Guo, Ping [2 ]
Zhang, Jifu [1 ]
Hu, Lihua [1 ]
机构
[1] Taiyuan Univ Sci & Technol, Sch Comp Sci & Technol, Taiyuan 030024, Peoples R China
[2] Beijing Normal Univ, Sch Syst Sci, Beijing 100875, Peoples R China
来源
VISUAL COMPUTER | 2024年 / 40卷 / 01期
关键词
Stacking pseudoinverse learner; Random projection; Attention mechanism; CLASSIFICATION; REGRESSION; NETWORKS;
D O I
10.1007/s00371-023-02780-7
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
The stacking pseudoinverse learning algorithm can effectively improve the classification accuracy of the model and reduce the training time. However, the effect of random projection blocks on the neural network is often ignored, which reduces the performance of stacked generalization. To improve the generalization performance of neural networks and obtain high-quality classification results, we propose a random projection stacking pseudoinverse learning algorithm based on attention mechanism named AM-RP stacking PILers. Firstly, the input weight matrices with specific distributions are randomly generated, and then different random projection blocks are obtained through a pseudoinverse learning algorithm. Secondly, the training results of different random projection blocks are taken as a new input dataset, and the idea of attention mechanism is introduced to assign different weights to different random projection blocks. Finally, we use the stacking pseudoinverse learning algorithm to train a single hidden layer neural network and obtain the classification results with high generalization performance. The experimental results on a total of 76788 images in three public datasets of Salinas, MNIST and COIL-20 show that our algorithm achieves better performance in accuracy, precision, recall, F1 score and training time.
引用
收藏
页码:273 / 285
页数:13
相关论文
共 50 条
  • [41] Remaining Useful Life Prediction of High-Speed Railroad Contact Network Based on Stacking Integrated Attention-LSTM-CNN Deep Learning
    Qu, Zhijian
    Zhu, Lin
    Ma, Shuaijun
    Zhang, Boyu
    ARABIAN JOURNAL FOR SCIENCE AND ENGINEERING, 2024, 49 (12) : 16137 - 16154
  • [42] Transformer-BLS: An efficient learning algorithm based on multi-head attention mechanism and incremental learning algorithms
    Fu, Rongrong
    Liang, Haifeng
    Wang, Shiwei
    Jia, Chengcheng
    Sun, Guangbin
    Gao, Tengfei
    Chen, Dan
    Wang, Yaodong
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 238
  • [43] DR-IIXRN : Detection Algorithm of Diabetic Retinopathy Based on Deep Ensemble Learning and Attention Mechanism
    Ai, Zhuang
    Huang, Xuan
    Fan, Yuan
    Feng, Jing
    Zeng, Fanxin
    Lu, Yaping
    FRONTIERS IN NEUROINFORMATICS, 2021, 15
  • [44] Underwater Target Detection Utilizing Polarization Image Fusion Algorithm Based on Unsupervised Learning and Attention Mechanism
    Cheng, Haoyuan
    Zhang, Deqing
    Zhu, Jinchi
    Yu, Hao
    Chu, Jinkui
    SENSORS, 2023, 23 (12)
  • [45] Method Based on Stacking-Attention to Find Decomposition Indicators of Discharge Mechanism in C4F7N-CO2-O2 Gas
    Wang, Cong
    Cui, Guangkai
    Gao, Xuke
    Chen, Geng
    Tu, Youping
    Zheng, Zhong
    Jin, Hua
    Yang, Yuan
    IEEE TRANSACTIONS ON DIELECTRICS AND ELECTRICAL INSULATION, 2024, 31 (06) : 3110 - 3119
  • [46] A hybrid feature weighted attention based deep learning approach for an intrusion detection system using the random forest algorithm
    Hashmi, Arshad
    Barukab, Omar M.
    Osman, Ahmad Hamza
    PLOS ONE, 2024, 19 (05):
  • [47] Battery state-of-health estimation based on random charge curve fitting and broad learning system with attention mechanism
    Dai, Houde
    Huang, Yiyang
    Zhu, Liqi
    Lin, Haijun
    Yu, Hui
    Lai, Yuan
    Yang, Yuxiang
    JOURNAL OF POWER SOURCES, 2025, 636
  • [48] Data-Aware Adaptive Pruning Model Compression Algorithm Based on a Group Attention Mechanism and Reinforcement Learning
    Yang, Zhi
    Zhai, Yuan
    Xiang, Yi
    Wu, Jianquan
    Shi, Jinliang
    Wu, Ying
    IEEE ACCESS, 2022, 10 : 82396 - 82406
  • [49] An explainable deep learning-based algorithm with an attention mechanism for predicting the live birth potential of mouse embryos
    Tokuoka, Yuta
    Yamada, Takahiro G.
    Mashiko, Daisuke
    Ikeda, Zenki
    Kobayashi, Tetsuya J.
    Yamagata, Kazuo
    Funahashi, Akira
    ARTIFICIAL INTELLIGENCE IN MEDICINE, 2022, 134
  • [50] A Dual Arrhythmia Classification Algorithm Based on Deep Learning and Attention Mechanism Incorporating Morphological-temporal Information
    Shan, Chudong
    Yuan, Zhiyong
    Qiu, Zekang
    He, Ziyang
    An, Panfeng
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,