中文

基于蒙特卡洛树搜索的代价敏感特征分类

机器学习 2021-02-16 v1

摘要

我们考虑代价敏感特征分类问题,即顺序选择特征子集以在分类误差与特征代价之间取得平衡。本文将此任务建模为 MDP 问题并使用 Advantage Actor Critic 算法求解。为了进一步提升智能体性能并使策略可解释,我们采用蒙特卡洛树搜索迭代更新策略。在此过程中,我们还考虑了其在不平衡数据集上的性能以及对缺失值的敏感性。我们在多个数据集上评估了模型,发现其优于其他方法。

关键词

引用

@article{arxiv.2102.07073,
  title  = {Costly Features Classification using Monte Carlo Tree Search},
  author = {Ziheng Chen and Jin Huang and Hongshik Ahn and Xin Ning},
  journal= {arXiv preprint arXiv:2102.07073},
  year   = {2021}
}