头版Folia Daily Briefing
← 返回头版
科技人工智能

8美元微控制器上成功训练小型语言模型

开源项目成功在价格仅为8美元的ESP32-S3微控制器上从零开始训练了一个包含31.9万参数的语言模型[1]。该项目采用克林贡语言作为测试语言,实现了完整的训练循环完全在芯片上进行,包括前向传播、反向传播和权重更新[1]。这一成果证明了微控制器可以独立学习,无需依赖预训练模型或外部数据源[1]。

项目采用单块transformer架构,配备单头因果注意力机制和共享权重设计[1]。训练过程中,权重占用内存1.3 MB,训练阶段总共需要数MB PSRAM,耗时数小时至数天,最终模型损失值约为1.87[1]。所有梯度通过手工推导的C代码实现,相对误差控制在1.07e-08以内,词汇表包含约31个符号[1]。该项目采用Apache 2.0开源许可证发布[1]。


ESP32-S3微控制器小语言模型训练克林贡语言边缘AI嵌入式机器学习

来源

  1. Hacker NewsAn SLM trained on $8 ESP32-S3