Convolutional neural networks (CNNs) are widely used learning models in the field of computer vision. In the process of convolution initialization, the selection of the initial class center has a significant impact on the subsequent recognition accuracy and efficiency. To solve this problem, this paper proposes a CNN image classification enhancement method combined with K-means++ clustering algorithm. Firstly, the optimized K-means++ algorithm is used to cluster the image dataset to generate more representative class centers, to reduce the impact of initial class center selection on the classification task. To meet the special needs in the field of image recognition, this paper introduces the weight adjustment mechanism of image feature space and optimizes the K-means++ algorithm to make the clustering process closer to the high-dimensional feature distribution of image data. Subsequently, these optimized class centers were input into the CNN for deep feature extraction and classification. Experimental results show that the classification accuracy of the proposed method is significantly improved on multiple public image datasets. The classification accuracy on both CIFAR-10 and ImageNet datasets improved more than $\mathbf{4 \%}$. In summary, the proposed method provides a new enhancement strategy for image classification tasks, which has high practical value and application prospects.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Research on Image Classification Enhancement Method Based on K-means++ Clustering and CNN


    Contributors:
    Zhao, Xia (author) / Zhang, Linfang (author) / Ling, Yaohui (author) / Zhang, Zhiran (author) / He, Yizhuo (author) / Wang, Fuyan (author)


    Publication date :

    2024-10-23


    Size :

    592506 byte




    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English



    Simplified method of kernel fuzzy c-means clustering for image texture classification

    Yunneng, Y. / Yang, W. / Gong, C. | British Library Online Contents | 2008


    Research on Speaker Recognition Method Based on Subtractive Clustering and Improved Fuzzy C-means Clustering Algorithm

    Cui, L.-y. / Xu, L. / Gu, S.-s. et al. | British Library Online Contents | 2008


    Efficient fuzzy c-means clustering for image data

    Chen, Y.-S. / Chen, B.-T. / Hsu, W.-H. | British Library Online Contents | 2005



    On Autonomous k-Means Clustering

    Elomaa, T. / Koivistoinen, H. | British Library Conference Proceedings | 2005