中文

基于Transformer架构的神经网络架构搜索方法

计算与语言 2025-11-03 v1 人工智能 机器学习 神经与进化计算

摘要

本文提出了一种基于Transformer架构的神经网络架构搜索方法,搜索不同编码器和解码器组合的交叉多头注意力计算方式。为了寻找具有更好翻译结果的神经网络结构,我们除了BLEU分数外,还考虑囊惑度作为算法的辅助评估指标,并通过多目标遗传算法不断改进种群中每个神经网络。实验结果表明,本算法搜索得到的神经网络结构均优于所有基线模型,引入辅助评估指标相较于仅考虑BLEU分数作为评估指标,能够找到更好的模型。

关键词

引用

@article{arxiv.2505.01314,
  title  = {A Transformer-based Neural Architecture Search Method},
  author = {Shang Wang and Huanrong Tang and Jianquan Ouyang},
  journal= {arXiv preprint arXiv:2505.01314},
  year   = {2025}
}

备注

GECCO 2023