English

Improving Deep Regression with Tightness

Machine Learning 2025-02-14 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

For deep regression, preserving the ordinality of the targets with respect to the feature representation improves performance across various tasks. However, a theoretical explanation for the benefits of ordinality is still lacking. This work reveals that preserving ordinality reduces the conditional entropy H(ZY)H(Z|Y) of representation ZZ conditional on the target YY. However, our findings reveal that typical regression losses do little to reduce H(ZY)H(Z|Y), even though it is vital for generalization performance. With this motivation, we introduce an optimal transport-based regularizer to preserve the similarity relationships of targets in the feature space to reduce H(ZY)H(Z|Y). Additionally, we introduce a simple yet efficient strategy of duplicating the regressor targets, also with the aim of reducing H(ZY)H(Z|Y). Experiments on three real-world regression tasks verify the effectiveness of our strategies to improve deep regression. Code: https://github.com/needylove/Regression_tightness.

Keywords

Cite

@article{arxiv.2502.09122,
  title  = {Improving Deep Regression with Tightness},
  author = {Shihao Zhang and Yuguang Yan and Angela Yao},
  journal= {arXiv preprint arXiv:2502.09122},
  year   = {2025}
}

Comments

ICLR 2025, Code: https://github.com/needylove/Regression_tightness

R2 v1 2026-06-28T21:42:49.684Z