arXiv 机器学习· Jo\~ao Pedro Silveira e Silva, M\'arcio Dorn, Bruno Iochins Grisci·· 6 小时前AI 评分28
用于神经网络多分类任务的特征选择层方法
A Feature Selection Layer Approach for Multi-Class Classification on Neural Networks
AI 导读
研究人员提出了针对神经网络特征选择层（Feature Selection Layer）的多项嵌入式扩展方案，用于解决传统嵌入方法难以捕捉多分类任务中标签特定关联的问题。实验表明，该方法在识别全局相关特征上比肩现有技术，并在捕捉标签特定相关性方面表现更优。该机制直接利用训练后神经网络的内部动态揭示单特征对特定标签预测的贡献，且在模型训练之外无需额外后处理。
来源：arXiv 机器学习 · arxiv.org