Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
zhaijianyang ead6340af6 | 2 years ago | |
---|---|---|
data/usps | 2 years ago | |
.gitignore | 2 years ago | |
ASPC.py | 2 years ago | |
MyModel.py | 2 years ago | |
README.md | 2 years ago | |
datasets.py | 2 years ago | |
metrics.py | 2 years ago | |
run_exp.py | 2 years ago |
Tensorflow implementation for our paper:
Deep clustering gains superior performance than conventional clustering by jointly performing feature learning and cluster
assignment. Although numerous deep clustering algorithms have emerged in various applications, most of them fail to learn robust
cluster-oriented features which in turn hurts the final clustering performance. To solve this problem, we propose a two-stage deep
clustering algorithm by incorporating data augmentation and self-paced learning. Specifically, in the first stage, we learn robust features
by training an autoencoder with examples that are augmented by random shifting and rotating the given clean examples. Then in the
second stage, we encourage the learned features to be cluster-oriented by alternatively finetuning the encoder with the augmented
examples and updating the cluster assignments of the clean examples. During finetuning the encoder, the target of each augmented
example in the loss function is the center of the cluster to which the clean example is assigned. The targets may be computed
incorrectly, and the examples with incorrect targets could mislead the encoder network. To stabilize the network training, we select
most confident examples in each iteration by utilizing the adaptive self-paced learning. Extensive experiments validate that our
algorithm outperforms the state of the arts on four image datasets.
Install Anaconda with Python 3.6 version (Optional).
Create a new env (Optional):
conda create -n aspc python=3.6 -y
source activate aspc # Linux
# or
conda activate aspc # Windows
Install required packages:
pip install tensorflow-gpu==1.10 scikit-learn h5py
git clone https://github.com/XifengGuo/ASPC-DA.git ASPC-DA
cd ASPC-DA
python run_exp.py
Adaptive Self-paced Deep Clustering with Data Augmentation
Python
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》