alephpi/Texo

A minimalist SOTA LaTeX OCR model with only 20M parameters, running in browser. Full training pipeline available for self-reproduction. | 超轻量SOTA LaTeX公式识别模型,仅20M参数量,可在浏览器中运行。训练全流程代码开源,以便自学复现。

解决的问题

Texo 是一个轻量级的开源 LaTeX OCR 模型,旨在将数学公式图像转换为 LaTeX 代码。它解决了 STEM 学习者和研究人员在数学笔记数字化过程中对快速、易用且精确工具的需求。

工作原理

Texo 是 PPFormulaNet-S 模型的蒸馏版本,随后在 UniMERNet-1M 数据集上进行了微调。它结合了模型编码器和解码器来识别数学图像中的模式,并将其转换为 LaTeX 命令的封闭词汇表。

适用对象

该项目主要面向需要快速将手写或印刷数学公式转换为 LaTeX 格式的 STEM 学生、AI 学习者和研究人员。

亮点

  • 极简设计:仅包含 2000 万个参数,比 UniMERNet-T (107M) 等其他 SOTA 模型小得多。
  • 高性能:在 UniMERNet-Test 数据集上的 BLEU 和编辑距离指标方面,保持了与大型模型相当的性能。
  • 消费级训练:可以在消费级 GPU 上进行训练,最低要求为 16GB GPU 显存。
  • 浏览器就绪:通过配套的 Web 项目支持直接在浏览器中运行。
  • 快速推理:针对轻量化和快速推理速度进行了优化。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • Dispatch