On Imbalanced Regression with Hoeffding Trees
Organizations: University of Oklahoma
Abstract
Many real-world applications generate continuous data streams for regression. Hoeffding trees and their variants have a long-standing tradition due to their effectiveness, either alone or as base models in broader ensembles. Recent batch-learning work shows that kernel density estimation (KDE) improves smoothed predictions in imbalanced regression [Yang et al., 2021], while hierarchical shrinkage (HS) provides post-hoc regularization for decision trees without modifying their structure [Agarwal et al., 2022]. We extend KDE to streaming settings via a telescoping formulation and integrate HS into incremental decision trees. Empirical evaluation on standard online regression benchmarks shows that KDE consistently improves early-stream performance, whereas HS provides limited gains. Our implementation is publicly available at: https://github.com/marinaAlchirch/DSFA_2026.
Figures & tables
| Predict-then-train | |||||
| set size used in | Outlier | Constructed | |||
| Dataset | #Features | #Examples | Algorithm 2 | removal | feature |
| Abalone | 8 | 4177 | 2089 | ✗ | ✗ |
| Semi | 229 | 19452 | - | ✓ | ✗ |
| California | 8 | 20635 | 10319 | ✓ | ✗ |
| NY Taxi | 10 | 19997 | 10001 | ✓ | ✓ |
| Model (potentially including HS, KDE, or both) | |||||||||
| HT | HAT | ||||||||
| HT | HAT | HT | HAT | +HS | +HS | ||||
| Dataset | Metric | HT | HAT | +HS | +HS | +KDE | +KDE | +KDE | +KDE |
| Abalone | MAE | 1.979 | 1.979 | 1.979 | 1.979 | 1.643 | 1.648 | 1.644 | 1.560 |
| RMSE | 2.634 | 2.634 | 2.634 | 2.634 | 2.350 | 2.350 | 2.347 | 2.233 | |
| WRMSE | 7.037 | 7.037 | 6.916 | 6.916 | 7.255 | 7.508 | 7.210 | 7.471 | |
| Model (potentially including KDE) | |||||||||
| HT | HAT | iSOUP | SGT | ||||||
| Dataset | Metric | HT | HAT | iSOUP | SGT | +KDE | +KDE | +KDE | +KDE |
| Abalone | MAE | 2.351 | 2.357 | 10.802 | 2.622 | 2.052 | 1.863 | 10.864 | 2.166 |
| RMSE | 3.175 | 3.186 | 11.249 | 4.051 | 2.776 | 2.633 | 11.318 | 3.138 | |
| WRMSE | 43.815 | 42.957 | 215.832 | 64.202 | 50.971 | 46.795 | 218.941 | 57.864 | |
| INVWRMSE | 16.944 | 17.018 | 58.676 | 21.451 | 14.539 | 13.826 | 58.988 | 16.430 | |
Decision Tree Learning
Decision tree learning builds hierarchical, rule-based models for prediction, balancing accuracy, computational cost, and interpretability.
See all 69 papers