跳到正文
OpenReview · Representation learning· Yun Yi; Haokui Zhang; Rong Xiao; Nannan Wang; Xiaoyu Wang·· 2023-09-22精选AI 评分74

NAR-Former V2:重新审视用于通用神经网络表示学习的 Transformer

NAR-Former V2: Rethinking Transformer for Universal Neural Network Representation Learning

AI 导读

基于摘要分析。该研究关注神经网络架构本身的表示学习,目标是在不实际训练或部署候选网络的情况下预测其属性,辅助网络设计与部署。作者提出 NAR-Former V2,以修改后的 Transformer 同时表示单元结构网络与完整网络。 方法先将网络视为图,通过 tokenizer 将其编码为序列,再将 GNN 的归纳表示学习能力引入 Transformer,以改善对未见架构的泛化;此外还提出若干面向图结构表示的修改。摘要未说明具体编码规则、归纳机制的实现、损失函数及这些修改的组成,不能据此判断其如何保留拓扑关系或处理不同规模的网络。 结果方面,作者报告:在 NNLQP 数据集的完整网络编码与延迟预测任务上,该方法明显优于 GNN 方法 NNLP;在 NASBench101 与 NASBench201 的单元结构网络准确率预测任务上,与其他先进方法取得高度可比的表现。摘要未提供具体分数、指标、划分或目标硬件条件,实验协议、消融及统计信息尚未验证。作者提供了代码链接,但代码可运行性与复现条件尚未核验。 平台推测(待验证):迁移假设是利用架构表示预测器降低 EEG 候选网络的逐一训练与部署评估成本,而非直接学习 EEG 信号。可复用部分是架构图编码与属性预测框架;需核对 tokenizer 对 EEG 常用算子的支持,并收集任务特定的架构—性能或硬件延迟标签。原方法依赖架构图及其目标属性监督,所需数据规模尚未验证。EEG 的低信噪比、被试和通道差异,以及小样本评估波动,可能使准确率标签不稳定,削弱架构层面的泛化。一个可证伪的小实验是在固定 EEG 数据划分与训练预算下,为一组候选架构采集性能标签,按架构留出测试,与 GNN 预测器比较误差及排序一致性;这属于迁移验证建议,不是论文已完成的实验。已有 EEG 相关工作尚未检索确认。

阅读价值

值得阅读其将 GNN 的归纳表示学习能力融入 Transformer、统一处理单元结构与完整网络的机制,但用于 EEG 网络设计的价值尚未验证。

来源:OpenReview · Representation learning · openreview.net