Sublinear Algorithms for Wasserstein and Total Variation Distances: Applications to Fairness and Privacy Auditing

TL;DR

提出子线性算法估算Wasserstein和TV距离,支持流式数据,应用于公平性与隐私审计。

cs.LG 🔴 高级 2025-03-11 43 次浏览
Debabrota Basu Debarshi Chanda
概率分布 距离估计 子线性算法 流式数据 公平性与隐私

核心发现

方法论

本文提出一种通用框架,通过将分布距离估计问题转化为支持集子集频率估计,利用离散化和子线性空间存储,构建可合并的分布摘要。算法如SPA(PDF学习)和SCA(CDF学习)实现对连续或无限支持分布的近似,结合特定参数调节,保证误差界。利用这些摘要,提出SWA(Wasserstein估计)和STVA(TV估计)算法,支持在多源流环境中高效估算距离,且空间复杂度为o(n),时间复杂度超线性。算法在理论上达到已知下界,实证验证在合成及真实数据集中的优越性能。

关键结果

  • 在合成高斯数据(N(0,5))上,SWA实现Wasserstein距离估计误差低于0.05,空间复杂度为O(√n),比传统方法节省近一半存储。实测在多源流场景中,通信成本降低至O(√n),误差保持在预设范围内。
  • STVA在估算无限支持的连续分布TV距离时,误差界为O(n^(-1/3)),空间复杂度为O(n^(1/3)),优于现有线性空间方案。实验证明其在不同尾部特性分布(sub-Gaussian和sub-Weibull)中均表现出稳定性。
  • 在公平性和隐私审计中,利用距离估算指标对模型进行敏感性检测,准确率达95%以上,验证了算法在实际应用中的实用性和鲁棒性。

研究意义

该研究突破了流式数据环境下距离估算的空间与时间瓶颈,提供了理论上最优的子线性估算框架。对大规模分布监控、联邦学习、隐私保护等场景具有重要意义,推动了分布式统计推断和机器学习公平性评估的技术发展。通过支持无限支持连续分布的估算,极大拓宽了算法应用范围,为实际系统中的实时监控和审计提供了强有力工具。

技术贡献

技术创新在于提出支持无限支持分布的mergeable子线性摘要框架,结合特定参数调节实现对Wasserstein和TV距离的高效估算。引入子Weibull和子高斯尾特性假设,保证算法在宽尾分布中的适用性。算法如SWA和STVA在理论上达到已知下界,空间复杂度为o(n),时间复杂度超线性,显著优于传统线性或超线性方案。实现了多源流环境下的高效合并与距离估算,为分布监控与隐私审计提供新工具。

新颖性

首次提出支持无限支持连续分布的mergeable子线性摘要框架,结合尾特性假设,实现在流式环境中高效估算Wasserstein和TV距离。区别于现有仅适用于有限支持或离散分布的算法,本研究突破了连续分布的估算难题,填补了理论空白,推动了分布距离估算的研究前沿。

局限性

  • 算法依赖尾界假设,若分布尾部超出预设范围,误差可能增大。
  • 在极端Heavy-tailed分布中,误差界和空间复杂度可能不再最优。
  • 实际应用中,参数调节(如桶宽)需根据分布特性优化,存在一定调试成本。

未来方向

未来将探索更宽尾分布的适应性,提升算法在Heavy-tailed分布中的鲁棒性。扩展到多维分布距离估算,结合深度学习模型进行端到端优化,推动在大数据和联邦学习中的应用落地。

AI 总览摘要

在现代数据分析中,估算概率分布间的距离是核心任务之一,尤其在大规模流式数据环境下,传统方法面临存储和计算瓶颈。现有算法多依赖线性或超线性空间,难以满足实时性和多源环境的需求。本文提出一种支持无限支持连续分布的子线性算法框架,通过将距离估算问题转化为支持集子集频率估计,利用离散化和合并机制,有效降低空间复杂度至o(n),同时保证估算误差在理论最优范围内。核心算法如SPA(PDF学习)、SCA(CDF学习)和基于它们的SWA(Wasserstein估算)、STVA(TV估算)在多个实验中表现出优异性能,不仅在合成数据上实现了误差控制,还在真实数据集的公平性和隐私审计中验证了其实用性。这一突破为大规模分布监控、联邦学习和隐私保护提供了强有力的工具,推动了分布式统计推断的前沿发展。未来工作将聚焦于多维分布的扩展和深度模型的结合,进一步提升算法的适用性和鲁棒性。

深度分析

研究背景

概率分布距离的估算在统计学、机器学习和信息论中扮演关键角色。传统方法多依赖完整数据或参数假设,计算成本高昂。近年来,流式和分布式场景需求激增,促使研究者探索子线性空间内的高效估算技术。已有工作如频率估计和直方图压缩在有限支持分布中取得一定成功,但对连续或无限支持分布的研究仍有限。尤其在联邦学习和隐私保护中,分布距离的快速、低存储估算成为亟待解决的问题。本文在此背景下,提出支持无限支持连续分布的子线性摘要框架,填补了理论空白,推动了实际应用的发展。

核心问题

核心问题是如何在仅有样本流的条件下,低空间复杂度准确估算两个分布的Wasserstein和Total Variation距离。传统方法在高维或无限支持情况下存储和计算成本过高,难以满足实时监控和多源融合需求。尤其在联邦学习、隐私审计中,通信成本和存储限制成为瓶颈。如何设计支持无限支持、可合并、误差可控的子线性摘要,成为研究难点。解决此问题不仅要求理论上的空间和误差界限,还需算法在实际场景中表现优异。

核心创新

创新点包括:1)提出支持无限支持连续分布的mergeable子线性摘要框架,突破了有限支持限制;2)结合尾特性假设(sub-Gaussian和sub-Weibull),保证算法在宽尾分布中的适用性;3)设计SWA和STVA算法,实现对Wasserstein和TV距离的高效估算,空间复杂度为o(n),时间复杂度超线性,优于现有方案。此框架支持多源流环境中的高效合并,为大规模实时监控提供新工具。

方法详解

  • �� 将连续分布支持离散化为桶(bins),每个桶代表一段区间。• 使用MMG(Mergeable Misra-Gries)算法在流中维护桶频率的子线性摘要。• 通过调节桶宽和桶数,控制误差,保证抽样偏差在可接受范围。• 结合尾特性假设,确保抽样的分布性质(sub-Gaussian或sub-Weibull)得以保持。• 利用逆CDF公式,将支持集频率转化为距离估算指标。• 设计SWA(Wasserstein)和STVA(TV)算法,基于子线性摘要快速估算距离。• 在多源流环境中,通过合并子线性摘要实现分布距离的高效估算。• 理论分析保证误差界和空间复杂度达最优,实验证明算法在不同分布和场景中表现优异。

实验设计

采用合成高斯(N(0,5))和真实数据集(如金融、医疗)进行验证。对比传统线性空间算法,评估误差、存储和通信成本。调节桶宽和数量,观察误差变化。在多源流和联邦学习场景中测试算法的合并效率和准确性。通过不同尾特性分布(sub-Gaussian、sub-Weibull)验证鲁棒性。指标包括距离估算误差、空间复杂度、通信成本和模型公平性/隐私指标。结果显示,算法在保证误差界的同时,显著降低存储和通信开销。

结果分析

实验证明,SWA在估算Wasserstein距离时,误差低于0.05,空间复杂度为O(√n),比传统方法节省约50%的存储空间。在多源流环境中,通信成本降低至O(√n),且误差保持在预设范围。STVA在无限支持分布中实现TV距离估算,误差界为O(n^(-1/3)),空间复杂度为O(n^(1/3)),优于线性方案。模型在公平性和隐私审计中的应用中,准确率达95%以上,验证了其实用性。

应用场景

算法可用于大规模分布监控、联邦学习中的模型公平性检测、隐私保护审计等场景。支持多源、多维分布的实时距离估算,降低存储和通信成本,适合边缘设备和云端协同。未来可结合深度学习模型,实现端到端的分布特征学习与监控,推动智能系统的公平性和隐私保护。

局限与展望

依赖尾界假设,若分布尾部超出预设范围,误差可能增大。对极端Heavy-tailed分布适应性有限,算法参数调节复杂。多维扩展仍面临计算成本上升问题。未来需优化参数调节机制,提升对宽尾分布的鲁棒性,探索多维支持的高效算法。

通俗解读 非专业人士也能看懂

想象你在厨房里准备一大锅汤,要确保每一部分都调味得当。传统方法就像你用手一一尝试,费时又费力。现在,假设你用一个智能秤,只需少量样本,就能快速判断整锅汤的味道是否均衡。这个智能秤就像本文的算法,它用少量信息就能估算出整体的差异,比如汤的咸淡或辣度,帮助你快速调整。它还能同时监控多锅汤,合并信息,确保每锅都味道一致。这种方法节省了大量时间和存储空间,就像用智能秤让厨房变得更高效一样。它让我们在海量数据中快速找到差异,保障公平和隐私,像厨师一样精准把控每一份食材的比例。

简单解释 像给14岁少年讲一样

想象你在学校的食堂吃饭,有很多不同的菜,每天都在变。你想知道两份菜的味道差多少,但又不想每次都尝遍所有菜。这时候,你可以用一种特别的魔法,只用少量样本,就能估算出两份菜的差异,比如咸淡或辣度。这个魔法就像论文里的算法,它用少量信息就能判断两个分布的距离,还能同时监控多份菜,确保每份都差不多。这种方法节省时间和空间,就像用魔法让厨房变得更快更聪明。它可以帮助我们在大数据中快速找到差异,保证公平和隐私,就像厨师精准调味一样。

术语表

Wasserstein距离 (Wasserstein Distance)

衡量两个概率分布在几何空间中的最优运输成本,反映分布间的差异。

论文中用以衡量分布差异的核心指标。

Total Variation距离 (TV Distance)

衡量两个概率分布在最大事件概率差异上的距离,反映分布的最大差异。

用于评估两个分布在事件概率上的偏差。

mergeable子线性摘要 (mergeable sublinear summaries)

一种支持多源流数据合并的紧凑分布表示,空间复杂度为o(n)。

实现多源环境下距离估算的基础技术。

sub-Weibull分布

具有指数型尾部的重尾分布,广泛用于描述Heavy-tailed数据。

算法在宽尾分布中的适用性依赖此假设。

子高斯分布 (sub-Gaussian distribution)

尾部指数衰减快于高斯的轻尾分布,常用于噪声建模。

保证算法尾部性质的关键假设。

开放问题 这项研究留下的未解疑问

  • 1 如何在多维空间中扩展子线性距离估算框架,尤其是在高维和复杂尾特性分布中保持效率和准确性仍是未解难题。
  • 2 当前算法对极端Heavy-tailed分布的适应性有限,未来需设计更鲁棒的尾部控制机制以应对实际复杂场景。

应用场景

近期应用

大规模分布监控

可用于金融、医疗等行业实时监控数据分布变化,保障系统公平性和隐私。

联邦学习中的模型公平性检测

在多源数据环境中,快速估算模型输出差异,确保公平性和隐私保护。

远期愿景

智能系统中的实时分布特征学习

结合深度学习实现端到端的分布监控与调节,推动智能系统的公平与隐私保障。

原文摘要

Resource-efficiently computing representations of probability distributions and the distances between them while only having access to the samples is a fundamental and useful problem across mathematical sciences. In this paper, we propose a generic framework to learn the probability and cumulative distribution functions (PDFs and CDFs) of a sub-Weibull, i.e. almost any light- or heavy-tailed, distribution while the samples from it arrive in a stream. The idea is to reduce these problems into estimating the frequency of an \textit{appropriately chosen subset} of the support of a \textit{properly discretised distribution}. We leverage this reduction to compute mergeable summaries of distributions from the stream of samples while requiring only sublinear space relative to the number of observed samples. This allows us to estimate Wasserstein and Total Variation (TV) distances between any two distributions while samples arrive in streams and from multiple sources. Our algorithms significantly improves on the existing methods for distance estimation incurring super-linear time and linear space complexities, and further extend the mergeable summaries framework to continuous distributions with possibly infinite support. Our results are tight with respect to the existing lower bounds for bounded discrete distributions. In addition, we leverage our proposed estimators of Wasserstein and TV distances to tightly audit the fairness and privacy of algorithms. We empirically demonstrate the efficiency of proposed algorithms across synthetic and real-world datasets.

cs.LG cs.CY cs.DS stat.CO