Gradient Aligned Regression via Pairwise Losses
Dixian Zhu, Tianbao Yang, Livnat Jerby
Abstract
Regression is a fundamental task in machine learning that has garnered extensive attention over the past decades. The conventional approach for regression involves employing loss functions that primarily concentrate on aligning model prediction with the ground truth for each individual data sample. Recent research endeavors have introduced novel perspectives by incorporating label similarity into regression through the imposition of additional pairwise regularization or contrastive learning on the latent feature space, demonstrating their effectiveness. However, there are two drawbacks to these approaches: (i) their pairwise operations in the latent feature space are computationally more expensive than conventional regression losses; (ii) they lack theoretical insights behind these methods. In this work, we propose GAR (Gradient Aligned Regression) as a competitive alternative method in label space, which is constituted by a conventional regression loss and two pairwise label difference losses for gradient alignment including magnitude and direction. GAR enjoys: i) the same level efficiency as conventional regression loss because the quadratic complexity for the proposed pairwise losses can be reduced to linear complexity; ii) theoretical insights from learning the pairwise label difference to learning the gradient of the ground truth function. We limit our current scope as regression on the clean data setting without noises, outliers or distributional shifts, etc. We demonstrate the effectiveness of the proposed method practically on two synthetic datasets and on eight extensive real-world tasks from six benchmark datasets with other eight competitive baselines. Running time experiments demonstrate the superior efficiency
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on9
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Deep symbolic regression: Recovering mathematical expressions from data via risk-seeking policy gradientsBrenden K. Petersen, Mikel Landajuela, T. Nathan Mundhenk, Cláudio Prata Santiago et al.ICLR 2021 · 444 citations
- Delving into Deep Imbalanced RegressionYuzhe Yang, Kaiwen Zha, Ying-Cong Chen, Hao Wang et al.ICML 2021 · 385 citations
- Balanced MSE for Imbalanced Visual RegressionJiawei Ren, Mingyuan Zhang, Cunjun Yu, Ziwei LiuCVPR 2022 · 163 citations
- Rank-N-Contrast: Learning Continuous Representations for RegressionKaiwen Zha, Peng Cao, Jeany Son, Yuzhe Yang et al.NeurIPS 2023 · 129 citations
Related papers
- ACCon: Angle-Compensated Contrastive Regularizer for Deep RegressionBotao Zhao, Xiaoyang Qu, Zuheng Kang, Junqing Peng et al.AAAI 2025
- ConR: Contrastive Regularizer for Deep Imbalanced RegressionMahsa Keramati, Lili Meng, R. David EvansICLR 2024 · 22 citations
- Advancing Semantic Textual Similarity Modeling: A Regression Framework with Translated ReLU and Smooth K2 LossBowen Zhang, Chunping LiEMNLP 2024 · 1 citation
- Unbiased Classification through Bias-Contrastive and Bias-Balanced LearningYoungkyu Hong, Eunho YangNeurIPS 2021 · 94 citations
- PRIME: Deep Imbalanced Regression with ProxiesJongin Lim, Sucheol Lee, Daeho Um, Sung-Un Park et al.ICML 2025
