Le, T., Nguyen, T., Yamada, M., Blanchet, J., & Nguyen, V. A. (2021). Adversarial Regression with Doubly Non-negative Weighting Matrices. ArXiv. /abs/2109.14875

View Publication

Abstract

Many machine learning tasks that involve predicting an output response can be solved by training a weighted regression model. Unfortunately, the predictive power of this type of models may severely deteriorate under low sample sizes or under covariate perturbations. Reweighting the training samples has aroused as an effective mitigation strategy to these problems. In this paper, we propose a novel and coherent scheme for kernel-reweighted regression by reparametrizing the sample weights using a doubly non-negative matrix. When the weighting matrix is confined in an uncertainty set using either the log-determinant divergence or the Bures-Wasserstein distance, we show that the adversarially reweighted estimate can be solved efficiently using first-order methods. Numerical experiments show that our reweighting strategy delivers promising results on numerous datasets.

Authors
Tam Le, Truyen Nguyen, Makoto Yamada, Jose Blanchet, Viet Anh Nguyen
Publication date
2021/12/6
Journal
Advances in Neural Information Processing Systems
Volume
34
Pages
16964-16976