arXiv:2604.04736cs.LGcs.AI2026-04

用多卡并行采样,让贝叶斯神经网络训练更快更省资源。

Sampling Parallelism for Fast and Efficient Bayesian Learning

  • 将参数采样任务分到多张GPU上并行处理,降低内存压力。
  • 样本数与算力成比例扩展时,性能接近线性加速。
  • 配合数据并行可提升增强多样性,减少训练轮次。

机器学习模型,尤其是深度神经网络,在医疗、环境预测和金融等高风险领域日益普及,可靠预测不确定性量化至关重要。然而,许多不确定性量化(UQ)方法因计算成本过高而难以应用。基于采样的贝叶斯学习方法(如贝叶斯神经网络,BNN)尤其昂贵,因为生成和评估多个参数样本会迅速耗尽内存与计算资源。为此,我们提出采样并行性(sampling parallelism),一种简单但高效的并行策略,直接针对采样过程这一主要瓶颈。通过将样本评估分布到多个GPU上,该方法在无需架构修改或复杂调参的情况下,显著降低内存占用和训练时间。我们在多个任务与模型上验证其性能,并以分布式数据并行(DDP)为基准进行对比。进一步表明,采样并行性可与现有策略结合,形成混合并行方案。实验显示,当样本数随计算资源同比增加时,性能近乎完美扩展,证明样本评估可干净地并行化。尽管在固定工作负载下DDP获得更高原始加速比,但采样并行性的优势在于:每张GPU对同一批次应用独立随机增强,提升了增强多样性,从而减少收敛所需训练轮次。

原文摘要 · Abstract (English)

Machine learning models, and deep neural networks in particular, are increasingly deployed in risk-sensitive domains such as healthcare, environmental forecasting, and finance, where reliable quantification of predictive uncertainty is essential. However, many uncertainty quantification (UQ) methods remain difficult to apply due to their substantial computational cost. Sampling-based Bayesian learning approaches, such as Bayesian neural networks (BNNs), are particularly expensive since drawing and evaluating multiple parameter samples rapidly exhausts memory and compute resources. These constraints have limited the accessibility and exploration of Bayesian techniques thus far. To address these challenges, we introduce sampling parallelism, a simple yet powerful parallelization strategy that targets the primary bottleneck of sampling-based Bayesian learning: the samples themselves. By distributing sample evaluations across multiple GPUs, our method reduces memory pressure and training time without requiring architectural changes or extensive hyperparameter tuning. We detail the methodology and evaluate its performance on a few example tasks and architectures, comparing against distributed data parallelism (DDP) as a baseline. We further demonstrate that sampling parallelism is complementary to existing strategies by implementing a hybrid approach that combines sample and data parallelism. Our experiments show near-perfect scaling when the sample number is scaled proportionally to the computational resources, confirming that sample evaluations parallelize cleanly. Although DDP achieves better raw speedups under scaling with constant workload, sampling parallelism has a notable advantage: by applying independent stochastic augmentations to the same batch on each GPU, it increases augmentation diversity and thus reduces the number of epochs required for convergence.

贝叶斯学习并行计算神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。