IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v12y2024i17p2749-d1471596.html
   My bibliography  Save this article

An End-to-End, Multi-Branch, Feature Fusion-Comparison Deep Clustering Method

Author

Listed:
  • Xuanyu Li

    (School of Computer Science and Technology, Hainan University, Haikou 570228, China
    Haikou Key Laboratory of Deep Learning and Big Data Application Technology, Hainan University, Haikou 570228, China)

  • Houqun Yang

    (School of Computer Science and Technology, Hainan University, Haikou 570228, China
    Haikou Key Laboratory of Deep Learning and Big Data Application Technology, Hainan University, Haikou 570228, China)

Abstract

The application of contrastive learning in image clustering in the field of unsupervised learning has attracted much attention due to its ability to effectively improve clustering performance. Extracting features for face-oriented clustering using deep learning networks has also become one of the key challenges in this field. Some current research focuses on learning valuable semantic features using contrastive learning strategies to accomplish cluster allocation in the feature space. However, some studies decoupled the two phases of feature extraction and clustering are prone to error transfer, on the other hand, features learned in the feature extraction phase of multi-stage training are not guaranteed to be suitable for the clustering task. To address these challenges, We propose an end-to-end multi-branch feature fusion comparison deep clustering method (SwEAC), which incorporates a multi-branch feature extraction strategy in the representation learning phase, this method completes the clustering center comparison between multiple views and then assigns clusters to the extracted features. In order to extract higher-level semantic features, a multi-branch structure is used to learn multi-dimensional spatial channel dimension information and weighted receptive-field spatial features, achieving cross-dimensional information exchange of multi-branch sub-features. Meanwhile, we jointly optimize unsupervised contrastive representation learning and clustering in an end-to-end architecture to obtain semantic features for clustering that are more suitable for clustering tasks. Experimental results show that our model achieves good clustering performance on three popular image datasets evaluated by three unsupervised evaluation metrics, which proves the effectiveness of end-to-end multi-branch feature fusion comparison deep clustering methods.

Suggested Citation

  • Xuanyu Li & Houqun Yang, 2024. "An End-to-End, Multi-Branch, Feature Fusion-Comparison Deep Clustering Method," Mathematics, MDPI, vol. 12(17), pages 1-14, September.
  • Handle: RePEc:gam:jmathe:v:12:y:2024:i:17:p:2749-:d:1471596
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/12/17/2749/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/12/17/2749/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:17:p:2749-:d:1471596. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.