Scikit-dimension: a Python package for intrinsic dimension estimation

Abstract

Dealing with uncertainty in applications of machine learning to real-life data critically depends on the knowledge of intrinsic dimensionality (ID). A number of methods have been suggested for the purpose of estimating ID, but no standard package to easily apply them one by one or all at once has been implemented in Python. This technical note introduces \texttt{scikit-dimension}, an open-source Python package for intrinsic dimension estimation. \texttt{scikit-dimension} package provides a uniform implementation of most of the known ID estimators based on scikit-learn application programming interface to evaluate global and local intrinsic dimension, as well as generators of synthetic toy and benchmark datasets widespread in the literature. The package is developed with tools assessing the code quality, coverage, unit testing and continuous integration. We briefly describe the package and demonstrate its use in a large-scale (more than 500 datasets) benchmarking of methods for ID estimation in real-life and synthetic data. The source code is available from https://github.com/j-bac/scikit-dimension , the documentation is available from https://scikit-dimension.readthedocs.io .

arXiv comment12 pages, 4 figures, 1 table

References in corpus (16)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• Array Programming with NumPy
• OpenML: networked science in machine learning
• Estimating the intrinsic dimension of datasets by a minimal neighborhood information
• Principal manifolds and graphs in practice: from molecular biology to dynamical systems
• Blessing of dimensionality: mathematical foundations of the statistical physics of data
• Robust And Scalable Learning Of Complex Dataset Topologies Via Elpigraph
• Correction of AI systems by linear discriminants: Probabilistic foundations
• The unreasonable effectiveness of small neural ensembles in high-dimensional brain
• Trajectories, bifurcations and pseudotime in large clinical datasets: applications to myocardial infarction and diabetes data
• Topological Grammars for Data Approximation
• Fractional norms and quasinorms do not help to overcome the curse of dimensionality
• On Adversarial Examples and Stealth Attacks in Artificial Intelligence Systems
• General stochastic separation theorems with optimal bounds
• Local intrinsic dimensionality estimators based on concentration of measure
• Geometry-Aware Maximum Likelihood Estimation of Intrinsic Dimension

Cited by in corpus (9)

• Domain Adaptation Principal Component Analysis: base linear method for learning with out-of-distribution data
• The Intrinsic Manifolds of Radiological Images and their Role in Deep Learning
• CaloChallenge 2022: A Community Challenge for Fast Calorimeter Simulation
• Compressing and forecasting atomic material simulations with descriptors
• Dataset-Adaptive Dimensionality Reduction
• Density Estimation via Binless Multidimensional Integration
• Relative intrinsic dimensionality is intrinsic to learning
• Intrinsic Dimension Estimating Autoencoder (IDEA) Using CancelOut Layer and a Projected Loss
• MDIntrinsicDimension: Dimensionality-Based Analysis of Collective Motions in Macromolecules from Molecular Dynamics Trajectories