Voronoi Density Estimator for High-Dimensional Data: Computation, Compactification and Convergence
Vladislav Polianskii, Giovanni Luca Marchetti, Alexander Kravberg, Anastasiia Varava, Florian T. Pokorny, Danica Kragic
TL;DR
This work introduces the Compactified Voronoi Density Estimator (CVDE) to overcome high-dimensional limitations of the classical Voronoi Density Estimator by compactifying Voronoi cells with a kernel-based measure. It develops efficient computational procedures via spherical-volume integration and hit-and-run sampling, with GPU-accelerated implementation, and proves convergence of the CVDE to the true density without requiring kernel bandwidth vanishing. Empirical results show CVDE outperforms KDE on synthetic and image-like data and remains effective in high dimensions where traditional VDE struggles. The approach offers a geometry-adaptive, nonparametric density estimator with favorable convergence properties and practical scalability for complex, high-dimensional data.
Abstract
The Voronoi Density Estimator (VDE) is an established density estimation technique that adapts to the local geometry of data. However, its applicability has been so far limited to problems in two and three dimensions. This is because Voronoi cells rapidly increase in complexity as dimensions grow, making the necessary explicit computations infeasible. We define a variant of the VDE deemed Compactified Voronoi Density Estimator (CVDE), suitable for higher dimensions. We propose computationally efficient algorithms for numerical approximation of the CVDE and formally prove convergence of the estimated density to the original one. We implement and empirically validate the CVDE through a comparison with the Kernel Density Estimator (KDE). Our results indicate that the CVDE outperforms the KDE on sound and image data.
