Categorical distance correlation under general encodings and its application to high-dimensional feature screening
Qingyang Zhang
TL;DR
This work extends distance correlation to categorical data by introducing general encodings (e.g., ordinal and semicircle) beyond one-hot, enabling $\mathrm{dCov}^{2}$ and $\mathrm{dVar}^{2}$ calculations that incorporate category spacings via $d^{X}_{ik}$ and $d^{Y}_{jl}$. It provides two sample estimators (maximum likelihood and a bias-corrected $U$-statistic) and derives their asymptotic null and alternative distributions using eigenstructures of encoding-specific matrices, facilitating p-value calculation and power analysis. The framework supports high-dimensional feature screening with sure screening guarantees under mild conditions and includes a data-driven change-point method for threshold selection. Simulations reveal that ordinal and semicircle encodings often outperform one-hot in monotone relationships, while one-hot can be advantageous for nonmonotone patterns; the method is demonstrated on the 2018 General Social Survey, identifying 32 relevant factors for subjective socioeconomic classification. Overall, the paper broadens the applicability of categorical distance correlation, offering principled encoding choices, rigorous theory, and practical screening tools for mixed nominal/ordinal data.
Abstract
In this paper, we extend distance correlation to categorical data with general encodings, such as one-hot encoding for nominal variables and semicircle encoding for ordinal variables. Unlike existing methods, our approach leverages the spacing information between categories, which enhances the performance of distance correlation. Two estimates including the maximum likelihood estimate and a bias-corrected estimate are given, together with their limiting distributions under the null and alternative hypotheses. Furthermore, we establish the sure screening property for high-dimensional categorical data under mild conditions. We conduct a simulation study to compare the performance of different encodings, and illustrate their practical utility using the 2018 General Social Survey data.
