Learning conditional distributions on continuous spaces - Archive ouverte HAL
Pré-Publication, Document De Travail Année : 2024

Learning conditional distributions on continuous spaces

Résumé

We investigate sample-based learning of conditional distributions on multi-dimensional unit boxes, allowing for different dimensions of the feature and target spaces. Our approach involves clustering data near varying query points in the feature space to create empirical measures in the target space. We employ two distinct clustering schemes: one based on a fixed-radius ball and the other on nearest neighbors. We establish upper bounds for the convergence rates of both methods and, from these bounds, deduce optimal configurations for the radius and the number of neighbors. We propose to incorporate the nearest neighbors method into neural network training, as our empirical analysis indicates it has better performance in practice. For efficiency, our training process utilizes approximate nearest neighbors search with random binary space partitioning. Additionally, we employ the Sinkhorn algorithm and a sparsity-enforced transport plan. Our empirical findings demonstrate that, with a suitably designed structure, the neural network has the ability to adapt to a suitable level of Lipschitz continuity locally. For reproducibility, our code is available at \url{https://github.com/zcheng-a/LCD_kNN}.
Fichier principal
Vignette du fichier
KernelLearning.pdf (5.13 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04615853 , version 1 (18-06-2024)

Identifiants

Citer

Cyril Bénézet, Ziteng Cheng, Sebastian Jaimungal. Learning conditional distributions on continuous spaces. 2024. ⟨hal-04615853⟩
16 Consultations
28 Téléchargements

Altmetric

Partager

More