A reinforcement learning guided adaptive cost-sensitive feature acquisition method

被引:0
|
作者
An, Chaojie
Zhou, Qifeng [1 ]
Yang, Shen
机构
[1] Xiamen Univ, Dept Automat, Xiamen, Peoples R China
关键词
Cost-sensitive; Feature acquisition; Recurrent neural network; Reinforcement learning; FEATURE-SELECTION; MUTUAL INFORMATION;
D O I
10.1016/j.asoc.2022.108437
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Most of the existing feature selection methods tend to pursue the learning performance of the selected feature subset while ignoring the costs of acquiring each feature. However, in real-world problems, we often face the tradeoff between model performance and feature costs because of limited resources. Moreover, in some applications (e.g. medical tests), features are acquired sequentially in the learning process instead of having known the information of the whole feature set in advance. To solve these problems we design a reinforcement learning agent to guide the cost-sensitive feature acquisition process and propose a deep learning-based model to select the informative and lower-cost features for each instance adaptively. The whole process of feature acquisition will be determined by an agent according to what it has observed from inputs. In particular, a Recurrent Neural Network (RNN) model will learn the knowledge from the current sample and the agent will give the instructions on whether the RNN model will continue to select the next feature or stop the sequential feature acquisition process. Moreover, the proposed method can also select the features per block thus it can deal with high dimensional data. We evaluate the effectiveness of the proposed method on a variety of datasets including benchmark datasets, gene datasets, and medical datasets. Compared with the state-of-the-art feature selection methods, the proposed method can achieve comparable learning accuracy while maintaining lower feature costs. (C)& nbsp;2022 Elsevier B.V. All rights reserved.
引用
收藏
页数:10
相关论文
共 50 条
  • [41] Adversarial Learning With Cost-Sensitive Classes
    Shen, Haojing
    Chen, Sihong
    Wang, Ran
    Wang, Xizhao
    IEEE TRANSACTIONS ON CYBERNETICS, 2023, 53 (08) : 4855 - 4866
  • [42] Cost-Sensitive Decision Tree Learning
    Vadera, Sunil
    PROCEEDINGS 2019 AMITY INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE (AICAI), 2019, : 4 - 5
  • [43] Cost-sensitive positive and unlabeled learning
    Chen, Xiuhua
    Gong, Chen
    Yang, Jian
    INFORMATION SCIENCES, 2021, 558 : 229 - 245
  • [44] Robust SVM for Cost-Sensitive Learning
    Jiangzhang Gan
    Jiaye Li
    Yangcai Xie
    Neural Processing Letters, 2022, 54 : 2737 - 2758
  • [45] Cost-Sensitive Action Model Learning
    Rao, Dongning
    Jiang, Zhihua
    INTERNATIONAL JOURNAL OF UNCERTAINTY FUZZINESS AND KNOWLEDGE-BASED SYSTEMS, 2016, 24 (02) : 167 - 193
  • [46] Roulette sampling for cost-sensitive learning
    Sheng, Victor S.
    Ling, Charles X.
    MACHINE LEARNING: ECML 2007, PROCEEDINGS, 2007, 4701 : 724 - +
  • [47] Cost-sensitive learning for defect escalation
    Sheng, Victor S.
    Gu, Bin
    Fang, Wei
    Wu, Jian
    KNOWLEDGE-BASED SYSTEMS, 2014, 66 : 146 - 155
  • [48] Robust SVM for Cost-Sensitive Learning
    Gan, Jiangzhang
    Li, Jiaye
    Xie, Yangcai
    NEURAL PROCESSING LETTERS, 2022, 54 (04) : 2737 - 2758
  • [49] Active learning for cost-sensitive classification
    Krishnamurthy, Akshay
    Agarwal, Alekh
    Huang, Tzu-Kuo
    Daumé Iii, Hal
    Langford, John
    Journal of Machine Learning Research, 2019, 20
  • [50] Cost-sensitive learning with neural networks
    Kukar, M
    Kononenko, I
    ECAI 1998: 13TH EUROPEAN CONFERENCE ON ARTIFICIAL INTELLIGENCE, PROCEEDINGS, 1998, : 445 - 449