Improving Image Clustering through Sample Ranking and Its Application to Remote Sensing Images

Image clustering is a very useful technique that is widely applied to various areas, including remote sensing. Recently, visual representations by self-supervised learning have greatly improved the performance of image clustering. To further improve the well-trained clustering models, this paper pro...

Full description

Bibliographic Details
Main Authors: Qinglin Li, Guoping Qiu
Format: Article
Language:English
Published: MDPI AG 2022-07-01
Series:Remote Sensing
Subjects:
Online Access:https://www.mdpi.com/2072-4292/14/14/3317
Description
Summary:Image clustering is a very useful technique that is widely applied to various areas, including remote sensing. Recently, visual representations by self-supervised learning have greatly improved the performance of image clustering. To further improve the well-trained clustering models, this paper proposes a novel method by first ranking samples within each cluster based on the confidence in their belonging to the current cluster and then using the ranking to formulate a weighted cross-entropy loss to train the model. For ranking the samples, we developed a method for computing the likelihood of samples belonging to the current clusters based on whether they are situated in densely populated neighborhoods, while for training the model, we give a strategy for weighting the ranked samples. We present extensive experimental results that demonstrate that the new technique can be used to improve the state-of-the-art image clustering models, achieving accuracy performance gains ranging from <inline-formula><math xmlns="http://www.w3.org/1998/Math/MathML" display="inline"><semantics><mrow><mn>2.1</mn><mo>%</mo></mrow></semantics></math></inline-formula> to <inline-formula><math xmlns="http://www.w3.org/1998/Math/MathML" display="inline"><semantics><mrow><mn>15.9</mn><mo>%</mo></mrow></semantics></math></inline-formula>. Performing our method on a variety of datasets from remote sensing, we show that our method can be effectively applied to remote sensing images.
ISSN:2072-4292