Wasserstein logistic regression with mixed features
File(s)2205.13501-2.pdf (1.2 MB)
Accepted version
Author(s)
Selvi, Aras
Belbasi, Reza
Haugh, Martin
Wiesemann, Wolfram
Type
Conference Paper
Abstract
Recent work has leveraged the popular distributionally robust optimization
paradigm to combat overfitting in classical logistic regression. While the resulting
classification scheme displays a promising performance in numerical experiments,
it is inherently limited to numerical features. In this paper, we show that distri-
butionally robust logistic regression with mixed (i.e., numerical and categorical)
features, despite amounting to an optimization problem of exponential size, ad-
mits a polynomial-time solution scheme. We subsequently develop a practically
efficient column-and-constraint approach that solves the problem as a sequence of
polynomial-time solvable exponential conic programs. Our model retains many
of the desirable theoretical features of previous works, but—in contrast to the
literature—it does not admit an equivalent representation as a regularized logistic
regression, that is, it represents a genuinely novel variant of logistic regression.
We show that our method outperforms both the unregularized and the regularized
logistic regression on categorical as well as mixed-feature benchmark instances.
paradigm to combat overfitting in classical logistic regression. While the resulting
classification scheme displays a promising performance in numerical experiments,
it is inherently limited to numerical features. In this paper, we show that distri-
butionally robust logistic regression with mixed (i.e., numerical and categorical)
features, despite amounting to an optimization problem of exponential size, ad-
mits a polynomial-time solution scheme. We subsequently develop a practically
efficient column-and-constraint approach that solves the problem as a sequence of
polynomial-time solvable exponential conic programs. Our model retains many
of the desirable theoretical features of previous works, but—in contrast to the
literature—it does not admit an equivalent representation as a regularized logistic
regression, that is, it represents a genuinely novel variant of logistic regression.
We show that our method outperforms both the unregularized and the regularized
logistic regression on categorical as well as mixed-feature benchmark instances.
Date Acceptance
2022-09-15
Citation
Advances in Neural Information Processing Systems, 35
ISBN
9781713871088
ISSN
1049-5258
Publisher
NeurIPs
Journal / Book Title
Advances in Neural Information Processing Systems
Volume
35
Copyright Statement
© 2022 The Author(s).
Source
NeurIPS 2022
Publication Status
Published
Start Date
2022-11-28
Finish Date
2022-12-09
Coverage Spatial
New Orleans, LA, USA