arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.02447cs.LGcs.AI

基于主动推理的判别式神经网络分类器优化

Active Inference-Based Optimization of Discriminative Neural Network Classifiers

Institute of Signal Processing and System Theory(信号处理与系统理论研究所);University of Stuttgart(斯图加特大学)

Faezeh Fallah

AI总结:

本文提出基于主动推理与广义 Kelly 准则的判别式神经网络优化方法,通过候选标签和期望自由能损失同时处理类别不平衡及参考标签错误。

摘要:

用于判别式神经网络分类器监督优化的常用目标函数(损失)要么基于分布,要么基于度量。基于分布的损失可能损害泛化能力,或导致分类偏向不平衡类别样本分布中的主导类别。基于度量的损失可使网络模型独立于任何分布,从而提高其泛化能力。然而,它们仍可能偏向主导类别,并且当某个类别在参考标签(真值)和预测标签中均不存在时,可能出现差异。本文提出一种新颖的优化过程,不仅解决训练样本类别样本分布的不平衡问题,还提供一种处理训练样本参考标签中错误的机制。这是通过提出一种新颖算法和一种用于优化的新颖目标函数实现的:该算法根据训练样本的先验概率及其在网络上当前估计的后验概率,寻找训练样本的候选分类标签。该算法源于将最优投注的广义 Kelly 准则转化为多类分类问题。所提出的目标函数是预期主动推理的期望自由能,能够融合候选标签、原始参考标签和训练样本的先验,同时仍保持基于分布的特性。将先验融入优化不仅有助于处理参考标签中的错误,还能通过将神经网络的注意力集中在重要但属于少数的前景类别上,减少对主导类别的分类偏差。

英文摘要:

Commonly used objective functions (losses) for a supervised optimization of discriminative neural network classifiers were either distribution-based or metric-based. The distribution-based losses could compromise the generalization or cause classification biases towards the dominant classes of an imbalanced class-sample distribution. The metric-based losses could make the network model independent of any distribution and thus improve its generalization. However, they could still be biased towards the dominant classes and could suffer from discrepancies when a class was absent in both the reference (ground truth) and the predicted labels. In this paper, we proposed a novel optimization process which not only tackled the unbalancedness of the class-sample distribution of the training samples but also provided a mechanism to tackle errors in the reference labels of the training samples. This was achieved by proposing a novel algorithm to find candidate classification labels of the training samples from their prior probabilities and the currently estimated posteriors on the network and a novel objective function for the optimizations. The algorithm was the result of casting the generalized Kelly criterion for optimal betting into a multiclass classification problem. The proposed objective function was the expected free energy of a prospective active inference and could incorporate the candidate labels, the original reference labels, and the priors of the training samples while still being distribution-based. The incorporation of the priors into the optimization not only helped to tackle errors in the reference labels but also allowed to reduce classification biases towards the dominant classes by focusing the attention of the neural network on important but minority foreground classes.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑