中文

HATSolver:基于层次注意力Transformer学习Gröbner基

机器学习 2025-12-18 v1 人工智能

摘要

在NeurIPS 2024,Kera等人引入了使用Transformer计算Gröbner基的方法。本文通过应用层次注意力Transformer(HATs)来解决通过Gröbner基计算求解多变量多项式方程组的问题。HAT架构包含树形归纳偏置,使其能够建模数据中存在的层次关系,从而相较于常规平面注意力模型实现显著的计算节省。我们推广到任意深度,并包含详细的计算成本分析。结合课程学习,我们的方法解决的实例远大于Kera等人(2024 Learning to compute Groebner bases)中的实例。

关键词

引用

@article{arxiv.2512.14722,
  title  = {HATSolver: Learning Groebner Bases with Hierarchical Attention Transformers},
  author = {Mohamed Malhou and Ludovic Perret and Kristin Lauter},
  journal= {arXiv preprint arXiv:2512.14722},
  year   = {2025}
}