arXiv:2409.11100cs.LGstat.ML2024-09被引 3

通过稀疏正则化直接优化变量权重,让朴素贝叶斯分类器更简洁鲁棒。

Fractional Naive Bayes (FNB): non-convex optimization for a parsimonious weighted selective naive Bayes classifier

  • 直接优化变量权重,引入基于先验惩罚成本的稀疏正则化
  • 在基准数据集上用较少变量达到与平均法相当的性能
  • 提出两阶段算法求解非凸优化,适合高维数据特征选择

针对输入变量极多的监督分类问题,研究朴素贝叶斯分类器的改进方法。朴素贝叶斯因简单、可扩展且在真实数据中表现良好而具有吸引力。当输入变量在给定目标变量下不满足条件独立假设时,变量选择与模型平均是常见的提升手段。在朴素贝叶斯框架下,模型平均等价于变量加权。本文聚焦于直接估计变量权重,提出一种对模型对数似然的稀疏正则化方法,考虑各输入变量的先验惩罚成本。相比以往基于平均的分类器,目标是获得变量更少、性能相当的简约稳健模型。变量权重的直接估计转化为非凸优化问题,本文提出并比较多种两阶段算法:首先通过凸松弛最小化目标函数,采用多种梯度方法;再以第一阶段结果为初值,使用局部优化方法求解原非凸问题。所提算法得到基于优化的加权朴素贝叶斯分类器,在多个基准数据集上评估,并与参考平均法分类器对比。

原文摘要 · Abstract (English)

We study supervised classification for datasets with a very large number of input variables. The naïve Bayes classifier is attractive for its simplicity, scalability and effectiveness in many real data applications. When the strong naïve Bayes assumption of conditional independence of the input variables given the target variable is not valid, variable selection and model averaging are two common ways to improve the performance. In the case of the naïve Bayes classifier, the resulting weighting scheme on the models reduces to a weighting scheme on the variables. Here we focus on direct estimation of variable weights in such a weighted naïve Bayes classifier. We propose a sparse regularization of the model log-likelihood, which takes into account prior penalization costs related to each input variable. Compared to averaging based classifiers used up until now, our main goal is to obtain parsimonious robust models with less variables and equivalent performance. The direct estimation of the variable weights amounts to a non-convex optimization problem for which we propose and compare several two-stage algorithms. First, the criterion obtained by convex relaxation is minimized using several variants of standard gradient methods. Then, the initial non-convex optimization problem is solved using local optimization methods initialized with the result of the first stage. The various proposed algorithms result in optimization-based weighted naïve Bayes classifiers, that are evaluated on benchmark datasets and positioned w.r.t. to a reference averaging-based classifier.

朴素贝叶斯变量选择非凸优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。