如何使用 torch.utils.checkpoint 梯度检查点技术以计算换空间训练超大模型
在训练深度学习模型,尤其是如Transformer这类拥有数百甚至数千层的超大模型时,GPU显存往往成为瓶颈。标准的反向传播算法需要存储前向传播中每层的所有中间激活值(Activations),以便在计算梯度时使用,这消耗了大量的显存。当模...
在训练深度学习模型,尤其是如Transformer这类拥有数百甚至数千层的超大模型时,GPU显存往往成为瓶颈。标准的反向传播算法需要存储前向传播中每层的所有中间激活值(Activations),以便在计算梯度时使用,这消耗了大量的显存。当模...
在深度学习领域,模型规模不断增大,对计算速度的要求也水涨船高。传统的FP32(单精度浮点数)运算虽然精度高,但计算量大、能耗高。为了解决这一问题,NVIDIA引入了专用的硬件加速单元——Tensor Core,并结合软件层面的混合精度(Mi...
在高风险应用场景,如信贷审批、招聘决策或司法判决中,AI模型的偏见(Bias)可能导致严重的社会不公和法律风险。因此,将公平性审计(Fairness Auditing)嵌入到模型部署和监控流程中,不再是一个“可选项”,而是一个基础设施的“必...
在现代人工智能,特别是大型语言模型(LLM)的训练中,GPU的核心计算能力(TFLOPS)固然重要,但常常被忽视的瓶颈是数据移动的速度。本文将详细解释GPU中使用的HBM(High Bandwidth Memory,高带宽显存)技术,以及为...
用公式和实例解释 GPU TFLOPS 如何计算,以及 FP32、FP16、BF16、FP8、FMA 和 Tensor Core 指标为什么不能直接横向比较。
作为一名搜索技术专家,理解Elasticsearch(简称ES)的近实时(NRT)搜索机制是掌握其高性能搜索能力的关键。NRT的实现核心在于Refresh操作。许多人误以为Refresh就是传统数据库中的“刷盘”或“提交”,但它在ES/Lu...
在深度学习模型训练和推理过程中,尤其是在使用GPU加速时,张量(Tensor)的内存管理是影响性能的关键因素。PyTorch张量的操作大致分为两类:返回“视图”(View)和返回“副本”(Copy/Clone)。不理解这两者的区别,可能导致...
在AI模型部署,尤其是涉及信贷、招聘或司法等高风险场景时,确保模型决策的公平性(Fairness)至关重要。传统的公平性指标(如群体平等机会或统计奇偶性)仅能发现群体偏差,但无法解释模型对特定个体的决策是否公平。 反事实分析(Counter...
向量搜索技术依赖高效的近似最近邻(ANN)算法来处理大规模数据集。其中,乘积量化(Product Quantization, PQ)因其卓越的压缩比和搜索速度而广受欢迎。然而,PQ是通过将高维向量拆分为多个子向量并独立量化来工作的,这一过程...
自动混合精度(Automatic Mixed Precision, AMP)训练是 PyTorch 1.6+ 版本引入的一项重要功能,它允许模型在训练过程中自动使用 FP16(半精度浮点数)进行计算,同时保留 FP32(单精度浮点数)来处理...