arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02456cs.LGcs.CL
收藏

tnGPS:通过大语言模型(LLMs)发现未知的张量网络结构搜索算法

tnGPS: Discovering Unknown Tensor Network Structure Search Algorithms via Large Language Models (LLMs)

School of Automation, Guangdong University of Technology(广东工业大学自动化学院);RIKEN Center for Advanced Intelligence Project (RIKEN-AIP)(理化学研究所先进智能中心);Tencent Inc.(腾讯公司)

Junhua Zeng, Chao Li, Zhun Sun, Qibin Zhao, Guoxu Zhou

AI总结:

针对张量网络结构搜索(TN-SS)现有手工启发式算法性能差的问题,提出基于大语言模型的自动算法发现框架tnGPS,通过模拟专家工作流的提示词流水线迭代生成新算法,在基准测试中超越了现有最先进方法。

摘要:

张量网络对于极高维表示非常高效,但其模型选择(即张量网络结构搜索,TN-SS)是一个极具挑战性的问题。尽管已有一些研究针对TN-SS,但现有的大多数算法都是人工设计的启发式方法,性能较差,且受限于维度灾难和局部收敛。在本研究中,我们跳出传统框架,研究如何利用大语言模型(LLMs)自动发现新的TN-SS算法,从而取代人类专家的参与。通过观察人类专家在研究中如何创新,我们对其通用工作流进行建模,并提出了一种名为tnGPS的自动算法发现框架。该框架是一个精细的提示词流水线,指导LLMs通过迭代优化和增强来生成新的TN-SS算法。实验结果表明,与当前最先进的方法相比,tnGPS发现的算法在基准测试中展现出更优越的性能。

英文摘要:

Tensor networks are efficient for extremely high-dimensional representation, but their model selection, known as tensor network structure search (TN-SS), is a challenging problem. Although several works have targeted TN-SS, most existing algorithms are manually crafted heuristics with poor performance, suffering from the curse of dimensionality and local convergence. In this work, we jump out of the box, studying how to harness large language models (LLMs) to automatically discover new TN-SS algorithms, replacing the involvement of human experts. By observing how human experts innovate in research, we model their common workflow and propose an automatic algorithm discovery framework called tnGPS. The proposed framework is an elaborate prompting pipeline that instruct LLMs to generate new TN-SS algorithms through iterative refinement and enhancement. The experimental results demonstrate that the algorithms discovered by tnGPS exhibit superior performance in benchmarks compared to the current state-of-the-art methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑