技术进展

REDI框架实现科学数据自动化AI就绪

Heooo 07月07日12时30分 2 阅读

「针对大型科学数据集在成为AI训练数据前需大量人工处理的痛点,研究团队推出开源框架REDI。该框架通过统一的五阶段流水线(摄取、预处理、转换、结构化、输出)实现自动化数据准备、就绪度评估、来源追踪和代理原生部署。配套工具SetGo可自动完成FAIR合规与目录发布。在气候、蛋白质组学、材料科学和核聚变四个领域验证中,REDI成功将所有原始数据转换为AI就绪状态,并在Frontier超算上展现出接近理想的百节点并行扩展能力。」

在科学计算与人工智能深度交融的时代,大型科学数据集的价值释放正面临一个关键瓶颈:数据准备。这些由领导级计算设施管理的海量数据,在成为AI训练数据之前,往往需要经历复杂且耗时的转换过程。然而,现有工具链缺乏一个能够统一自动化转换、就绪度评估、来源追踪和代理原生部署的完整框架。针对这一空白,研究团队在arXiv上发表了题为《Automated Data Readiness for Scientific AI》的论文,正式推出开源框架REDI。

REDI的核心设计理念是通过一个统一的五阶段流水线来解决数据准备难题。该流水线包括摄取、预处理、转换、结构化与输出五个阶段。每个阶段都配备了专门的仪表化工具,用于记录处理过程中的元数据,确保整个流程的可重复性。这种设计不仅让数据科学家能够清晰追溯每一步操作,还使得REDI能够作为一个“代理可调用技能”进行部署,从而无缝集成到更广泛的AI工作流中。

与REDI配套发布的还有工具SetGo,其核心功能是自动化实现FAIR(可发现、可访问、可互操作、可重用)合规性,并自动将处理后的数据集发布到目录中。这一组合让数据准备从一次性的、依赖专家经验的手工活,转变为标准化、可复用的社区资产。

为了验证REDI的跨领域适用性,研究团队在四个截然不同的科学领域进行了全面评估:气候科学、蛋白质组学、材料科学和核聚变研究。在每一个案例中,REDI都成功将原始数据转换为AI就绪状态,其输出结果经过了领域专家参考标准的严格验证。以气候科学案例为例,REDI在Frontier超级计算机上展现了接近理想的并行扩展能力,在100个计算节点上实现了近乎线性的性能提升,这证明了其处理极端规模数据集的潜力。

通过来源追踪仪表化分析,研究团队揭示了流水线中的性能瓶颈。分析显示,文件I/O操作是主导整个流水线成本的主要因素,而文件格式的选择则成为影响整体效率的首要优化杠杆。这一发现为未来进一步优化REDI性能指明了方向:通过智能选择数据格式和优化I/O策略,可以显著加快数据准备速度。

REDI的发布标志着科学AI领域数据准备流程的一次重要升级。它将原本繁琐、耗时且容易出错的数据转换过程,转变为一种可重复、可审计、可扩展的自动化操作。对于从事气候建模、蛋白质结构预测、新材料发现和核聚变模拟的研究人员而言,REDI提供了一个强大的工具,让他们能够将更多精力集中在模型设计和科学洞察上,而非数据清洗的泥潭中。随着科学AI对高质量训练数据的需求持续增长,像REDI这样的自动化数据就绪平台将成为加速科学发现的关键基础设施。

# REDI # 科学AI # 数据准备 # 开源框架 # FAIR合规

来源:Heooo AI工具导航