Databricks UCDataset GPU 训练省去表导出步骤
Databricks 于 11 月 9 日推出测试版 UCDataset,旨在缓解 GPU 训练延迟,团队无需先导出其数据表。Databricks UCDataset GPU 训练将其数据治理系统 Unity Catalog 与机器学习框架 PyTorch 连接起来。该公司称,在一项模拟推荐工作负载中,涉及一张 300 GB 特征表和八块 NVIDIA H100 GPU,其训练速度提升了 10 倍。该结果比较的是其加载器与导出的 Parquet 副本,后者是以列式文件格式存储的独立数据集。
眼下的比较对象,是直接访问数据表与单独的导出流水线。Databricks 现有的 AI Runtime 指引也建议,在通过机器学习软件生态系统 Hugging Face 加载大型数据表之前先将其导出。这使该公告具备了一个具体参照点:Databricks 自身内部已经确立的工作流。
对于机器学习工程师和平台负责人而言,实际问题是,取消这一导出步骤是否也能简化可靠的生产训练。答案取决于工作负载、访问策略和恢复要求。更快的输入路径值得关注,但团队仍需获得可比的模型结果,并保留能够说明所用数据的可靠记录。
Databricks UCDataset GPU 训练让数据表成为直接输入
此次发布改变的是受治理数据表与训练任务之间的连接方式,而非模型架构。
Databricks 的 UCDataset 测试版公告介绍了直接从云存储读取、在训练进程间自动分发,以及与现有 PyTorch 循环的集成。初始版本支持在 Unity Catalog 中注册的 Delta 表;这类表通过事务历史来组织数据文件。
该公开公告于 11 月 9 日发布。该公司的 11 月发布说明则将这项测试版功能列在 11 月 6 日。这些日期区分了产品更新与后续的博客公告,但两者都不改变其测试版状态。
其架构意义在于移除了中间数据产物。导出工作流在数据团队维护的数据表与训练代码消费的文件之间建立了一道边界。团队必须决定这些文件何时刷新、由哪个任务生成,以及准备过程失败时如何处理。
直接访问数据表改变了这道边界。训练应用可以通过数据表识别输入,而不是依赖单独维护的导出位置。对于符合条件的工作负载,这提供了淘汰准备代码及其相关运营决策的机会。
但这一机会应以比“训练不再需要准备工作”更审慎的方式理解。特征仍需恰当定义,标签仍需验证,实验仍需商定纳入标准。更直接地读取数据,并不能证明这些数据正确代表了问题。
设想一个推荐团队正在评估预测模型的改动。其工程师希望在保持输入样本群体稳定的同时比较模型行为。取消导出可以缩短达成这一比较的路径,但若同时改变样本群体,结果将更难解读。
这一差别很重要,因为更简单的基础设施和更简单的实验是两种不同结果。团队可以减轻基础设施负担,却意外使实验控制变得更弱;也可以在移除冗余准备阶段的同时,保留严谨的实验控制。
因此,对该版本最恰当的理解是具体而有限的。Databricks 正在为熟悉的“数据表到文件”交接方式提供集成式替代方案。这一替代方案是否改善整个工作流,取决于现有交接目前实现了什么。
对于导出主要是为了适配训练读取器的团队而言,替换看起来颇具吸引力。对于将导出用作获批实验产物的团队,采用则需要有意地为这一控制措施寻找替代方案。
导出流水线让平台团队承受压力
受影响最大的团队,是那些仅为实现训练而维护第二条数据路径的团队。
Databricks 的 AI Runtime 数据加载文档介绍了针对大型数据表的导出式方法。它还区分了数据表输入与存储在 Unity Catalog volumes(其受治理文件容器)中的文件输入。新的测试版在这一既有环境中增加了另一种选择。
导出流水线的职责不止于生成可读取的文件。其所有者需要决定命名规范、保留策略、刷新触发条件、故障处理和访问权限。即使是规模不大的工作流,只要多个实验消费其输出,也可能成为共享依赖。
这次发布首先给负责这些依赖的平台工程师带来压力。他们现在有理由审视哪些环节仍然必要。当最初的兼容性问题已有另一种已文档化的解决方案时,保留导出就需要更清晰的理由。
这并不意味着每一次导出都是浪费。经过有意冻结的数据集可支持外部协作、受控审批或特定训练栈。这些都是实质性要求。若某项导出只是因为没人重新审视旧有加载器决策,则应作出不同评估。
短期内应采取的措施是审查工作流,而不是立即承诺迁移。平台负责人需要了解其准备任务除格式转换以外还提供了什么。否则,他们可能会在删除冗余工作的同时,也删除有用的控制措施。
模型开发者面临相关压力。当输入连接更容易被替换时,训练代码中隐藏的假设会更加显眼。依赖特定文件排列方式的项目,可能比其负责人最初预期的需要更多调整。
预算负责人也需要进行比加速器运行时间更广泛的比较。导出工作流会在训练前后消耗工程关注。只要能移除有意义的维护工作并保留所需行为,替代方案即使未达到头条基准结果,也可能改善运营。
相反的结果同样可能出现。加载器可以改善某个已测量的训练阶段,却在其他环节增加迁移工作或恢复复杂性。这不会否定技术改进本身,但会改变特定团队的采用决策。
因此,有效评估应将持续收益与过渡成本分开。团队应记录哪些任务消失、哪些职责转移,以及哪些检查仍然必要。这会使运营理由可供审查,而非依赖一种笼统的简化承诺。
对于正在评估符合条件的数据表工作负载的现有 Databricks 用户而言,这种压力是即时的。更广泛的平台决策则需要更多证据。仅凭一个测试版加载器,并不能证明组织应迁移其数据或替换训练框架。
实际变化在于,导出流水线现在需要证明自身存在的价值。其价值应源于明确要求,而不是历史惯性。
其机制是一条更短、交接更少的路径
技术主张是减少数据表存储与分布式训练之间的协调工作。
分布式训练将模型工作拆分到多个进程中。其输入系统必须将预期样本提供给每个进程,且不发生意外重复。PyTorch 的 可迭代数据集文档警告,简单的多进程处理可能导致数据重复,因为工作进程会获得数据集副本。
这一警告说明,加载器维护不只是读取文件。训练任务可能成功执行,却消费了非预期的样本分布。没有抛出异常,并不能证明实验正确处理了预期数据。
Databricks 表示,UCDataset 的自动分区处理了这类协调工作。其工程吸引力在于减少了应用特定的输入逻辑。团队随后可将检查重点放在最终行为和平台契约上。
应在产品改变的边界上评估其收益。如果任务在等待样本,更好的输入路径就有改善空间;如果模型计算已经占主导地位,减少输入工作对总训练耗时的影响就会更小。
这也是为何基准测试的工作负载很重要。包含大型特征表的推荐任务描述了一种可识别的用例,但并不能证明每种模型、特征表示或训练配置都会获得相同结果。
与 Ray(一个分布式应用框架)也存在有用的对比。其 Databricks 表读取器接受仓库标识符,并使用 Databricks statement execution API。它为已经围绕 Ray 组织的应用提供了另一条集成路径。
这一比较揭示的是设计差异,而不是谁更优。基于仓库的读取器可以适合围绕查询执行构建的工作流。直接表读取器则可以适合已准备好所选训练数据的工作负载。
相关决策在于,每个团队希望将准备与执行职责置于何处。转移职责可以减少交接,但也可能暴露此前由上游处理的假设。公平的比较应让这些职责保持可见。
例如,假设某个训练导出包含决定哪些观测值符合条件的业务规则。用数据表引用替换其输出,并不会自动保留这些规则。工程师必须确认等效筛选发生在何处,以及审查者如何检查它。
同样地,之前接收标准化特征表示的训练代码可能依赖于该表示。新的输入连接应依据相同契约进行评判。否则,性能比较也会变成一次未被承认的数据处理比较。
Databricks UCDataset GPU 训练最适合被理解为一种针对性的输入路径变更。它让团队有理由质疑不必要的交接,同时保留使训练结果值得信赖的检查。
更广泛的教训是架构性的:每一个被移除的阶段,都应说明其职责由何处承担。当剩余组件仍覆盖重要工作时,更短的流水线才有价值。
基准测试需要在工作负载层面解读
所报告的提速结果支持进一步调查,但生产采用仍需来自目标工作负载的证据。
Databricks 将这一头条结果描述为针对模拟生产工作负载的公司基准测试。该公告未提供独立验证。在内部讨论该结果或比较潜在基础设施变更时,读者应保留这一差别。
基准测试回答的是特定配置下一个有边界的问题。采用决策则问的是,同一种方法是否能改善另一种运行环境。这两个问题有所重叠,但不能互相替代。
有意义的评估应保持模型、输入样本群体和质量标准一致。否则,表面上的改进可能只是工作发生了变化。若实验使用了不同样本,或采用不同标准评判结果,执行速度更快的意义便有限。
比较还需要一个定义明确的起点。从训练开始时计量回答的是一个问题;从可用源数据到达、经由完整流程得到已评估模型来计量,则回答的是另一个问题。
两种计量都可能有用。前者隔离训练行为,后者反映用户实际经历的工作流程。分别报告它们,能够避免将局部改进误认为同等幅度的运营改进。
基线质量同样重要。团队应与其在生产环境中能够合理维持的导出流程进行比较。一个不必要地薄弱的基线会让新加载器看起来更好,却无法澄清工程师实际面临的决策。
重复执行很重要,因为运行条件会变化。一次成功运行可以发现一个有前景的方向,但对于日常行为的证据有限。评估应包含足够多的观测,以区分可靠的改进与异常有利的一次执行。
模型质量也应纳入该评估。加载器是训练系统的一部分,因此数据顺序或消费方式的变化值得关注。只有在确信所得实验仍有意义的前提下,可比的耗时才有参考价值。
这些是评估原则,并非声称该 beta 未能满足这些原则。现有公告提供了测试的理由,但并未免除团队在测试开始前定义成功标准的必要性。
买方也不应将训练速度比率直接换算为预算预测。实际支出取决于资源分配、周边服务以及运行管道所需的工作。这种关系需要在组织自身的环境中进行测量。
即使无法复现宣传中的数字,商业价值仍可能令人信服。淘汰脆弱的数据准备任务,或让实验输入更容易审查,都可能具有实际价值。这些成果应被明确记录,而非隐藏在性能口号背后。
对于 Databricks UCDataset GPU 训练,可信的采用理由应结合耗时、可比的模型结果和减少的运维工作。这种组合比单独考察速度数字更有用。
Beta 限制决定生产环境决策
在性能测试开始之前,兼容性和实验控制就可能决定是否具备资格。
UCDataset 文档列出了具体限制。不支持带有行过滤器或列掩码的表,也不支持视图、物化视图、流式表、浅克隆,以及包含 VARIANT 列的表。
这些限制使预期的输入对象成为实际的起点。团队应先确认其获批的训练数据是否可通过受支持的表获取。在检查这一条件之前评估速度,可能会浪费精力。
策略限制尤其值得关注。行过滤器限制某个身份可查看的记录,而列掩码会隐藏或转换受保护的值。依赖这些控制措施的组织,在考虑 beta 之前需要获得批准的兼容方案。
这是一项治理决策,而不是性能调整。基准测试结果并不能证明削弱访问策略是合理的。相关利益相关方需要就训练身份被允许使用哪些数据达成一致。
实验控制则是另一项议题。文档支持选择数值型表版本。这为识别特定输入状态提供了方法,无需为每个实验专门创建导出文件。
然而,记录该状态仍是团队的责任。模型研究需要在结果与输入之间建立可辩护的关联。如果表在两次执行之间发生变化,仅记录表名并不足够。
清晰的记录应将选定的输入状态与模型配置及评估决策关联起来。这并不意味着要把每一次实验都变成文档项目,而是要保留足够的背景信息,让另一位工程师能够理解结果。
维护这些背景信息的团队,可以从用于技术文档和实验讨论的可搜索知识库中受益。这类文档可与训练系统自身的记录共同支持审查。
该 beta 还会在各个 epoch 中使用相同的打乱文件顺序,也就是对数据集进行完整遍历。它不提供可配置的随机打乱种子。因此,对顺序有特定要求的团队需要评估这一行为是否适合其实验。
恢复能力是另一项明确边界:加载器无法对读取位置进行检查点保存,因此无法在一个 epoch 中途恢复。保存训练状态的模型检查点并不能独立解决该输入位置的限制。
其运营影响取决于工作负载。对一种应用而言,重新开始输入消费可能可以接受;对另一种应用而言,则可能成本高昂。当恢复行为会实质影响交付时,团队应将中断执行纳入评估。
数组列会以 Python 列表形式到达,且文档说明其速度低于数值列。因此,测试性能时,代表性的特征数据很重要。一个方便的测试表未必反映预期模型使用的数据表示。
综合来看,这些限制界定了采用范围。它们为工程师和治理负责人提供了具体问题,而无需臆测产品是否已适用于每一种企业工作流。
适当的决策可能是采用、有限试点,或继续使用既有路径。只要与明确的工作负载要求相关联,每一种选择都有其合理性。
2027 年初前值得关注的事项
下一批有价值的证据将涉及可复现的结果、兼容的治理方式和可靠的恢复能力。
第一个信号是其他团队能够评估的详细工作负载比较。强有力的证据应说明基线、输入特征、训练配置和质量标准,也应解释耗时从何处开始计量。
此类报告会强化开展更广泛评估的理由。它能帮助工程师识别与自身相似的工作负载,避免从无关示例中外推。来自现有生产用户的结果,也会在模拟工作负载之外提供更多背景。
即使证据显示改进集中于特定配置,它仍然有价值。这会缩小产品最适用的用例范围,并使采用决策更精确。有边界的结果比无限制的主张更具可操作性。
第二个信号是受策略控制表的兼容性进展。支持组织已依赖的控制措施,将扩大符合资格的工作负载范围。关于获批替代方案的详细指导,也会帮助团队评估现实可行的路径。
这一信号之所以重要,是因为数据访问要求先于加载器选择。组织不能在批准迁移后,再将策略兼容性视为次要问题。若资格范围仍然狭窄,采用将继续集中于兼容数据集。
第三个信号是恢复能力和数据顺序控制方面的进展。可恢复性或可配置排序的文档化改进,将解决 beta 中可见的具体顾虑。详细的运营示例也将阐明团队当前如何应对这些要求。
这些发展将强化适用于以下训练任务的理由:中断处理或实验可重复性会驱动基础设施选择。如果这些方面保持不变,团队应继续在这些边界内评估该 beta。
观察窗口是未来数月,而不是已承诺的发布计划。Databricks 尚未确定本文讨论的未来交付日期。这些是应监测的具体信号,而非归因于该公司的承诺。
更广泛的采用问题在于,直接读取表是否会成为常规的平台选择。这需要一个可重复的运营案例:兼容的访问方式、可理解的输入、可靠的执行和有用的性能。没有任何单一基准能够证明所有这些条件。
对于当前用户而言,此次发布已带来实际决策。他们可以识别那些主要为满足训练兼容性而维护的导出文件,并考察该 beta 是否消除了这一要求。结果应指导具体工作流,而非形成对整个通用平台的判断。
从一个符合条件、且既有导出流程已被充分理解的训练工作负载开始。在改变其输入路径之前,先定义比较方式,包括数据身份、模型质量、耗时和中断行为。在审查结果期间保留既有工作流。询问平台负责人哪些维护职责会消失,并询问模型负责人该实验是否仍具可比性。Databricks UCDataset GPU 训练提供了一个简化基于表训练的具体机会,但其价值将在这些有记录的决策中最为清晰。您当前管道中的哪一项导出,是因为训练读取器需要它而存在?又需要哪些证据,才能证明可以将其退役?



