Guardado en:
Detalles Bibliográficos
Autores principales: Stewart, Lawrence, Bach, Francis, Berthet, Quentin
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:https://arxiv.org/abs/2502.02996
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_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