Automatic deep spare clustering with a dynamic population-based evolutionary algorithm using reinforcement learning and transfer learning

被引:1
作者
Hadikhani, Parham [1 ]
Lai, Daphne Teck Ching [1 ]
Ong, Wee-Hong [1 ]
Nadimi-Shahraki, Mohammad H. [2 ]
机构
[1] Univ Brunei Darussalam, Sch Digital Sci, Darussalam, Brunei
[2] Islamic Azad Univ, Fac Comp Engn, Najafabad Branch, Najafabad, Iran
关键词
Unsupervised learning; Deep clustering; Reinforcement learning; Transfer learning; Evolutionary algorithm; Differential evolution; Auto-encoder; Dimension reduction; GENETIC ALGORITHM; DATA SET; NUMBER; SIZE; SEPARATION;
D O I
10.1016/j.imavis.2024.105258
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Clustering data effectively remains a significant challenge in machine learning, particularly when the optimal number of clusters is unknown. Traditional deep clustering methods often struggle with balancing local and global search, leading to premature convergence and inefficiency. To address these issues, we introduce ADSCDPE-RT (Automatic Deep Sparse Clustering with a Dynamic Population-based Evolutionary Algorithm using Reinforcement Learning and Transfer Learning), a novel deep clustering approach. ADSC-DPE-RT builds on Multi-Trial Vector-based Differential Evolution (MTDE), an algorithm that integrates sparse auto-encoding and manifold learning to enable automatic clustering without prior knowledge of cluster count. However, MTDE's fixed population size can lead to either prolonged computation or premature convergence. Our approach introduces a dynamic population generation technique guided by Reinforcement Learning (RL) and Markov Decision Process (MDP) principles. This allows for flexible adjustment of population size, preventing premature convergence and reducing computation time. Additionally, we incorporate Generative Adversarial Networks (GANs) to facilitate dynamic knowledge transfer between MTDE strategies, enhancing diversity and accelerating convergence towards the global optimum. This is the first work to address the dynamic population issue in deep clustering through RL, combined with Transfer Learning to optimize evolutionary algorithms. Our results demonstrate significant improvements in clustering performance, positioning ADSC-DPE-RT as a competitive alternative to state-of-the-art deep clustering methods.
引用
收藏
页数:16
相关论文
共 96 条
[1]   A meta-heuristic density-based subspace clustering algorithm for high-dimensional data [J].
Agarwal, Parul ;
Mehta, Shikha ;
Abraham, Ajith .
SOFT COMPUTING, 2021, 25 (15) :10237-10256
[2]  
[Anonymous], 1997, The EM Algorithm and Extensions
[3]  
[Anonymous], 2024, Image Vis. Comput., V145
[4]   An extensive comparative study of cluster validity indices [J].
Arbelaitz, Olatz ;
Gurrutxaga, Ibai ;
Muguerza, Javier ;
Perez, Jesus M. ;
Perona, Inigo .
PATTERN RECOGNITION, 2013, 46 (01) :243-256
[5]  
Shah SA, 2018, Arxiv, DOI [arXiv:1803.01449, DOI 10.48550/ARXIV.1803.01449]
[6]   Self-supervised deep geometric subspace clustering network [J].
Baek, Sangwon ;
Yoon, Gangjoon ;
Song, Jinjoo ;
Yoon, Sang Min .
INFORMATION SCIENCES, 2022, 610 :235-245
[7]   Population size reduction for the differential evolution algorithm [J].
Brest, Janez ;
Maucec, Mirjam Sepesy .
APPLIED INTELLIGENCE, 2008, 29 (03) :228-247
[8]   Efficient Deep Embedded Subspace Clustering [J].
Cai, Jinyu ;
Fan, Jicong ;
Guo, Wenzhong ;
Wang, Shiping ;
Zhang, Yunhe ;
Zhang, Zhao .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :21-30
[9]  
Calinski T., 1974, Communications in Statistics-theory and Methods, V3, P1, DOI [DOI 10.1080/03610927408827101, 10.1080/03610927408827101, https://doi.org/10.1080/03610927408827101]
[10]   Deep Clustering for Unsupervised Learning of Visual Features [J].
Caron, Mathilde ;
Bojanowski, Piotr ;
Joulin, Armand ;
Douze, Matthijs .
COMPUTER VISION - ECCV 2018, PT XIV, 2018, 11218 :139-156