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 条
[21]   Parallel evolutionary computation for solving complex CFD optimization problems:: A review and some nozzle applications [J].
Galvan, B ;
Greiner, D ;
Périaux, J ;
Sefrioui, M ;
Winter, G .
PARALLEL COMPUTATIONAL FLUID DYNAMICS: NEW FRONTIERS AND MULTI-DISCIPLINARY APPLICATIONS, PROCEEDINGS, 2003, :573-604
[22]  
Gong L., 2024, Attributed graph clustering with dual redundancy reduction
[23]   Generative Adversarial Networks [J].
Goodfellow, Ian ;
Pouget-Abadie, Jean ;
Mirza, Mehdi ;
Xu, Bing ;
Warde-Farley, David ;
Ozair, Sherjil ;
Courville, Aaron ;
Bengio, Yoshua .
COMMUNICATIONS OF THE ACM, 2020, 63 (11) :139-144
[24]   Adaptive Self-Paced Deep Clustering with Data Augmentation [J].
Guo, Xifeng ;
Liu, Xinwang ;
Zhu, En ;
Zhu, Xinzhong ;
Li, Miaomiao ;
Xu, Xin ;
Yin, Jianping .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2020, 32 (09) :1680-1693
[25]  
Guo XF, 2017, PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1753
[26]   Human Activity Discovery With Automatic Multi-Objective Particle Swarm Optimization Clustering With Gaussian Mutation and Game Theory [J].
Hadikhani, Parham ;
Lai, Daphne Teck Ching ;
Ong, Wee-Hong .
IEEE TRANSACTIONS ON MULTIMEDIA, 2024, 26 :420-435
[27]   Automatic Deep Sparse Multi-Trial Vector-based Differential Evolution clustering with manifold learning and incremental technique [J].
Hadikhani, Parham ;
Lai, Daphne Teck Ching ;
Ong, Wee-Hong ;
Nadimi-Shahraki, Mohammad H. .
IMAGE AND VISION COMPUTING, 2023, 136
[28]   A Novel Skeleton-Based Human Activity Discovery Using Particle Swarm Optimization With Gaussian Mutation [J].
Hadikhani, Parham ;
Lai, Daphne Teck Ching ;
Ong, Wee-Hong .
IEEE TRANSACTIONS ON HUMAN-MACHINE SYSTEMS, 2023, 53 (03) :538-548
[29]   Improved data clustering using multi-trial vector-based differential evolution with Gaussian crossover [J].
Hadikhani, Parham ;
Lai, Daphne Teck Ching ;
Ong, Wee-Hong ;
Nadimi-Shahraki, Mohammad H. .
PROCEEDINGS OF THE 2022 GENETIC AND EVOLUTIONARY COMPUTATION CONFERENCE COMPANION, GECCO 2022, 2022, :487-490
[30]   An adaptive hybrid algorithm for social networks to choose groups with independent members [J].
Hadikhani, Parham ;
Hadikhani, Pooria .
EVOLUTIONARY INTELLIGENCE, 2020, 13 (04) :695-703