arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.09630cs.LG
收藏

Predictive K-means with local models

Orange Labs(Orange实验室);AgroParisTech, Université Paris-Saclay(巴黎高科农业学院,巴黎萨克雷大学);LIM-EA2525, Université de La Réunion(留尼汪大学LIM-EA2525实验室)

Vincent Lemaire, Oumaima Alaoui Ismaili, Antoine Cornuéjols, Dominique Gay

英文摘要:

Supervised classification can be effective for prediction but sometimes weak on interpretability or explainability (XAI). Clustering, on the other hand, tends to isolate categories or profiles that can be meaningful but there is no guarantee that they are useful for labels prediction. Predictive clustering seeks to obtain the best of the two worlds. Starting from labeled data, it looks for clusters that are as pure as possible with regards to the class labels. One technique consists in tweaking a clustering algorithm so that data points sharing the same label tend to aggregate together. With distance-based algorithms, such as k-means, a solution is to modify the distance used by the algorithm so that it incorporates information about the labels of the data points. In this paper, we propose another method which relies on a change of representation guided by class densities and then carries out clustering in this new representation space. We present two new algorithms using this technique and show on a variety of data sets that they are competitive for prediction performance with pure supervised classifiers while offering interpretability of the clusters discovered.

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

↑