Building Bridges between Regression, Clustering, and Classification

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Stewart, Lawrence, Bach, Francis, Berthet, Quentin
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929718679306240
author Stewart, Lawrence
Bach, Francis
Berthet, Quentin
author_facet Stewart, Lawrence
Bach, Francis
Berthet, Quentin
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2502_02996
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Building Bridges between Regression, Clustering, and Classification
Stewart, Lawrence
Bach, Francis
Berthet, Quentin
Machine Learning
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.
title Building Bridges between Regression, Clustering, and Classification
topic Machine Learning
url https://arxiv.org/abs/2502.02996