ArXiv

DADIR: Density-Aware Data-level Imbalanced Regression Framework

Authors
Shermin Shahbazi, Hossein Mohammadi, Mohsen Afsharchi
Categories
cs.LG, cs.AI
arXiv
https://arxiv.org/abs/2607.17178v1
PDF
https://arxiv.org/pdf/2607.17178v1

Brief

DADIR addresses imbalanced regression by jointly exploiting target-density and local feature structure. It adaptively partitions the target space (DAAP), trains a density-regularized CVAE to retain sparse-region features (DR-CVAE), and performs clustering-based latent oversampling to synthesize realistic minority samples. Reported experiments (abstract) indicate consistent gains—especially on underrepresented targets—using balanced data with off-the-shelf regressors. Full paper content was not provided, only the abstract.

Why it matters

DADIR (by Shermin Shahbazi, Hossein Mohammadi, Mohsen Afsharchi; arXiv:2607.17178v1, 2026-07-19) introduces three components—Density-Aware Adaptive Partitioning (DAAP), a Density-Regularized Conditional VAE (DR-CVAE), and latent-space data balancing—to detect minority target regions and generate structurally consistent synthetic samples while preserving sparse-region representations.

Key details

  • Experiments on multiple imbalanced regression benchmarks show consistent improvements in predictive performance—notably in underrepresented target regions—while also improving overall accuracy; the produced balanced datasets can be used with existing regression models without changing architectures or loss functions.
Source evidence

Abstract

Imbalanced learning addresses predictive modeling problems with underrepresented regions of the data distribution. Although widely studied in classification, imbalanced regression remains challenging because of continuous target variables and heterogeneous density distributions. Existing data-level methods often rely on fixed target partitioning or synthetic sample generation without jointly considering density variations and local feature-space structure. We propose DADIR, a Density-Aware Data-level Imbalanced Regression framework that exploits density information throughout the balancing process. DADIR comprises three components: (1) Density-Aware Adaptive Partitioning (DAAP), which recursively partitions the target space according to density variations; (2) a Density-Regularized Conditional Variational Autoencoder (DR-CVAE), which preserves sparse-region representations while learning latent features; and (3) latent-space data balancing, which combines feature-level clustering with oversampling to generate structurally consistent synthetic samples. Together, these components identify minority regions more effectively, preserve sparse-region information, and generate realistic synthetic data. The resulting balanced dataset can be used directly with existing regression models without modifying their architecture or learning objective. Experiments on diverse imbalanced regression datasets demonstrate consistent improvements in predictive performance, particularly in underrepresented regions, while also improving overall accuracy.