Building Bridges between Regression, Clustering, and Classification
Machine Learning
2025-02-19 v2 Machine Learning
Abstract
Regression, the task of predicting a continuous scalar target y based on some features x is one of the most fundamental tasks in machine learning and statistics. It has been observed and theoretically analyzed that the classical approach, meansquared error minimization, can lead to suboptimal results when training neural networks. In this work, we propose a new method to improve the training of these models on regression tasks, with continuous scalar targets. Our method is based on casting this task in a different fashion, using a target encoder, and a prediction decoder, inspired by approaches in classification and clustering. We showcase the performance of our method on a wide range of real-world datasets.
Cite
@article{arxiv.2502.02996,
title = {Building Bridges between Regression, Clustering, and Classification},
author = {Lawrence Stewart and Francis Bach and Quentin Berthet},
journal= {arXiv preprint arXiv:2502.02996},
year = {2025}
}