通过学习核函数提升多分类贝叶斯分类性能
Learning Kernels by Alignment for Multiclass Bayes Classification

- 用可学习的马氏距离替代余弦相似度进行核对齐
- 在CIFAR-10等数据集上准确率更高、收敛更快
- 适合需要可靠置信度估计的医疗预测场景
核方法将数据表示与决策分离,但通常需预先选定核函数。本文表明核函数可通过核对齐来学习,并基于新提出的协同学习与推理(CLaI)框架进行发展。我们证明协同学习本质上是核对齐过程,其中嵌入向量被训练以使诱导相似性匹配由标签导出的目标核。同时证明协同推理等价于使用Parzen窗密度估计的核贝叶斯分类。受此启发,我们将CLaI推广为使用可学习的马氏距离代替余弦相似度,并扩展至多分类任务。在CIFAR-10、PathMNIST和SleepEDF数据集上,马氏形式相比余弦变体提升准确率、加速收敛并降低校准误差。辅助实验进一步验证了这些联系:CLaI生成的潜在信号与高斯过程形式一致,且在脓毒症预测中达到良好校准效果。这些结果建立了一个统一表示学习、核对齐与贝叶斯分类的原则性学习核框架,并自然延伸至多分类设置。
原文摘要 · Abstract (English)
Kernel methods separate data representation from decision-making, but typically require the kernel to be chosen in advance. We show that this kernel can instead be learned by alignment, and develop the resulting framework through the recently introduced Collaborative Learning and Inference (CLaI). We show that Collaborative Learning can be viewed as a kernel alignment process, in which an embedding is trained so that its induced similarity matches a label-derived target kernel. We also prove that Collaborative Inference is equivalent to kernel Bayes classification with Parzen-window density estimation. Motivated by these perspectives, we generalise CLaI by replacing cosine similarity with a learned Mahalanobis distance and extend it to multiclass classification. On CIFAR-10, PathMNIST, and SleepEDF, the Mahalanobis formulation improves accuracy, converges faster, and yields lower calibration error than the cosine-based variant. Auxiliary experiments further support these connections, showing that CLaI produces latent signals of the same form as a Gaussian process, while achieving competitive calibration on sepsis prediction. Together, these results establish a principled learned-kernel framework that unifies representation learning, kernel alignment, and Bayesian classification, and extends naturally to the multiclass setting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。