B2P-Corr 框架解决 PCC/CCC 相关性损失的小批量优化偏差问题
B2P-Corr: Batch-to-Population Gradient Estimators for Non-Decomposable Correlation Losses
训练回归模型要用 PCC 或 CCC 做损失的话,这篇给了在小批量下估计梯度不偏的方案,两个变体开销都很小,可以直接试。
arXiv 论文 B2P-Corr 针对 Pearson(PCC)与 Concordance(CCC)相关性系数在随机优化中不可分解的问题,提出用全局矩素描将相关性目标转为可分解伪损失。方法包含两个 O(B) 复杂度的变体:B2P-EMA 用指数移动平均加 stop-gradient 实现渐近梯度追踪,B2P-U 用跨批次 U 统计量估计器加蓄水池稳定化。论文量化了朴素小批量优化的 O(B^-1) 偏差,并给出 B2P-EMA 在快素描/慢参数机制下的移动目标追踪证明。在 2 个合成数据集和 3 个真实数据集上,B2P-Corr 以可忽略的开销取得比朴素小批量优化更高的相关性。
B2P-Corr: Batch-to-Population Gradient Estimators for Non-Decomposable Correlation Losses
Correlation metrics such as the Pearson Correlation Coefficient (PCC) and Concordance Correlation Coefficient (CCC) are standard regression metrics and learning objectives, but their non-decomposable nature poses a challenge for stochastic optimization. Since these metrics depend on dataset-level moments (mean, variance, covariance), naive mini-batch training with local statistics yields biased, high-variance gradients whose expectation does not align with the population gradient. To bridge this ``batch-to-population'' (B2P) gap, we propose \textbf{B2P-Corr}, a novel framework that transforms correlation objectives into decomposable pseudo-losses driven by lightweight global ``moment sketches''. We introduce two efficient variants: \textbf{B2P-EMA}, which utilizes exponential moving averages and stop-gradient operators to ensure asymptotic gradient tracking with $O(B)$ cost ($B$ is batch size); and \textbf{B2P-U}, which uses sampled cross-batch U-statistic estimators and a reservoir-based stabilization with an $O(B)$ implementation. Theoretically, we quantify the $O(B^{-1})$ bias of naive mini-batch optimization, prove a moving-target tracking result for B2P-EMA under a fast-sketch / slow-parameter regime, and provide an idealized control-variate analysis for B2P-U. Experiments on two synthetic and three real-world datasets demonstrate B2P-Corr achieves stable optimization and higher correlation than naive mini-batch optimization with negligible overhead.