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 条
  • [31] Reinforcement learning-based cost-sensitive classifier for imbalanced fault classification
    Zhang, Xinmin
    Fan, Saite
    Song, Zhihuan
    SCIENCE CHINA-INFORMATION SCIENCES, 2023, 66 (11)
  • [32] Disease Diagnosis with Cost-Sensitive Grouped Features Based on Deep Reinforcement Learning
    Sataer, Yikemaiti
    Gao, Zhiqiang
    Li, Xuelian
    2023 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN, 2023,
  • [33] Cost-sensitive sample shifting in feature space
    Zhenchong Zhao
    Xiaodan Wang
    Chongming Wu
    Lei Lei
    Pattern Analysis and Applications, 2020, 23 : 1689 - 1707
  • [34] A novel active cost-sensitive learning method for intrusion detection
    Long, Jun
    Yin, Jian-Ping
    Zhu, En
    Zhao, Wen-Tao
    PROCEEDINGS OF 2008 INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND CYBERNETICS, VOLS 1-7, 2008, : 1099 - 1104
  • [35] Cost-sensitive sample shifting in feature space
    Zhao, Zhenchong
    Wang, Xiaodan
    Wu, Chongming
    Lei, Lei
    PATTERN ANALYSIS AND APPLICATIONS, 2020, 23 (04) : 1689 - 1707
  • [36] IMCStacking: Cost-sensitive stacking learning with feature inverse mapping for imbalanced problems
    Cao, Chenjie
    Wang, Zhe
    KNOWLEDGE-BASED SYSTEMS, 2018, 150 : 27 - 37
  • [37] Active Learning for Cost-Sensitive Classification
    Krishnamurthy, Akshay
    Agarwal, Alekh
    Huang, Tzu-Kuo
    Daume, Hal, III
    Langford, John
    INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 70, 2017, 70
  • [38] Cost-sensitive learning of SVM for ranking
    Xu, Jun
    Cao, Yunbo
    Li, Hang
    Huang, Yalou
    MACHINE LEARNING: ECML 2006, PROCEEDINGS, 2006, 4212 : 833 - 840
  • [39] Cost-Sensitive Learning in Answer Extraction
    Wiegand, Michael
    Leidner, Jochen L.
    Klakow, Dietrich
    SIXTH INTERNATIONAL CONFERENCE ON LANGUAGE RESOURCES AND EVALUATION, LREC 2008, 2008, : 711 - 714
  • [40] Active Learning for Cost-Sensitive Classification
    Krishnamurthy, Akshay
    Agarwal, Alekh
    Huang, Tzu-Kuo
    Daume, Hal, III
    Langford, John
    JOURNAL OF MACHINE LEARNING RESEARCH, 2019, 20