α-MeanShift++: Improving MeanShift++ for Image Segmentation

• Main Author: Hanhoon Park
• Format: Article
• Language: English
• Published: IEEE 2021-01-01
• Series: IEEE Access
• Subjects: Clustering; mean shift algorithm; MeanShift++; image segmentation
• Online Access: https://ieeexplore.ieee.org/document/9541345/

MeanShift is one of the popular clustering algorithms and can be used to partition a digital image into semantically meaningful regions in an unsupervised manner. However, due to its prohibitively high computational complexity, a grid-based approach, called MeanShift&#x002B;&#x002B;, has recently been proposed and succeeded to surprisingly reduce the computational complexity of MeanShift. Nevertheless, we found that MeanShift&#x002B;&#x002B; still has computational redundancy and there is room for improvement in terms of accuracy and runtime; thus, we propose an improvement to MeanShift&#x002B;&#x002B;, named <inline-formula> <tex-math notation="LaTeX">$\alpha $ </tex-math></inline-formula>-MeanShift&#x002B;&#x002B;. We first attempt to minimize the computational redundancy by using an additional hash table. Then, we introduce a speedup factor (<inline-formula> <tex-math notation="LaTeX">$\alpha $ </tex-math></inline-formula>) to reduce the number of iterations required until convergence, and we use more neighboring grid cells for the same bandwidth to improve accuracy. Through intensive experiments on image segmentation benchmark datasets, we demonstrate that <inline-formula> <tex-math notation="LaTeX">$\alpha $ </tex-math></inline-formula>-MeanShift&#x002B;&#x002B; can run 4.1-<inline-formula> <tex-math notation="LaTeX">$4.6\times $ </tex-math></inline-formula> faster on average (but up to <inline-formula> <tex-math notation="LaTeX">$7\times $ </tex-math></inline-formula>) than MeanShift&#x002B;&#x002B; and achieve better image segmentation quality.