Task-moment mismatch: diagnosing failures of coverage-based methods on heavy-tailed distributions in coreset selection
摘要
Coreset construction aims to identify a small, weighted subset of training data that approximates the full dataset’s statistical properties, thereby enabling efficient model training without significant performance loss. Existing coreset selection methods typically optimize for geometric coverage or mean matching, under the implicit assumption that these objectives transfer meaningfully to arbitrary downstream tasks. In this study, it is demonstrated that this assumption can fail when the downstream task depends on statistical moments that the selection objective often does not preserve. A common underlying phenomenon termed the task-moment mismatch is identified, wherein coreset objectives that neglect task-relevant statistical moments yield worse performance than uniform random sampling. The objective of this work is threefold: (1) to diagnose failure modes of coverage-based coreset selection on heavy-tailed distributions, (2) to characterize how optimizing for lower-order moments can degrade the preservation of higher-order statistical structure required by certain tasks, and (3) to provide a constructive demonstration that moment-aware selection can address this mismatch. A greedy selection procedure termed Hierarchical Moment-Preserving (HMP) selection is proposed, which targets specific moment tensors through a two-stage hierarchical approach. Experimental evaluation across synthetic distributions and realistic signals parameterized from real-world sources reveals that k-means++ exhibits