基于加权融合惩罚的快速树推断

Fast Tree Inference With Weighted Fusion Penalties

Journal of Computational and Graphical Statistics · 2015
被引 24
ABS 3

中文导读

提出一种多维融合惩罚框架,用于在大量条件下对相似条件进行融合和聚合,以简化解释并提取数据主要特征。通过ℓ1和ℓ∞范数推导出同伦算法,可精确恢复整个树结构,并证明距离递减权重可产生平衡树结构。

Abstract

Given a dataset with many features observed in a large number of conditions, it is desirable to fuse and aggregate conditions that are similar to ease the interpretation and extract the main characteristics of the data. This article presents a multidimensional fusion penalty framework to address this question when the number of conditions are large. If the fusion penalty is encoded by an ℓq-norm, we prove for uniform weights that the path of solutions is a tree that is suitable for interpretability. For the ℓ1 and ℓ∞-norms, the path is piecewise linear and we derive a homotopy algorithm to recover exactly the whole tree structure. For weighted ℓ1-fusion penalties, we demonstrate that distance-decreasing weights lead to balanced tree structures. For a subclass of these weights that we call “exponentially adaptive,” we derive an O(nlog(n)) homotopy algorithm and we prove an asymptotic oracle property. This guarantees that we recover the underlying structure of the data efficiently both from a statistical and a computational point of view. We provide a fast implementation of the homotopy algorithm for the single feature case, as well as an efficient embedded cross-validation procedure that takes advantage of the tree structure of the path of solutions. Our proposal outperforms its competing procedures on simulations both in terms of timings and prediction accuracy. As an example we consider phenotypic data: given one or several traits, we reconstruct a balanced tree structure and assess its agreement with the known taxonomy. Supplementary materials for this article are available online.

统计学机器学习数据融合算法