APPARATUS AND METHOD OF IMAGE CLUSTERING

The present invention relates to an apparatus and method for clustering. The apparatus for clustering according to one embodiment of the present invention includes: a transformed image generating unit generating transformed images by transforming unlabeled original images; a pre-learning unit generating a feature vector regarding each transformed image using an artificial neural network-based encoder and training the encoder based on the feature vector regarding each transformed image; a pseudo label generating unit generating a feature vector regarding each unlabeled learning image using the trained encoder, performing clustering of the learning images based on the feature vector regarding each unlabeled learning image, and generating a pseudo label regarding at least one of the learning images based on the clustering result; and an additional learning unit generating a prediction label regarding the at least one learning image using a classification model including a classifier generating a prediction label regarding an image input to the trained encoder based on the trained encoder and the feature vector generated by the trained encoder and training the classification model based on the prediction label and the pseudo label regarding the at least one learning image. 클러스터링 방법 및 장치가 개시된다. 본 발명의 일 실시예에 따른 클러스터링 장치는 라벨링 되지 않은 복수의 원본 이미지 각각을 변형하여 복수의 변형 이미지를 생성하는 변형 이미지 생성부; 인공 신경망 기반의 인코더를 이용하여 상기 복수의 변형 이미지 각각에 대한 특징 벡터를 생성하고, 상기 복수의 변형 이미지 각각에 대한 특징 벡터에 기초하여 상기 인코더를 학습하는 사전 학습부; 상기 학습된 인코더를 이용하여 라벨링되지 않은 복수의 학습 이미지 각각에 대한 특징 벡터를 생성하고, 상기 복수의 학습 이미지 각각에 대한 특징 벡터에 기초하여 상기 복수의 학습 이미지에 대한 군집화를 수행하고, 상기 군집화 결과에 기초하여 상기 복수의 학습 이미지 중 하나 이상의 학습 이미지에 대한 의사 라벨을 생성하는 의사 라벨 생성부; 및 상기 학습된 인코더 및 상기 학습된 인코더에 의해 생성된 특징 벡터에 기초하여 상기 학습된 인코더로 입력된 이미지에 대한 예측 라벨을 생성하는 분류기를 포함하는 분류 모델을 이용하여 상기 하나 이상의 학습 이미지에 대한 예측 라벨을 생성하고, 상기 하나 이상의 학습 이미지에 대한 의사 라벨 및 예측 라벨에 기초하여 상기 분류 모델을 학습하는 추가 학습부를 포함한다.