中文

kindling：一个用于在R中生成、训练和调优神经网络的高级torch接口

统计计算 2026-07-04 v1
作者: Antoine Soetewey , Joshua Marie
在 arXiv 上查看 ↗ PDF ↗

摘要

{kindling}是一个R包，为{torch}（R中PyTorch的原生实现）提供了用于定义、训练和调优神经网络的高级接口。它支持多层感知机和循环架构（RNN、LSTM、GRU），同时减少了手工编写{torch}模型定义和训练循环所需的样板代码。该包围绕三个抽象层次组织：返回可检查、未求值的{torch::nn\_module()}表达式的代码生成函数；直接从公式和数据框拟合模型的直接训练函数；以及{tidymodels}注册的模型规格，使得神经网络可以使用{tidymodels}用户在其他模型类型中已经依赖的{tune}、{dials}、{recipes}和{workflows}基础设施进行拟合、调优和评估。这种设计使分析人员无需离开{tidymodels}生态系统即可从探索性训练转向系统性超参数搜索，同时保留检查或修改生成模型代码的能力，而不是将其视为黑盒。

关键词

R包 torch接口 神经网络 tidymodels 超参数调优 代码生成

引用

@article{arxiv.2607.03832,
  title  = {kindling: A Higher-Level torch Interface for Generating, Training, and Tuning Neural Networks in R},
  author = {Antoine Soetewey and Joshua Marie},
  journal= {arXiv preprint arXiv:2607.03832},
  year   = {2026}
}