Prevalidated ridge regression is a highly-efficient drop-in replacement for logistic regression for high-dimensional data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dempster, Angus, Webb, Geoffrey I., Schmidt, Daniel F.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910902102523904
author Dempster, Angus
Webb, Geoffrey I.
Schmidt, Daniel F.
author_facet Dempster, Angus
Webb, Geoffrey I.
Schmidt, Daniel F.
contents Logistic regression is a ubiquitous method for probabilistic classification. However, the effectiveness of logistic regression depends upon careful and relatively computationally expensive tuning, especially for the regularisation hyperparameter, and especially in the context of high-dimensional data. We present a prevalidated ridge regression model that closely matches logistic regression in terms of classification error and log-loss, particularly for high-dimensional data, while being significantly more computationally efficient and having effectively no hyperparameters beyond regularisation. We scale the coefficients of the model so as to minimise log-loss for a set of prevalidated predictions derived from the estimated leave-one-out cross-validation error. This exploits quantities already computed in the course of fitting the ridge regression model in order to find the scaling parameter with nominal additional computational expense.
format Preprint
id arxiv_https___arxiv_org_abs_2401_15610
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Prevalidated ridge regression is a highly-efficient drop-in replacement for logistic regression for high-dimensional data
Dempster, Angus
Webb, Geoffrey I.
Schmidt, Daniel F.
Machine Learning
Logistic regression is a ubiquitous method for probabilistic classification. However, the effectiveness of logistic regression depends upon careful and relatively computationally expensive tuning, especially for the regularisation hyperparameter, and especially in the context of high-dimensional data. We present a prevalidated ridge regression model that closely matches logistic regression in terms of classification error and log-loss, particularly for high-dimensional data, while being significantly more computationally efficient and having effectively no hyperparameters beyond regularisation. We scale the coefficients of the model so as to minimise log-loss for a set of prevalidated predictions derived from the estimated leave-one-out cross-validation error. This exploits quantities already computed in the course of fitting the ridge regression model in order to find the scaling parameter with nominal additional computational expense.
title Prevalidated ridge regression is a highly-efficient drop-in replacement for logistic regression for high-dimensional data
topic Machine Learning
url https://arxiv.org/abs/2401.15610