cs.LGFeb 24, 2026

High-Dimensional Robust Mean Estimation with Untrusted Batches

Authors: Maryam Aliakbarpour, Vladimir Braverman, Yuhan Liu, Junze Yin

Organizations: Department of Computer Science, Ken Kennedy Institute, Rice University. · Johns Hopkins University. · Rice University.

Abstract

We study high-dimensional mean estimation in a collaborative setting where data is contributed by NN users in batches of size nn. In this environment, a learner seeks to recover the mean μμ of a true distribution PP from a collection of sources that are both statistically heterogeneous and potentially malicious. We formalize this challenge through a double corruption landscape: an ε\varepsilon-fraction of users are entirely adversarial, while the remaining ``good'' users provide data from distributions that are related to PP, but deviate by a proximity parameter αα. Unlike existing work on the untrusted batch model, which typically measures this deviation via total variation distance in discrete settings, we address the continuous, high-dimensional regime under two natural variants for deviation: (1) good batches are drawn from distributions with a mean-shift of α\sqrtα, or (2) an αα-fraction of samples within each good batch are adversarially corrupted. In particular, the second model presents significant new challenges: in high dimensions, unlike discrete settings, even a small fraction of sample-level corruption can shift empirical means and covariances arbitrarily. We provide two Sum-of-Squares (SoS) based algorithms to navigate this tiered corruption. Our algorithms achieve the minimax-optimal error rate O(ε/n+d/nN+α)O(\sqrt{\varepsilon/n} + \sqrt{d/nN} + \sqrtα), demonstrating that while heterogeneity αα represents an inherent statistical difficulty, the influence of adversarial users is suppressed by a factor of 1/n1/\sqrt{n} due to the internal averaging afforded by the batch structure.

Figures & tables

Explore similar work

CardsList
  1. Sample Complexity Bounds for Robust Mean Estimation with Mean-Shift Contamination

    Feb 25, 2026Ilias Diakonikolas, Giannis Iakovidis, Daniel M. Kane +1Robust Mean EstimationRobust Estimation

  2. High-Dimensional Gaussian Mean Estimation under Realizable Contamination

    Mar 17, 2026Ilias Diakonikolas, Daniel M. Kane, Thanasis PittasData ContaminationStatistical Learning Theory

  3. Mean Estimation from Coarse Data: Characterizations and Efficient Algorithms

    Feb 26, 2026Alkis Kalavasis, Anay Mehrotra, Manolis Zampetakis +2Query Identifiability