Skip to main content

Gradients weights improve regression and classification

Author(s): Kpotufe, S; Boularias, A; Schultz, T; Kim, K

To refer to this page use:
Abstract: In regression problems over ℝd, the unknown function f often varies more in some coordinates than in others. We show that weighting each coordinate i according to an estimate of the variation of f along coordinate i -- e.g. the L1 norm of the ith-directional derivative of f -- is an efficient way to significantly improve the performance of distance-based regressors such as kernel and k-NN regressors. The approach, termed Gradient Weighting (GW), consists of a first pass regression estimate fn which serves to evaluate the directional derivatives of f, and a second-pass regression estimate on the re-weighted data. The GW approach can be instantiated for both regression and classification, and is grounded in strong theoretical principles having to do with the way regression bias and variance are affected by a generic feature-weighting scheme. These theoretical principles provide further technical foundation for some existing feature-weighting heuristics that have proved successful in practice. We propose a simple estimator of these derivative norms and prove its consistency. The proposed estimator computes efficiently and easily extends to run online. We then derive a classification version of the GW approach which evaluates on real-worlds datasets with as much success as its regression counterpart.
Publication Date: 2016
Citation: Kpotufe, Samory, Abdeslam Boularias, Thomas Schultz, and Kyoungok Kim. "Gradients weights improve regression and classification." The Journal of Machine Learning Research 17, no. 22 (2016): pp. 1-34.
ISSN: 1532-4435
EISSN: 1533-7928
Pages: 1 - 34
Type of Material: Journal Article
Journal/Proceeding Title: Journal of Machine Learning Research
Version: Final published version. Article is made available in OAR by the publisher's permission or policy.

Items in OAR@Princeton are protected by copyright, with all rights reserved, unless otherwise indicated.