Representation Learning With Dual Autoencoder for Multi-Label Classification

被引:1
|
作者
Zhu, Yi [1 ,2 ,3 ]
Yang, Yang [1 ]
Li, Yun [1 ]
Qiang, Jipeng [1 ]
Yuan, Yunhao [1 ]
Zhang, Runmei [4 ]
机构
[1] Yangzhou Univ, Sch Informat Engn, Yangzhou 225009, Jiangsu, Peoples R China
[2] Hefei Univ Technol, Key Lab Knowledge Engn Big Data, Minist Educ, Hefei 230009, Peoples R China
[3] Hefei Univ Technol, Sch Comp Sci & Informat Engn, Hefei 230009, Peoples R China
[4] Anhui Jianzhu Univ, Sch Elect & Informat Engn, Hefei 230022, Peoples R China
基金
中国国家自然科学基金;
关键词
Training; Manifolds; Independent component analysis; Data models; Correlation; Linear programming; Licenses; Multi-label classification; dual autoencoder; RICA; manifold regularization; representation learning; NETWORK;
D O I
10.1109/ACCESS.2021.3096194
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Multi-label classification aims to deal with the problem that an object may be associated with one or more labels, which is a more difficult task due to the complex nature of multi-label data. The crucial problem of multi-label classification is the more robust and higher-level feature representation learning, which can reduce non-helpful feature attributes from the input space prior to training. In recent years, deep learning methods based on autoencoders have achieved excellent performance in multi-label classification for the advantages of powerful representations learning ability and fast convergence speed. However, most existing autoencoder-based methods only rely on the single autoencoder model, which pose challenges for multi-label feature representations learning and fail to measure similarities between data spaces. To address this problem, in this paper, we propose a novel representation learning method with dual autoencoder for multi-label classification. Compared to the existing autoencoder-based methods, our proposed method can capture different characteristics and more abstract features from data by the serially connection of two different types of autoencoders. More specifically, firstly, the algorithm of Reconstruction Independent Component Analysis (RICA) in sparse autoencoder is trained on patches on all training and test dataset for robust global feature representations learning. Secondly, with the output of RICA, stacked autoencoder with manifold regularization (SAMR) is introduced to ameliorate the quality of multi-label features learning. Comprehensive experiments on several real-world data sets demonstrate the effectiveness of our proposed approach compared with several competing state-of-the-art methods.
引用
收藏
页码:98939 / 98947
页数:9
相关论文
共 50 条
  • [1] Supervised representation learning for multi-label classification
    Ming Huang
    Fuzhen Zhuang
    Xiao Zhang
    Xiang Ao
    Zhengyu Niu
    Min-Ling Zhang
    Qing He
    [J]. Machine Learning, 2019, 108 : 747 - 763
  • [2] Supervised representation learning for multi-label classification
    Huang, Ming
    Zhuang, Fuzhen
    Zhang, Xiao
    Ao, Xiang
    Niu, Zhengyu
    Zhang, Min-Ling
    He, Qing
    [J]. MACHINE LEARNING, 2019, 108 (05) : 747 - 763
  • [3] Dual Sum-Product Networks Autoencoder for Multi-Label Classification
    Wang S.
    Zhang H.
    Chen J.
    [J]. Journal of Shanghai Jiaotong University (Science), 2020, 25 (05) : 665 - 673
  • [4] Gaussian Mixture Variational Autoencoder with Contrastive Learning for Multi-Label Classification
    Bai, Junwen
    Kong, Shufeng
    Gomes, Carla
    [J]. INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 162, 2022,
  • [5] Multi-label Learning via Supervised Autoencoder
    Lian, Siming
    Liu, Jianwei
    Lu, Runkun
    Luo, Xionglin
    [J]. 2018 37TH CHINESE CONTROL CONFERENCE (CCC), 2018, : 9416 - 9421
  • [6] Learning Dual Low-Rank Representation for Multi-Label Micro-Video Classification
    Lu, Wei
    Li, Desheng
    Nie, Liqiang
    Jing, Peiguang
    Su, Yuting
    [J]. IEEE TRANSACTIONS ON MULTIMEDIA, 2023, 25 : 77 - 89
  • [7] Label-aware graph representation learning for multi-label image classification
    Chen, Yilu
    Zou, Changzhong
    Chen, Jianli
    [J]. NEUROCOMPUTING, 2022, 492 : 50 - 61
  • [8] Multi-label classification using a cascade of stacked autoencoder and extreme learning machines
    Law, Anwesha
    Ghosh, Ashish
    [J]. NEUROCOMPUTING, 2019, 358 : 222 - 234
  • [9] Geometrically Preserved Dual Projections Learning for Multi-label Classification
    Zhi-Feng Liu
    Rui-Hang Cai
    Timothy Apasiba Abeo
    Cong-Hua Zhou
    Xiang-Jun Shen
    [J]. Neural Processing Letters, 2023, 55 : 7369 - 7392
  • [10] Geometrically Preserved Dual Projections Learning for Multi-label Classification
    Liu, Zhi-Feng
    Cai, Rui-Hang
    Abeo, Timothy Apasiba
    Zhou, Cong-Hua
    Shen, Xiang-Jun
    [J]. NEURAL PROCESSING LETTERS, 2023, 55 (06) : 7369 - 7392