arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.13634cs.CV

A Deep Learning Object Detection Method for an Efficient Clusters Initialization

Univ. Bourgogne Franche-Comté (UBFC)(勃艮第-弗朗什-孔泰大学(UBFC));FEMTO-ST Institute(FEMTO-ST研究院);American University of Beirut(贝鲁特美国大学);University of Bejaia(贝贾亚大学)

Raphaël Couturier, Hassan N. Noura, Ola Salman, Abderrahmane Sider

英文摘要:

Clustering is an unsupervised machine learning method grouping data samples into clusters of similar objects. In practice, clustering has been used in numerous applications such as banking customers profiling, document retrieval, image segmentation, and e-commerce recommendation engines. However, the existing clustering techniques present significant limitations, from which is the dependability of their stability on the initialization parameters (e.g. number of clusters, centroids). Different solutions were presented in the literature to overcome this limitation (i.e. internal and external validation metrics). However, these solutions require high computational complexity and memory consumption, especially when dealing with big data. In this paper, we apply the recent object detection Deep Learning (DL) model, named YOLO-v5, to detect the initial clustering parameters such as the number of clusters with their sizes and centroids. Mainly, the proposed solution consists of adding a DL-based initialization phase making the clustering algorithms free of initialization. Two model solutions are provided in this work, one for isolated clusters and the other one for overlapping clusters. The features of the incoming dataset determine which model to use. Moreover, The results show that the proposed solution can provide near-optimal clusters initialization parameters with low computational and resources overhead compared to existing solutions.

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

↑