当前位置:首页 > 报告详情

Grace Blackwell 上使用 Spark Connect ML 扩展 XGBoost.pdf

上传人: Fl****zo 编号:718645 2025-06-22 65页 3.12MB

1、Forward-looking StatementThis presentation has been prepared for informational purposes only.The information set forth herein does not purport to be complete or contain all relevant information.Statements contained herein are made as of the date of this presentation unless stated otherwise.This pres

2、entation and the accompanying oral commentary may contain forward-looking statements.In some cases,forward-looking statements can be identified by terms such as“may”,“will”,“should”,“expects”,“plans”,“anticipates”,“could”,“intends”,“projects”,“believes”,“estimates”,“predicts”,or“continue”,or the neg

3、ative of these words or other similar terms or expressions that concern Databricks expectations,strategy,plans,or intentions.Forward-looking statements are based on information available at the time those statements are made and are inherently subject to risks and uncertainties that could cause actu

4、al results to differ materially from those expressed in or suggested by the forward-looking statements.Forward-looking statements should not be read as a guarantee of future performance or outcomes.Except as required by law,Databricks does not undertake any obligation to publicly update or revise an

5、y forward-looking statement,whether as a result of new information,future developments or otherwise.2Scaling XGBoost With Spark Connect ML on Grace BlackwellJiaming Yuan,Bobby Wang4/12/2025Why Are We Here?Main theme of the talkScalingScalingScalingWe are here to scaleXGBoost trainingAgendaIntroducti

6、onExternal Memory Model TrainingScaling with NVLink-C2CJiaming YuanSpark Connect MLRun XGBoost JVM package over spark connectBobby WangPart IIntroduction to the Out-of-Core XGBoostJiaming YuanIntroductionXGBoostIntroductionGradient boosting decision treesSupports multiple target typesRegressionClass

word格式文档无特别注明外均可编辑修改,预览文件经过压缩,下载原文更清晰!
三个皮匠报告文库所有资源均是客户上传分享,仅供网友学习交流,未经上传用户书面授权,请勿作商用。
本文主要内容是介绍如何使用Databricks的Spark Connect ML和NVLink-C2C技术来扩展XGBoost算法在GPU上的训练。关键点如下: 1. XGBoost适用于多种类型的目标,如回归、分类等,并支持类别特征。 2. 使用外部内存模型训练(Out-of-Core)来处理大于GPU内存的数据集。 3. NVLink-C2C技术提供了CPU和GPU之间的高带宽连接,显著减少数据传输开销。 4. 在单个GPU上,使用NVLink-C2C技术的XGBoost可训练的最大数据集达1.5TB。 5. Spark Connect ML允许在Spark集群上运行JVM-based的XGBoost,实现了无需修改代码即可在CPU和GPU集群上运行。 6. 性能对比显示,GPU训练时间仅为CPU的1/20,成本效益显著。 核心数据引用: - 最大数据集训练:1TB以上。 - GPU训练时间:13.4秒。 - CPU训练时间:275秒。 - 成本对比:具体数值未给出,但强调GPU更具成本效益。
"如何规模化XGBoost训练?" "Spark Connect ML如何助力XGBoost?" "NVLink-C2C技术如何提升GPU计算?"
客服
商务合作
小程序
服务号
折叠